Update app.py

app.py

@@ -102,8 +102,31 @@ def read_json(r, step):
     return body
 
 
+PATHS = {"generate": "/generate", "get-status": "/get-status", "get-file": "/get-file"}
+
+
+def route_exists(path):
+    """GET a route without parameters. Only a router-level 404 means it's missing.
+    A GET never creates a job or spends credits."""
+    try:
+        r = requests.get(f"{BASE}{path}", headers=HEADERS, timeout=15)
+    except Exception as e:
+        print(f"[startup] probe GET {path} failed: {e}")
+        return False
+    missing = False
+    if r.status_code == 404:
+        try:
+            missing = r.json().get("detail") == "Not Found"
+        except ValueError:
+            missing = True
+    print(f"[startup] probe GET {path} -> HTTP {r.status_code}"
+          f"{' (route missing)' if missing else ' (route exists)'}")
+    return not missing
+
+
 def startup_check():
-    """Log whether the API is reachable and the key works (key is never printed)."""
+    """Log whether the API is reachable and the key works (key is never printed),
+    and pick the right path for each endpoint (with or without /v2)."""
     if not API_KEY:
         print("[startup] DRAMA_API_KEY is not set; demo is switched off.")
         return
@@ -114,6 +137,14 @@ def startup_check():
             print(f"[startup] GET {path} -> HTTP {r.status_code}: {r.text[:300]}")
         except Exception as e:
             print(f"[startup] GET {path} failed: {e}")
+    for name in PATHS:
+        for candidate in (f"/{name}", f"/v2/{name}"):
+            if route_exists(candidate):
+                PATHS[name] = candidate
+                break
+        else:
+            print(f"[startup] WARNING: no route found for {name}; keeping {PATHS[name]}")
+    print(f"[startup] using paths: {PATHS}")
 
 
 def build_drama(mood, excited, angry):
@@ -134,7 +165,7 @@ def build_drama(mood, excited, angry):
 def submit_job(video, audio, drama):
     with open(video, "rb") as v, open(audio, "rb") as a:
         r = requests.post(
-            f"{BASE}/generate",
+            f"{BASE}{PATHS['generate']}",
             headers=HEADERS,
             data={
                 "skip_find_face": "1",
@@ -157,14 +188,14 @@ def submit_job(video, audio, drama):
 
 
 def job_status(request_id):
-    r = requests.get(f"{BASE}/get-status", headers=HEADERS,
+    r = requests.get(f"{BASE}{PATHS['get-status']}", headers=HEADERS,
                      params={"request_id": request_id}, timeout=30)
     body = read_json(r, "get-status")
     return str((body.get("data") or {}).get("status", "")).lower()
 
 
 def download(request_id, workdir):
-    r = requests.get(f"{BASE}/get-file", headers=HEADERS,
+    r = requests.get(f"{BASE}{PATHS['get-file']}", headers=HEADERS,
                      params={"request_id": request_id}, timeout=300, stream=True)
     if "video" not in r.headers.get("Content-Type", ""):
         raise gr.Error("The result isn't ready to download yet. Please try again.")