Update app.py

app.py

@@ -5,11 +5,9 @@ import pathlib
 import requests
 import gradio as gr
 
-# Endpoint Kie AI
 BASE = "https://api.kie.ai/api/v1"
 CREATE_URL = f"{BASE}/jobs/createTask"
 
-# Utility: headers dengan API key (BYOK atau dari Secrets)
 def make_headers(user_key: str | None):
     api_key = (user_key or os.getenv("KIE_API_KEY") or "").strip()
     if not api_key:
@@ -19,20 +17,15 @@ def make_headers(user_key: str | None):
         "Content-Type": "application/json",
     }
 
-# Buat task text-to-video
 def create_task(headers, prompt: str, aspect_ratio: str, duration: str | None):
     payload = {
         "model": "sora-2-text-to-video",
         "input": {
             "prompt": prompt,
-            # Sesuaikan dengan yang didukung Kie AI (portrait/landscape/9:16/16:9)
             "aspect_ratio": aspect_ratio,
-            # Demi kepatuhan, jangan bantu hapus watermark
-            "remove_watermark": False,
+            "remove_watermark": False,   # patuhi ToS
         },
     }
-
-    # Durasi opsional — hanya sertakan kalau diisi. Jika API Kie AI menolak, hapus field ini.
     if duration:
         try:
             d = int(duration)
@@ -43,10 +36,11 @@ def create_task(headers, prompt: str, aspect_ratio: str, duration: str | None):
 
     r = requests.post(CREATE_URL, headers=headers, json=payload, timeout=60)
     if r.status_code >= 400:
-        raise gr.Error(f"Gagal createTask: {r.status_code} {r.text}")
+        raise gr.Error(f"createTask gagal: {r.status_code} {r.text}")
 
     data = r.json()
-    # Ambil task id dan (jika ada) status_url yang diberikan server
+
+    # Ekstrak task_id & status_url jika disediakan
     task_id = (
         data.get("task_id") or data.get("taskId") or data.get("id") or
         data.get("data", {}).get("taskId") or data.get("data", {}).get("id")
@@ -55,19 +49,18 @@ def create_task(headers, prompt: str, aspect_ratio: str, duration: str | None):
         data.get("status_url") or data.get("statusUrl") or
         data.get("data", {}).get("status_url") or data.get("data", {}).get("statusUrl")
     )
+    raw_preview = json.dumps(data)[:400]
 
     if not task_id:
-        raise gr.Error(f"Tidak menemukan task_id di response: {json.dumps(data)[:500]}")
-
-    # Jika status_url tidak disediakan, coba pattern umum (cek docs Kie AI jika berbeda)
-    if not status_url:
-        status_url = f"{BASE}/jobs/task/{task_id}"
+        raise gr.Error(f"Tidak menemukan task_id pada response createTask: {raw_preview}")
 
-    return task_id, status_url, data
+    return task_id, status_url, raw_preview
 
-# Baca status + result url dari berbagai bentuk response
 def infer_status_and_result(obj: dict):
-    status = obj.get("status") or obj.get("state") or obj.get("task_status") or obj.get("data", {}).get("status")
+    status = (
+        obj.get("status") or obj.get("state") or
+        obj.get("task_status") or obj.get("data", {}).get("status")
+    )
     progress = obj.get("progress") or obj.get("data", {}).get("progress")
     result_url = (
         obj.get("result_url") or obj.get("output_url") or obj.get("video_url") or
@@ -76,16 +69,80 @@ def infer_status_and_result(obj: dict):
     )
     return status, progress, result_url
 
-# Polling status sampai selesai
-def poll_until_done(headers, status_url: str, interval=5, max_wait=60*40):
+def probe_status_url(headers, task_id: str):
+    # Coba beberapa pola umum sampai ada yang 200 OK
+    candidates = [
+        f"{BASE}/jobs/getTask?taskId={task_id}",
+        f"{BASE}/jobs/get-task?taskId={task_id}",
+        f"{BASE}/jobs/task?taskId={task_id}",
+        f"{BASE}/jobs/{task_id}",
+        f"{BASE}/jobs/status/{task_id}",
+        f"{BASE}/jobs/getTask/{task_id}",
+        f"{BASE}/jobs/task/{task_id}",
+    ]
+    for url in candidates:
+        try:
+            r = requests.get(url, headers=headers, timeout=30)
+            if r.status_code < 400:
+                # pastikan respons memang status task
+                try:
+                    data = r.json()
+                except Exception:
+                    continue
+                status, prog, _ = infer_status_and_result(data)
+                if status is not None:
+                    return url, f"[INFO] Menemukan status URL: {url} (status={status}, progress={prog})"
+        except Exception:
+            pass
+    # Beberapa API memerlukan POST untuk getTask
+    post_url = f"{BASE}/jobs/getTask"
+    try:
+        r = requests.post(post_url, headers=headers, json={"taskId": task_id}, timeout=30)
+        if r.status_code < 400:
+            data = r.json()
+            status, prog, _ = infer_status_and_result(data)
+            if status is not None:
+                return post_url + " (POST)", "[INFO] Menemukan status via POST /jobs/getTask"
+    except Exception:
+        pass
+    return None, "[ERROR] Tidak menemukan endpoint status yang cocok. Mohon cek dokumentasi Kie AI."
+
+def poll_until_done(headers, status_url: str, task_id: str, interval=5, max_wait=60*40):
     waited = 0
     last_status = None
