Update app.py

app.py

@@ -53,7 +53,7 @@ def process_video(drive_link, progress=gr.Progress()):
             t2 = time.time()
             exec_time = t2 - t1
             out = st["output"]
-            dl_link = to_direct_download(out["drive_link"]) or out["drive_link"]
+            drive_links = out.get("drive_links", [])
             break
         if status == "FAILED":
             progress(1.0, desc="Job Failed")
@@ -65,21 +65,28 @@ def process_video(drive_link, progress=gr.Progress()):
     progress(1.0, desc="Completed")
     delay = t1 - t0
 
-    details_md = f"""
-**Drive Link:**  
-`{dl_link}`
-
-**Job ID:** `{job_id}`  
-**Status:** `{status}`  
+    details_md = f"**Job ID:** `{job_id}`  \n**Status:** `{status}`\n\n**Submission Delay:** `{delay:.1f}s`  \n**Execution Time:** `{exec_time:.1f}s`\n\n"
+
+    # Compose all download links, one per drive
+    links_md = ""
+    button_md = ""
+    for item in drive_links:
+        drive = item.get("drive", "Unknown")
+        web_link = item.get("webViewLink", "")
+        dl_link = to_direct_download(web_link) or web_link
+        links_md += f"**{drive} Drive Link:**  \n`{web_link}`\n\n"
+        # Add download button per drive
+        button_md += (
+            f'<a href="{web_link}" target="_blank" style="display:inline-block;padding:8px 16px;background:#4CAF50;color:white;border-radius:4px;text-decoration:none;margin-right:8px;">'
+            f'Open {drive} Download</a> '
+        )
 
-**Submission Delay:** `{delay:.1f}s`  
-**Execution Time:** `{exec_time:.1f}s`  
-"""
+    details_md = links_md + details_md
 
-    # Render a styled HTML link inside Markdown
-    button_md = f'<a href="{dl_link}" target="_blank" style="display:inline-block;padding:8px 16px;background:#4CAF50;color:white;border-radius:4px;text-decoration:none;">Open Download</a>'
+    # Optionally, show first link as the main download link
+    primary_dl_link = drive_links[0]["webViewLink"] if drive_links else ""
 
-    return "✅ Completed!", dl_link, details_md, button_md
+    return "✅ Completed!", primary_dl_link, details_md, button_md
 
 def create_ui():
     with gr.Blocks(theme=gr.themes.Soft()) as app: