Update downloader.py

downloader.py

@@ -1,26 +1,11 @@
-import os, sys, time, pathlib
+import os
+import time
+import pathlib
+import shutil
+import gradio as gr
 from openai import OpenAI, BadRequestError
 
-API_KEY = os.getenv("OPENAI_API_KEY")
-if not API_KEY:
-    print("ERROR: set OPENAI_API_KEY dulu.")
-    sys.exit(1)
-
-client = OpenAI(api_key=API_KEY)
-
-MODEL = "sora-2"
-PROMPT = (
-    "YouTuber, black t-shirt, in a professional studio setup, sitting at his desk "
-    "talking directly into a Shure SM7B microphone on a boom arm. He says with a smirk: "
-    "'Wait... did an AI just generate me to teach you about AI? That's pretty meta...' "
-    "Dark background with a soft blue LED light glow. Professional lighting setup with key light and rim light. "
-    "Shot on Sony FX6, cinematic depth of field, crisp focus, vertical format 9:16."
-)
-SIZE = "720x1280"
-SECONDS = "12"  # wajib string dan hanya boleh "4" | "8" | "12"
-
-outdir = pathlib.Path("output")
-outdir.mkdir(parents=True, exist_ok=True)
+ALLOWED_SECONDS = {"4", "8", "12"}
 
 def as_percent(v):
     try:
@@ -29,39 +14,116 @@ def as_percent(v):
     except Exception:
         return 0.0
 
