From d7309571000855c40cba218672cd7c5e14fc505f Mon Sep 17 00:00:00 2001
From: sudan45 <sudan1bhandari@gmail.com>
Date: Mon, 13 Jan 2025 12:25:18 +0545
Subject: [PATCH] Update status of draft entry

---
 apps/deepl_integration/handlers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/apps/deepl_integration/handlers.py b/apps/deepl_integration/handlers.py
index 7731c5ab11..43771d6a5b 100644
--- a/apps/deepl_integration/handlers.py
+++ b/apps/deepl_integration/handlers.py
@@ -1376,10 +1376,10 @@ def save_data(cls, lead, data_url):
                 project=lead.project,
                 lead=lead,
                 excerpt=model_preds['text'],
-                prediction_status=DraftEntry.PredictionStatus.STARTED,
+                prediction_status=DraftEntry.PredictionStatus.DONE,
                 type=DraftEntry.Type.AUTO
             )
-            if model_preds['geolocations']:
+            if model_preds.get('geolocations'):
                 geo_areas_qs = GeoAreaGqlFilterSet(
                     data={'titles': [geo['entity'] for geo in model_preds['geolocations']]},
                     queryset=GeoArea.get_for_project(lead.project)