Update app.py

app.py

@@ -1,69 +1,66 @@
+import os
 import sys
 import time
+import pathlib
 from openai import OpenAI
 
-openai = OpenAI()
+# Ambil API key dari environment (set di Space Secrets atau lokal)
+API_KEY = os.getenv("OPENAI_API_KEY")
+if not API_KEY:
+    print("ERROR: Set environment variable OPENAI_API_KEY terlebih dahulu.")
+    sys.exit(1)
 
-# --------------------------------------------------------------
-# Create a new video
-# --------------------------------------------------------------
+client = OpenAI(api_key=API_KEY)
 
-# Takes about 2 minutes
-video = openai.videos.create(
-    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="4",
+MODEL = "sora-2"       # atau "gpt-video-1" tergantung akses akunmu
+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."
 )
-
-print("Video generation started:", video)
-
-# --------------------------------------------------------------
-# List all videos
-# --------------------------------------------------------------
-
-time.sleep(3)
-videos_list = openai.videos.list().data
-
-# --------------------------------------------------------------
-# Check the last's video status
-# --------------------------------------------------------------
-
-video = openai.videos.retrieve(videos_list[0].id)
-print("Video status:", video.status)
-
-# --------------------------------------------------------------
-# Download video
-# --------------------------------------------------------------
-
-progress = getattr(video, "progress", 0)
-bar_length = 30
-
-while video.status in ("in_progress", "queued"):
-    video = openai.videos.retrieve(video.id)
-    progress = getattr(video, "progress", 0)
-
-    filled_length = int((progress / 100) * bar_length)
-    bar = "=" * filled_length + "-" * (bar_length - filled_length)
-    status_text = "Queued" if video.status == "queued" else "Processing"
-
-    sys.stdout.write(f"\r{status_text}: [{bar}] {progress:.1f}%")
-    sys.stdout.flush()
-    time.sleep(2)
-
-sys.stdout.write("\n")
-
-if video.status == "failed":
-    message = getattr(
-        getattr(video, "error", None), "message", "Video generation failed"
+SIZE = "720x1280"      # 9:16
+SECONDS = 10           # durasi 10 detik
+
+outdir = pathlib.Path("output")
+outdir.mkdir(parents=True, exist_ok=True)
+
+def as_percent(v):
+    try:
+        x = float(v or 0)
+        return x * 100 if 0 <= x <= 1 else x
+    except Exception:
+        return 0.0
+
+# 1) Create job
+try:
+    job = client.videos.create(
+        model=MODEL,
+        prompt=PROMPT,
+        size=SIZE,
+        seconds=SECONDS,
     )
-    print(message)
+    vid_id = job.id
+    print(f"Video job created: {vid_id} (status={job.status})")
+except Exception as e:
+    s = str(e)
+    if "billing_hard_limit_reached" in s:
+        print("Gagal membuat video: limit/saldo akun OpenAI habis. Tambahkan kredit/naikkan hard limit atau gunakan API key lain.")
+    elif "insufficient_quota" in s:
+        print("Gagal: insufficient_quota. Cek Billing/kuota akun kamu.")
+    elif "invalid_api_key" in s:
+        print("Gagal: API key tidak valid.")
+    else:
+        print("Gagal membuat video:", s)
     sys.exit(1)
 
-print("Video generation completed:", video)
-print("Downloading video content...")
-
-content = openai.videos.download_content(video.id, variant="video")
-content.write_to_file(f"output/{video.id}.mp4")
-
-print(f"Wrote output/{video.id}.mp4")
+# 2) Poll sampai selesai
+bar_len = 30
+while True:
+    job = client.videos.retrieve(vid_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.capitalize():<11} [{bar}] {progress:5.1f}%")