Update app.py

app.py

@@ -53,11 +53,15 @@ def get_translators(platform):
     else:
         client = pymongo.MongoClient(os.environ.get("MONGO_URI"))
         source_db = client.deepsync_prod
-    translators = source_db.translators.find({})
+    translators = list(source_db.translators.find({}))
     transaltors_data = []
+    if len(translators):
+        desired_ids = [t["translator"] for t in translators]
+        companies_cursor = source_db.companies.find({'_id': {'$in': desired_ids}})
+        companies = {c["_id"]: c for c in companies_cursor}
     for translator in translators:
         try:
-            data = source_db.companies.find_one({"_id": translator["translator"]})
+            data = companies[translator["translator"]]
             transaltors_data.append({
                 "_id": str(translator["_id"]),
                 "Name": data["name"],
@@ -111,11 +115,15 @@ def get_translator_names(platform):
     else:
         client = pymongo.MongoClient(os.environ.get("MONGO_URI"))
         source_db = client.deepsync_prod
-    translators = source_db.translators.find({})
+    translators = list(source_db.translators.find({}))
     translators_names = []
+    if len(translators):
+        desired_ids = [t["translator"] for t in translators]
+        companies_cursor = source_db.companies.find({'_id': {'$in': desired_ids}})
+        companies = {c["_id"]: c for c in companies_cursor}
     for translator in translators:
         try:
-            data = source_db.companies.find_one({"_id": translator["translator"]})
+            data = companies[translator["translator"]]
             translators_names.append(data["name"])
         except:
             pass
@@ -177,7 +185,7 @@ def update_translator(platform, translator_name, source_languages, target_langua
         {"_id": translator_company_ids[company_id]}, 
         {"$set": updateable_data}
     )
-    return f"Updated translator accent succesfully."
+    return f"Updated translator details succesfully."
 
 
 def get_videos_under_review(status_keys):
@@ -190,18 +198,21 @@ def get_videos_under_review(status_keys):
         response = requests.get(API_URL, headers=headers)
         if response.status_code == 200:
             options = [f"{str(v['_id'])} / {v['title']}" for v in response.json()['data'] if 'title' in v]
-            return gr.Dropdown.update(choices=options, value=None)
+            return gr.Dropdown.update(choices=options, value=None), ""
         else:
-            return gr.Dropdown.update(choices=[], value=None)
+            return gr.Dropdown.update(choices=[], value=None), ""
     else:
         options = []
+        count = {}
         for status in status_keys:
             API_URL = f"https://api.dev-env.deepsync.co/api/v1/dashboard/dubbed/videos/review/requested/{video_status[status]}"
             response = requests.get(API_URL, headers=headers)
             if response.status_code == 200:
                 current_options = [f"{str(v['_id'])} / {v['title']}" for v in response.json()['data'] if 'title' in v]
+                count[status] = len(current_options)
                 options.extend(current_options)
-        return gr.Dropdown.update(choices=options, value=None)
+        data_text = "\n".join([f"{k} : {v}" for k,v in count.items()])
+        return gr.Dropdown.update(choices=options, value=None), data_text
 
 
 def get_traslators_assign(video):
@@ -211,27 +222,58 @@ def get_traslators_assign(video):
     video = source_db.dubbedvideos.find_one({"_id": ObjectId(video_id)})
     source_language = video["sourceLanguage"].split("-")[0]
     target_language = video["targetLanguage"].split("-")[0]
-    translators = source_db.translators.find({"sourceAccent": {"$elemMatch": {"$eq": source_language}}, "targetAccent": {"$elemMatch": {"$eq": target_language}}})
+    translators = list(source_db.translators.find({"sourceAccent": {"$elemMatch": {"$eq": source_language}}, "targetAccent": {"$elemMatch": {"$eq": target_language}}}))
     translators_list = []
+    if len(translators):
+        desired_ids = [t["translator"] for t in translators]
+        companies_cursor = source_db.companies.find({'_id': {'$in': desired_ids}})
+        companies = {c["_id"]: c for c in companies_cursor}
     for translator in translators:
-        company = source_db.companies.find_one({"_id": translator["translator"]})
-        translators_list.append(f"{company['name']} / {str(translator['_id'])}")
+        company = companies[translator["translator"]]
+        translators_list.append(f"{company['name']} / {company['email']} / {str(translator['_id'])}")
     
     button = gr.Button.update(interactive=False)
+    current_translator = None 
+    current_qa = None
     if "videoReviewInformation" in video:
         if video["videoReviewInformation"]["status"] in {"NOT_STARTED", "DUBBING_STARTED", "WAITING_QA"}:
             button = gr.Button.update(interactive=True)
-    return gr.Dropdown.update(choices=translators_list), language_maps[source_language], language_maps[target_language], button
+        if video["videoReviewInformation"].get("assignedTranslator") is not None:
+            str_id = str(video["videoReviewInformation"].get("assignedTranslator"))
+            value = [v for v in translators_list if str_id in v]
+            current_translator = value[0] if len(value) else None
+        if video["videoReviewInformation"].get("assignedQA") is not None:
+            str_id = str(video["videoReviewInformation"].get("assignedTranslator"))
+            value = [v for v in translators_list if str_id in v]
+            current_qa = value[0] if len(value) else None
+    return gr.Dropdown.update(choices=translators_list, value=current_translator), gr.Dropdown.update(choices=translators_list, value=current_qa), language_maps[source_language], language_maps[target_language], button, button
 
 
-def assign_translator(translator, video, role):
+def assign_translator(translator, video):
     translator_id = translator.rsplit("/", maxsplit=1)[1].strip()
     video_id = video.split("/", maxsplit=1)[0].strip()
     headers = {
         "x-api-key": os.environ.get('X-API-KEY'),
         "Content-Type": "application/json"
     }
-    role = "QA" if role == "QA" else "translator"
+    role = "translator"
+    API_URL = f"https://api.dev-env.deepsync.co/api/v1/dashboard/dubbed/video/assign/{role}"
+    data = {
+        "videoId": video_id,
+        "userId": translator_id
+    }
+    response = requests.post(API_URL, json=data, headers=headers)
+    return str(response.status_code)
+
+
+def assign_qa(qa, video):
+    translator_id = qa.rsplit("/", maxsplit=1)[1].strip()
+    video_id = video.split("/", maxsplit=1)[0].strip()
+    headers = {
+        "x-api-key": os.environ.get('X-API-KEY'),
+        "Content-Type": "application/json"
+    }
+    role = "QA"
     API_URL = f"https://api.dev-env.deepsync.co/api/v1/dashboard/dubbed/video/assign/{role}"
     data = {
         "videoId": video_id,
@@ -271,14 +313,18 @@ with gr.Blocks() as demo:
     with gr.Tab("Assign"):
         gr.Markdown("Assign Translator / QA")
         filter_videos_assign = gr.CheckboxGroup(['Not Started', 'Dubbing Started', 'Waiting QA', 'Sent to Client', 'Approved', 'Rejected', 'Churned', 'Re-review Requested'], label="Filter videos. Do not select any of them for videos without request info.")
+        video_details = gr.Textbox(label="Video Details")
         refresh_assign = gr.Button("Load Videos")
         under_review_videos_list = gr.Dropdown([], label="Video IDs")
         with gr.Row():
             source_language_box = gr.Textbox(label="Source Language")
             target_language_box = gr.Textbox(label="Target Language")
-        translator_name_assign = gr.Dropdown([], label="Translators")
-        role_assign = gr.Radio(["Translator", "QA"], label="Role", value="Translator")
-        submit_assign_button = gr.Button("Assign Role")
+        with gr.Row():
+            translator_name_assign = gr.Dropdown([], label="Translator")
+            submit_tr_assign_button = gr.Button("Assign Translator")
+        with gr.Row():
+            qa_name_assign = gr.Dropdown([], label="QA")
+            submit_qa_assign_button = gr.Button("Assign QA")
         assign_logs = gr.Textbox(label="Logs")
 
     
@@ -296,9 +342,10 @@ with gr.Blocks() as demo:
     update_language.click(update_translator, [source_platform_update, translator_name, source_languages_update, target_languages_update, contact_num_update, pricing_update], result_update)
 
     # For Assign
-    refresh_assign.click(get_videos_under_review, filter_videos_assign, under_review_videos_list)
-    under_review_videos_list.change(get_traslators_assign, under_review_videos_list, [translator_name_assign, source_language_box, target_language_box, submit_assign_button])
-    submit_assign_button.click(assign_translator, [translator_name_assign, under_review_videos_list, role_assign], assign_logs)
+    refresh_assign.click(get_videos_under_review, filter_videos_assign, [under_review_videos_list, video_details])
+    under_review_videos_list.change(get_traslators_assign, under_review_videos_list, [translator_name_assign, qa_name_assign, source_language_box, target_language_box, submit_tr_assign_button, submit_qa_assign_button])
+    submit_tr_assign_button.click(assign_translator, [translator_name_assign, under_review_videos_list], assign_logs)
+    submit_qa_assign_button.click(assign_qa, [qa_name_assign, under_review_videos_list], assign_logs)
 
 
 if __name__=="__main__":