Update app.py

app.py

@@ -9,7 +9,8 @@ import requests
 load_dotenv()
 
 video_status = {
-    "Not Started": "NOT_STARTED",
+    "Not Assigned": "NOT_STARTED",
+    "Under Review": "NOT_STARTED",
     "Dubbing Started": "DUBBING_STARTED",
     "Waiting QA": "WAITING_QA",
     "Sent to Client": "SENT_TO_CLIENT",
@@ -234,7 +235,13 @@ def get_videos_under_review(platform, status_keys):
             API_URL = f"{api_url}/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]
+                response_data = response.json()['data']
+                if status == "Not Assigned":
+                    response_data = list(filter(lambda x: x["videoReviewInformation"].get("assignedTranslator") is None, response_data))
+                if status == "Under Review":
+                    response_data = list(filter(lambda x: x["videoReviewInformation"].get("assignedTranslator") is not None, response_data))
+                current_options = [f"{str(v['_id'])} / {v['title']}" for v in response_data if 'title' in v]
+                
                 count[status] = len(current_options)
                 options.extend(current_options)
         data_text = "\n".join([f"{k} : {v} videos" for k,v in count.items()])
@@ -371,7 +378,7 @@ with gr.Blocks() as demo:
     with gr.Tab("Assign"):
         gr.Markdown("Assign Translator / QA")
         source_platform_assign = gr.Radio(["dev", "prod"], value="dev", label="Source Platform")
-        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.")
+        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.")
         video_details = gr.Textbox(label="Video Details")
         refresh_assign = gr.Button("Load Videos")
         under_review_videos_list = gr.Dropdown([], label="Video IDs")