Update app.py

app.py

@@ -1,55 +1,49 @@
-# app.py (Space publik)
+import streamlit as st
+import google.generativeai as genai
 import os
-from huggingface_hub import snapshot_download
-import sys
-
-# ---- 1) Set ENV sebelum Streamlit jalan (W A J I B) ----
-# HOME yang writable (agar tidak tulis ke '/.streamlit')
-os.environ.setdefault("HOME", "/home/user")
-# Matikan XSRF/CORS + naikan limit upload
-os.environ.setdefault("STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION", "false")
-os.environ.setdefault("STREAMLIT_SERVER_ENABLE_CORS", "false")
-os.environ.setdefault("STREAMLIT_BROWSER_GATHER_USAGE_STATS", "false")
-os.environ.setdefault("STREAMLIT_SERVER_MAX_UPLOAD_SIZE", "1024")
-
-# (opsional) siapkan config.toml sebagai backup
-try:
-    cfg_dir = os.path.join(os.environ["HOME"], ".streamlit")
-    os.makedirs(cfg_dir, exist_ok=True)
-    cfg_path = os.path.join(cfg_dir, "config.toml")
-    if not os.path.exists(cfg_path):
-        with open(cfg_path, "w", encoding="utf-8") as f:
-            f.write(
-                "[server]\n"
-                "enableXsrfProtection = false\n"
-                "enableCORS = false\n"
-                "maxUploadSize = 1024\n"
-                "\n[browser]\n"
-                "gatherUsageStats = false\n"
-            )
-except Exception:
-    pass
-
-# ---- 2) Download Space privat ----
-LOCAL_DIR = snapshot_download(
-    repo_id="sharul20001/veo",        # <== pastikan benar
-    repo_type="space",
-    use_auth_token=os.getenv("HF_TOKEN"),
-)
-
-# ---- 3) Jalankan streamlit script dari Space privat ----
-streamlit_script = os.path.join(LOCAL_DIR, "app.py)
-if not os.path.exists(streamlit_script):
-    print(f"File tidak ditemukan: {streamlit_script}")
-    sys.exit(1)
-
-os.execvp(
-    "streamlit",
-    [
-        "streamlit",
-        "run",
-        streamlit_script,
-        "--server.address", "0.0.0.0",
-        "--server.port", os.getenv("PORT", "7860"),
-    ],
-)
+
+# ==============================
+# 1. Setup API Key
+# ==============================
+API_KEY = os.getenv("GOOGLE_API_KEY")  # nanti set di HuggingFace "Repository Secrets"
+if not API_KEY:
+    st.error("⚠️ GOOGLE_API_KEY belum diset di Secrets.")
+else:
+    genai.configure(api_key=API_KEY)
+
+# ==============================
+# 2. UI Streamlit
+# ==============================
+st.set_page_config(page_title="Veo‑3 Video Generator 🚀")
+st.title("🎬 Veo‑3 Video Generator")
+
+prompt = st.text_area("Masukkan deskripsi video yang ingin dibuat:", 
+                      "Seekor naga terbang di angkasa dengan gaya anime")
+
+duration = st.slider("Durasi video (detik)", 2, 10, 4)
+
+if st.button("🚀 Generate Video"):
+    if not API_KEY:
+        st.warning("Silakan set GOOGLE_API_KEY terlebih dahulu.")
+    else:
+        with st.spinner("Sedang generate video dari Veo‑3..."):
+            try:
+                # ==============================
+                # 3. Panggilan API Veo‑3
+                # ==============================
+                response = genai.generate_video(
+                    model="veo-3",
+                    prompt=prompt,
+                    duration=duration,
+                )
+
+                video_url = response.video.uri  # link URL video hasil (Veo API balikin URI ke file mp4)
+
+                # ==============================
+                # 4. Tampilkan hasil
+                # ==============================
+                st.video(video_url)
+                st.success("✅ Video berhasil dibuat.")
+
+            except Exception as e:
+                st.error(f"Terjadi error: {e}")