Update app.py

app.py

@@ -12,10 +12,7 @@ def make_headers(user_key: str | None):
     api_key = (user_key or os.getenv("KIE_API_KEY") or "").strip()
     if not api_key:
         raise gr.Error("Masukkan KIE API key atau set KIE_API_KEY di Secrets.")
-    return {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json",
-    }
+    return {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
 
 def create_task(headers, prompt: str, aspect_ratio: str, duration: str | None):
     payload = {
@@ -23,7 +20,7 @@ def create_task(headers, prompt: str, aspect_ratio: str, duration: str | None):
         "input": {
             "prompt": prompt,
             "aspect_ratio": aspect_ratio,
-            "remove_watermark": False,  # patuh ToS
+            "remove_watermark": False,
         },
     }
     if duration:
@@ -57,23 +54,17 @@ def create_task(headers, prompt: str, aspect_ratio: str, duration: str | None):
 def infer_status_and_result(obj: dict):
     status = obj.get("status") or obj.get("state") or (obj.get("data") or {}).get("status")
     progress = obj.get("progress") or (obj.get("data") or {}).get("progress")
-    # cari link video pada beberapa kemungkinan field
-    for key in [
-        "result_url","output_url","video_url","download_url",
-    ]:
+    for key in ["result_url","output_url","video_url","download_url"]:
         if key in obj and isinstance(obj[key], str) and obj[key].startswith("http"):
             return status, progress, obj[key]
     data = obj.get("data") or {}
-    for key in [
-        "result_url","output_url","video_url","download_url",
-    ]:
+    for key in ["result_url","output_url","video_url","download_url"]:
         v = data.get(key)
         if isinstance(v, str) and v.startswith("http"):
             return status, progress, v
     return status, progress, None
 
 def probe_status_url(headers, task_id: str):
-    # Coba beberapa pola umum sampai ada yang OK
     candidates = [
         f"{BASE}/jobs/getTask?taskId={task_id}",
         f"{BASE}/jobs/get-task?taskId={task_id}",
@@ -96,7 +87,7 @@ def probe_status_url(headers, task_id: str):
                     return url
         except Exception:
             pass
-    # Beberapa layanan pakai POST getTask
+    # POST variant
     post_url = f"{BASE}/jobs/getTask"
     try:
         r = requests.post(post_url, headers=headers, json={"taskId": task_id}, timeout=30)
@@ -195,7 +186,6 @@ def runtime_info(user_api_key):
     masked = (key[:4] + "..." + key[-4:]) if key else "(none)"
     return f"Key source: {src}\nKey hint: {masked}\nBase: {BASE}"
 
-# ============ Gradio UI (Pure Gradio) ============
 with gr.Blocks(title="Kie AI — Sora‑2 Text‑to‑Video (Polling)") as demo:
     gr.Markdown("## Kie AI — Sora‑2 Text‑to‑Video\n- Gunakan API key Kie AI (bukan OpenAI).\n- remove_watermark dimatikan (False).")
     api = gr.Textbox(label="KIE API Key (opsional, BYOK)", type="password", placeholder="kie-...")
@@ -220,4 +210,6 @@ with gr.Blocks(title="Kie AI — Sora‑2 Text‑to‑Video (Polling)") as demo:
     run.click(run_job, inputs=[api, prompt, aspect, duration], outputs=[status, video])
     check.click(runtime_info, inputs=[api], outputs=[info])
 
-# Penting untuk Spaces: expose variable 'demo'
+# Tambahkan launch jika Space type = Python
+if __name__ == "__main__":
+    demo.queue().launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", "7860")))