Return correct page info

AI4Bharat · Jul 24, 2023 · a2f63d9 · a2f63d9
1 parent 930e66e
commit a2f63d9
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 23 deletions.
diff --git a/backend/translation/views.py b/backend/translation/views.py
@@ -1266,6 +1266,15 @@ def save_translation(request):
                         task.status = "COMPLETE"
                         task.save()
                         translation_obj.save()
+                        response = check_if_translation_correct(translation_obj, task)
+                        if type(response) == dict:
+                            return Response(
+                                {
+                                    "data": response["data"],
+                                    "message": response["message"],
+                                },
+                                status=status.HTTP_400_BAD_REQUEST,
+                            )
                         delete_indices = []
                         for index, sentence in enumerate(
                             translation_obj.payload["payload"]
@@ -1277,15 +1286,6 @@ def save_translation(request):
                         for ind in delete_indices:
                             translation_obj.payload["payload"].pop(ind)
                         translation_obj.save()
-                        response = check_if_translation_correct(translation_obj, task)
-                        if type(response) == dict:
-                            return Response(
-                                {
-                                    "data": response["data"],
-                                    "message": response["message"],
-                                },
-                                status=status.HTTP_400_BAD_REQUEST,
-                            )
                         message = change_active_status_of_next_tasks(
                             task, translation_obj
                         )
@@ -1374,6 +1374,15 @@ def save_translation(request):
                         limit, payload, start_offset, end_offset, translation_obj
                     )
                     translation_obj.save()
+                    translation_obj.save()
+                    task.status = "COMPLETE"
+                    task.save()
+                    response = check_if_translation_correct(translation_obj, task)
+                    if type(response) == dict:
+                        return Response(
+                            {"data": response["data"], "message": response["message"]},
+                            status=status.HTTP_400_BAD_REQUEST,
+                        )
                     delete_indices = []
                     for index, sentence in enumerate(
                         translation_obj.payload["payload"]
@@ -1384,15 +1393,6 @@ def save_translation(request):
                     delete_indices.reverse()
                     for ind in delete_indices:
                         translation_obj.payload["payload"].pop(ind)
-                    translation_obj.save()
-                    task.status = "COMPLETE"
-                    task.save()
-                    response = check_if_translation_correct(translation_obj, task)
-                    if type(response) == dict:
-                        return Response(
-                            {"data": response["data"], "message": response["message"]},
-                            status=status.HTTP_400_BAD_REQUEST,
-                        )
                     message = change_active_status_of_next_tasks(task, translation_obj)
                 else:
                     translation_obj = (

diff --git a/backend/voiceover/utils.py b/backend/voiceover/utils.py
@@ -402,11 +402,16 @@ def get_bad_sentences(translation_obj, target_language):
 def get_bad_sentences_in_progress(translation_obj, target_language):
     problem_sentences = []
     translation = translation_obj.payload
+    compare_with_index = -1
+    last_valid_index = -1
     for ind, text in enumerate(translation["payload"]):
-        if not compare_time(text["end_time"], text["start_time"])[0]:
+        if (
+            "text" in text.keys()
+            and not compare_time(text["end_time"], text["start_time"])[0]
+        ):
             problem_sentences.append(
                 {
-                    "index": ind % 50,
+                    "index": (ind % 50) + 1,
                     "page_number": (ind // 50) + 1,
                     "start_time": text["start_time"],
                     "end_time": text["end_time"],
@@ -415,8 +420,6 @@ def get_bad_sentences_in_progress(translation_obj, target_language):
                 }
             )
         if ind != 0 and ind < len(translation["payload"]):
-            compare_with_index = -1
-            last_valid_index = -1
             compare = False
             if "text" in translation["payload"][ind - 1] and "text" in text.keys():
                 compare_with_index = ind - 1
@@ -438,7 +441,7 @@ def get_bad_sentences_in_progress(translation_obj, target_language):
             ):
                 problem_sentences.append(
                     {
-                        "index": ind % 50,
+                        "index": (ind % 50) + 1,
                         "page_number": (ind // 50) + 1,
                         "start_time": text["start_time"],
                         "end_time": text["end_time"],