+    used_post_mode = status_url.endswith("(POST)")  # indikator dari probe
+
     while waited <= max_wait:
-        r = requests.get(status_url, headers=headers, timeout=60)
+        try:
+            if used_post_mode:
+                r = requests.post(status_url.replace(" (POST)", ""), headers=headers, json={"taskId": task_id}, timeout=60)
+            else:
+                r = requests.get(status_url, headers=headers, timeout=60)
+        except Exception as e:
+            yield f"[WARN] Poll error: {e}", None
+            time.sleep(interval)
+            waited += interval
+            continue
+
+        if r.status_code == 404:
+            yield f"[WARN] Poll 404 di {status_url} — mencoba cari endpoint lain...", None
+            new_url, msg = probe_status_url(headers, task_id)
+            yield msg, None
+            if not new_url:
+                raise gr.Error("Gagal menemukan endpoint status. Cek logs dan dokumentasi Kie AI.")
+            status_url = new_url
+            used_post_mode = status_url.endswith("(POST)")
+            continue
+
         if r.status_code >= 400:
             yield f"[WARN] Poll failed: {r.status_code} {r.text}", None
         else:
-            data = r.json()
+            try:
+                data = r.json()
+            except Exception:
+                yield "[WARN] Response status bukan JSON yang valid.", None
+                data = {}
+
             status, progress, result_url = infer_status_and_result(data)
             if status != last_status:
                 yield f"{status} - progress: {progress}", None
@@ -96,14 +153,13 @@ def poll_until_done(headers, status_url: str, interval=5, max_wait=60*40):
                 yield "Selesai (server) — mengunduh hasil...", result_url
                 return
             if s in {"failed", "error", "canceled", "cancelled"}:
-                raise gr.Error(f"Task gagal: {json.dumps(data, indent=2)}")
+                raise gr.Error(f"Task gagal:\n{json.dumps(data, indent=2)}")
 
         time.sleep(interval)
         waited += interval
 
     raise gr.Error("Timeout menunggu task selesai.")
 
-# Unduh file hasil
 def download_result(url: str, task_id: str):
     if not url:
         raise gr.Error("Server tidak mengirimkan URL hasil.")
@@ -118,22 +174,29 @@ def download_result(url: str, task_id: str):
                     f.write(chunk)
     return str(out)
 
-# Gradio handler utama
 def run_job(user_api_key, prompt, aspect_ratio, duration):
     headers = make_headers(user_api_key)
 
     # 1) create task
-    task_id, status_url, raw_create = create_task(headers, prompt, aspect_ratio, duration)
-    yield f"Task dibuat: {task_id}\nPolling: {status_url}\nResponse: {json.dumps(raw_create)[:400]}...", None
+    task_id, status_url, raw_preview = create_task(headers, prompt, aspect_ratio, duration)
+    yield f"Task dibuat: {task_id}\nStatus URL (server): {status_url or '-'}\nCreate resp: {raw_preview}", None
+
+    # 2) tentukan status_url valid
+    if not status_url:
+        found, msg = probe_status_url(headers, task_id)
+        yield msg, None
+        if not found:
+            raise gr.Error("Tidak bisa menentukan endpoint status. Mohon kirim potongan full response createTask di sini.")
+        status_url = found
 
-    # 2) poll
+    # 3) poll
     result_link = None
-    for status_msg, maybe_url in poll_until_done(headers, status_url):
+    for status_msg, maybe_url in poll_until_done(headers, status_url, task_id):
         yield status_msg, None
         if maybe_url:
             result_link = maybe_url
 
-    # 3) download & tampilkan
+    # 4) download & tampilkan
     path = download_result(result_link, task_id)
     yield f"Selesai ✅ (task {task_id})", path
 
@@ -141,10 +204,10 @@ def runtime_info(user_api_key):
     key = (user_api_key or os.getenv("KIE_API_KEY") or "").strip()
     src = "BYOK (user input)" if user_api_key else ("ENV KIE_API_KEY" if os.getenv("KIE_API_KEY") else "NONE")
     masked = (key[:4] + "..." + key[-4:]) if key else "(none)"
-    return f"Key source: {src}\nKey hint: {masked}\nBase URL: {BASE}"
+    return f"Key source: {src}\nKey hint: {masked}\nBase: {BASE}"
 
-with gr.Blocks(title="Kie AI — Sora 2 Text-to-Video") as demo:
-    gr.Markdown("## Kie AI — Sora‑2 Text‑to‑Video (Polling)\n- Gunakan API key Kie AI (bukan OpenAI).")
+with gr.Blocks(title="Kie AI — Sora‑2 Text‑to‑Video") as demo:
+    gr.Markdown("## Kie AI — Sora‑2 Text‑to‑Video (Polling + Auto-probe status URL)")
     api = gr.Textbox(label="KIE API Key (opsional, BYOK)", type="password", placeholder="kie-...")
     prompt = gr.Textbox(
         label="Prompt",
@@ -161,7 +224,7 @@ with gr.Blocks(title="Kie AI — Sora 2 Text-to-Video") as demo:
             value="portrait",
             label="Aspect ratio"
         )
-        duration = gr.Textbox(value="", label="Duration (detik, opsional)", placeholder="contoh: 12 (kosongkan jika ragu)")
+        duration = gr.Textbox(value="", label="Duration (detik, opsional)", placeholder="mis. 12 (kosongkan jika ragu)")
     run = gr.Button("Generate")
     status = gr.Textbox(label="Status")
     video = gr.Video(label="Hasil video")