Update main.py

main.py

@@ -33,10 +33,9 @@ WORKFLOW_FILE  = os.getenv("WORKFLOW_FILE", "video-download.yml")
 # e.g. https://my-api.hf.space  or  https://xxxxx.ngrok.io
 PUBLIC_URL     = os.getenv("PUBLIC_URL", "https://your-api-public-url.com")
 
-N8N_TRIGGER_URL = os.getenv(
-    "N8N_TRIGGER_URL",
-    "https://egauto-n8n.hf.space/webhook/run%20gethup%20action"
-)
+# اختياري - لو موجود الـ API هو اللي يشغل n8n
+# لو None، n8n هو اللي يبعت callback للـ API
+N8N_TRIGGER_URL = os.getenv("N8N_TRIGGER_URL", None)
 
 VIDEOS_DIR = Path("videos")
 VIDEOS_DIR.mkdir(exist_ok=True)
@@ -47,9 +46,11 @@ jobs: dict[str, dict] = {}
 
 # ─── App ───────────────────────────────────────────────────────────────────────
 app = FastAPI(
-    title="Video Download Middleware",
+    title="EGDownloader API",
     description="Bridges client ↔ n8n/GitHub Action video pipeline",
     version="1.0.0",
+    docs_url="/",
+    redoc_url="/redoc",
 )
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(message)s")
@@ -80,8 +81,16 @@ async def download_shorts(body: DownloadRequest):
 @app.post("/download", summary="Start a video download job")
 async def start_download(body: DownloadRequest):
     """
-    Triggers the n8n → GitHub Action pipeline.
-    Returns a job_id you can poll with GET /status/{job_id}.
+    طريقتين للشغل:
+
+    1) Push mode (الـ default - اللي عندك دلوقتي):
+       N8N_TRIGGER_URL = None
+       n8n هو اللي يشغل GitHub Action وبعدين يبعت callback على /n8n/callback/{job_id}
+       الـ client بس يعمل polling على /status/{job_id}
+
+    2) Trigger mode:
+       N8N_TRIGGER_URL = "https://..."
+       الـ API هو اللي يبعت trigger لـ n8n، وn8n يرد بـ callback
     """
     job_id = str(uuid.uuid4())
 
@@ -93,29 +102,39 @@ async def start_download(body: DownloadRequest):
         "run_id":     None,
         "error":      None,
         "created_at": time.time(),
+        "mode":       "trigger" if N8N_TRIGGER_URL else "push",
     }
 
-    # Build the payload n8n expects (mirrors the "run action" webhook pinData)
-    payload = {
-        "url":         body.url,
-        "cookies_txt": body.cookies_txt or "",
-        # Tell n8n where to POST the finished file
-        "n8n_webhook": f"{PUBLIC_URL}/n8n/callback/{job_id}",
-    }
+    # ── Trigger mode: الـ API يشغل n8n ────────────────────────────────────────
+    if N8N_TRIGGER_URL:
+        payload = {
+            "url":         body.url,
+            "cookies_txt": body.cookies_txt or "",
+            "n8n_webhook": f"{PUBLIC_URL}/n8n/callback/{job_id}",
+        }
+        try:
+            async with httpx.AsyncClient(timeout=15) as client:
+                r = await client.post(N8N_TRIGGER_URL, json=payload)
+                r.raise_for_status()
+            jobs[job_id]["status"] = "processing"
+            log.info(f"[{job_id}] triggered n8n → {r.status_code}")
+        except Exception as exc:
+            jobs[job_id]["status"] = "failed"
+            jobs[job_id]["error"]  = str(exc)
+            log.error(f"[{job_id}] failed to trigger n8n: {exc}")
+            raise HTTPException(status_code=502, detail=f"Could not trigger pipeline: {exc}")
 
-    try:
-        async with httpx.AsyncClient(timeout=15) as client:
-            r = await client.post(N8N_TRIGGER_URL, json=payload)
-            r.raise_for_status()
-        jobs[job_id]["status"] = "processing"
-        log.info(f"[{job_id}] job triggered → n8n responded {r.status_code}")
-    except Exception as exc:
-        jobs[job_id]["status"] = "failed"
-        jobs[job_id]["error"]  = str(exc)
-        log.error(f"[{job_id}] failed to trigger n8n: {exc}")
-        raise HTTPException(status_code=502, detail=f"Could not trigger pipeline: {exc}")
+    # ── Push mode: n8n هو اللي هيبعت callback لما يخلص ───────────────────────
+    else:
+        jobs[job_id]["status"] = "pending"
+        log.info(f"[{job_id}] job created (push mode) - waiting for n8n callback")
 
-    return {"job_id": job_id, "status": "processing"}
+    return {
+        "job_id":  job_id,
+        "status":  jobs[job_id]["status"],
+        "mode":    jobs[job_id]["mode"],
+        "callback_url": f"{PUBLIC_URL}/n8n/callback/{job_id}",
+    }
 
 
 # ══════════════════════════════════════════════════════════════════════════════