Upload 4 files

app.py

@@ -426,6 +426,14 @@ def run_v3_status():
     except Exception as e:
         return f"Error: {e}"
 
+def run_v3_retry():
+    cmd = ["python", "pipeline/orchestrator.py", "retry"]
+    try:
+        res = subprocess.run(cmd, capture_output=True, text=True, cwd=os.path.dirname(os.path.abspath(__file__)))
+        return f"--- RETRY OUTPUT ---\n{res.stdout}\n{res.stderr}"
+    except Exception as e:
+        return f"Error: {e}"
+
 with gr.Blocks() as app:
     gr.Markdown("# YouTube Video Orchestrator Host")
     gr.Markdown("Deploy this to Hugging Face Spaces or run locally. It fetches pending links from your database and delegates them to FastAPI workers or Kaggle.")
@@ -442,12 +450,14 @@ with gr.Blocks() as app:
                 v3_launch_btn = gr.Button("Launch Workers (--push)", variant="primary")
                 v3_sync_btn = gr.Button("Sync DB & Status")
                 v3_status_btn = gr.Button("Check Status")
+                v3_retry_btn = gr.Button("Retry Failed Videos")
                 
             v3_output_log = gr.Textbox(label="Terminal Output", lines=15)
             
             v3_launch_btn.click(fn=run_v3_launch, inputs=[v3_max_workers, v3_videos], outputs=v3_output_log)
             v3_sync_btn.click(fn=run_v3_sync, inputs=[], outputs=v3_output_log)
             v3_status_btn.click(fn=run_v3_status, inputs=[], outputs=v3_output_log)
+            v3_retry_btn.click(fn=run_v3_retry, inputs=[], outputs=v3_output_log)
 
         with gr.Tab("Database Sync"):
             gr.Markdown("Sync (Add) new API keys and YouTube Links into your database.")

pipeline/orchestrator.py

@@ -179,7 +179,7 @@ class Tracker:
 
     def retry_failed(self, max_retries=3):
         with self._conn() as c:
-            r = c.execute("UPDATE videos SET status=? WHERE status IN (?,?) AND retry_count<?",
+            r = c.execute("UPDATE videos SET status=?, retry_count=retry_count+1 WHERE status IN (?,?) AND retry_count<?",
                           (S_PENDING, S_FAILED, S_TIMEOUT, max_retries))
             return r.rowcount