-try:
-    job = client.videos.create(model=MODEL, prompt=PROMPT, size=SIZE, seconds=SECONDS)
-    print(f"Video job created: {job.id} status={job.status}")
-except BadRequestError as e:
-    s = str(e)
-    if "billing_hard_limit_reached" in s or "insufficient_quota" in s:
-        print("Kredit/limit akun habis. Tambah limit di Billing atau pakai API key lain.")
-    else:
-        print("Gagal membuat video:", s)
-    sys.exit(1)
-
-# polling
-bar_len = 30
-while True:
-    job = client.videos.retrieve(job.id)
-    status = getattr(job, "status", "unknown")
-    progress = as_percent(getattr(job, "progress", 0))
-    filled = int((progress / 100) * bar_len)
-    bar = "=" * filled + "-" * (bar_len - filled)
-    sys.stdout.write(f"\r{status:<11} [{bar}] {progress:5.1f}%")
-    sys.stdout.flush()
-    if status in {"succeeded", "failed", "canceled"}:
-        break
-    time.sleep(2)
-print()
-
-if job.status != "succeeded":
-    msg = getattr(getattr(job, "error", None), "message", job.status)
-    print("Video generation failed:", msg)
-    sys.exit(1)
-
-print("Downloading video...")
-blob = client.videos.download_content(job.id, variant="video")
-outfile = outdir / f"{job.id}.mp4"
-blob.write_to_file(str(outfile))
-print("Saved:", outfile)
+def generate_video(prompt, seconds, size, model, user_api_key, trim_to_10s):
+    # Ambil API key: dari input user atau dari env
+    api_key = (user_api_key or os.getenv("OPENAI_API_KEY") or "").strip()
+    if not api_key:
+        raise gr.Error("Masukkan OpenAI API key atau set OPENAI_API_KEY di Secrets.")
+
+    # seconds wajib STRING dan salah satu dari 4/8/12
+    seconds = str(seconds).strip()
+    if seconds not in ALLOWED_SECONDS:
+        raise gr.Error("Durasi 'seconds' harus salah satu dari: 4, 8, atau 12 detik.")
+
+    client = OpenAI(api_key=api_key)
+
+    # 1) Create job
+    try:
+        job = client.videos.create(
+            model=model,
+            prompt=prompt,
+            size=size,
+            seconds=seconds,   # penting: string "4" | "8" | "12"
+        )
+    except BadRequestError as e:
+        s = str(e)
+        if "billing_hard_limit_reached" in s or "insufficient_quota" in s:
+            raise gr.Error("Gagal: limit/kredit akun OpenAI habis. Tambah limit di Billing atau gunakan API key lain.")
+        if "invalid_type" in s and "seconds" in s:
+            raise gr.Error("Gagal: parameter 'seconds' harus string ('4','8','12'), bukan angka.")
+        raise gr.Error(f"Gagal membuat video: {s}")
+
+    yield f"Job dibuat: {job.id}. Status: {job.status}", None
+
+    # 2) Polling progress
+    last_progress = -1
+    while True:
+        time.sleep(2)
+        job = client.videos.retrieve(job.id)
+        status = getattr(job, "status", "unknown")
+        progress = as_percent(getattr(job, "progress", 0))
+        if int(progress) != int(last_progress):
+            yield f"{status} - {progress:.1f}%", None
+            last_progress = progress
+
+        if status == "succeeded":
+            break
+        if status in {"failed", "canceled"}:
+            msg = getattr(getattr(job, "error", None), "message", status)
+            raise gr.Error(f"Gagal: {msg}")
+
+    # 3) Download hasil
+    tmp_dir = pathlib.Path("/tmp")
+    tmp_dir.mkdir(exist_ok=True, parents=True)
+    raw_path = tmp_dir / f"{job.id}.mp4"
+
+    blob = client.videos.download_content(job.id, variant="video")
+    blob.write_to_file(str(raw_path))
+
+    out_path = raw_path
+
+    # 4) Opsi: trim ke 10 detik (pakai ffmpeg jika tersedia)
+    if trim_to_10s:
+        ffmpeg = shutil.which("ffmpeg")
+        if ffmpeg:
+            trimmed_path = tmp_dir / f"{job.id}_10s.mp4"
+            # -t 10: batasi durasi 10 detik; -c copy: tanpa re-encode (cepat)
+            # jika format tidak mengizinkan -c copy, fallback re-encode
+            code = os.system(f'{ffmpeg} -y -i "{raw_path}" -t 10 -c copy "{trimmed_path}"')
+            if code != 0:
+                # fallback re-encode
+                os.system(f'{ffmpeg} -y -i "{raw_path}" -t 10 -vf "scale=iw:ih" -c:v libx264 -preset veryfast -crf 20 -c:a aac -b:a 128k "{trimmed_path}"')
+            if trimmed_path.exists():
+                out_path = trimmed_path
+        else:
+            yield "ffmpeg tidak tersedia, melewati proses trim. Hasil tetap durasi asli.", None
+
+    yield "Selesai ✅", str(out_path)
+
+
+with gr.Blocks(title="Sora-2 Video Generator") as demo:
+    gr.Markdown("## Sora-2 Video Generator (HF Space) — BYOK ready")
+    gr.Markdown("- Durasi wajib: 4, 8, atau 12 detik (aturan API). Jika butuh 10 detik, centang Trim to 10s setelah render.")
+    with gr.Row():
+        api = gr.Textbox(label="OpenAI API Key (opsional)", type="password", placeholder="sk-...")
+        model = gr.Textbox(value="sora-2", label="Model")
+    prompt = gr.Textbox(
+        label="Prompt",
+        lines=8,
+        value=(
+            "YouTuber, black t-shirt, in a professional studio setup, sitting at his desk "
+            "talking directly into a Shure SM7B microphone on a boom arm. He says with a smirk: "
+            "'Wait... did an AI just generate me to teach you about AI? That's pretty meta...' "
+            "Dark background with a soft blue LED light glow. Professional lighting setup with key light and rim light. "
+            "Shot on Sony FX6, cinematic depth of field, crisp focus, vertical format 9:16."
+        ),
+    )
+    with gr.Row():
+        size = gr.Dropdown(
+            ["720x1280", "1080x1920", "1280x720", "1920x1080"],
+            value="720x1280",
+            label="Resolution",
+        )
+        seconds = gr.Dropdown(["4", "8", "12"], value="12", label="Duration (sec)")
+        trim10 = gr.Checkbox(value=False, label="Trim to 10s after render")
+    run = gr.Button("Generate")
+    status = gr.Textbox(label="Status", lines=2)
+    video = gr.Video(label="Hasil (unduh dari tombol di player)")
+
+    run.click(
+        fn=generate_video,
+        inputs=[prompt, seconds, size, model, api, trim10],
+        outputs=[status, video],
+    )
+
+demo.launch()