Update app.py

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 from bson import ObjectId
 from dotenv import load_dotenv
 import requests
+from collections import defaultdict
 
 load_dotenv()
 
@@ -214,7 +215,8 @@ def update_translator(platform, translator_name, source_languages, target_langua
     return f"Updated translator details succesfully."
 
 
-def get_videos_under_review(platform, status_keys):
+def get_videos_under_review(platform, status_key):
+    status_keys = [status_key]
     if platform == "dev":
         api_url = "https://api.dev-env.deepsync.co/api/v1"
     else:
@@ -284,13 +286,27 @@ def get_traslators_assign(platform, video):
 
     translators = list(source_db.translators.find({"sourceAccent": {"$elemMatch": {"$eq": source_language}}, "targetAccent": {"$elemMatch": {"$eq": target_language}}}))
     translators_list = []
+
+    pending_videos = []
+    for status in ["NOT_STARTED", "DUBBING_STARTED", "WAITING_QA"]:
+        API_URL = f"{api_url}/dashboard/dubbed/videos/review/requested/{status}"
+        response = requests.get(API_URL, headers=HEADERS)
+        if response.status_code == 200:
+            response_data = response.json()['data']
+            pending_videos.extend(response_data)
+    translator_pv_count = defaultdict(lambda: 0)
+    for pv in pending_videos:
+        if video.get("videoReviewInformation", {}).get("assignedTranslator") is not None:
+            trs = str(video.get("videoReviewInformation", {}).get("assignedTranslator"))
+            translator_pv_count[trs] += 1
+
     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 = companies[translator["translator"]]
-        translators_list.append(f"{company['name']} / {company['email']} / {str(translator['_id'])}")
+        translators_list.append(f"{company['name']} / {company['email']} / {str(translator['_id'])} / Pending Videos: {translator_pv_count[str(translator['_id'])]}")
     
     button = gr.Button.update(interactive=False)
     current_translator = None 
@@ -310,7 +326,7 @@ def get_traslators_assign(platform, video):
 
 
 def assign_translator(platform, translator, video):
-    translator_id = translator.rsplit("/", maxsplit=1)[1].strip()
+    translator_id = translator.split("/")[2].strip()
     video_id = video.split("/", maxsplit=1)[0].strip()
     role = "translator"
     if platform == "dev":
@@ -329,7 +345,7 @@ def assign_translator(platform, translator, video):
 
 
 def assign_qa(platform, qa, video):
-    translator_id = qa.rsplit("/", maxsplit=1)[1].strip()
+    translator_id = qa.split("/")[2].strip()
     video_id = video.split("/", maxsplit=1)[0].strip()
     role = "QA"
     if platform == "dev":
@@ -378,7 +394,7 @@ with gr.Blocks() as demo:
     with gr.Tab("Assign"):
         gr.Markdown("Assign Translator / QA")
         source_platform_assign = gr.Radio(["dev", "prod"], value="prod", label="Source Platform")
-        filter_videos_assign = gr.CheckboxGroup(['Not Assigned', 'Under Review', '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.")
+        filter_videos_assign = gr.Radio(['Not Assigned', 'Under Review', '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")