Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,80 +9,77 @@ Original file is located at
 
 import streamlit as st
 import requests
-import json
+import os
 
 # Konfigurasi Halaman
-st.set_page_config(page_title="AI Scene Director", page_icon="🎬", layout="wide")
+st.set_page_config(page_title="AI Video Scene Director", page_icon="🎬", layout="wide")
 
-# Masukkan Token Hugging Face Anda di sini atau via Sidebar
-API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B-Instruct"
-# Tips: Ganti API_KEY di bawah dengan token asli Anda
-headers = {"Authorization": "Bearer YOUR_HUGGINGFACE_TOKEN_HERE"}
+# Daftar Model AI (Urutan dari yang paling pintar)
+MODELS = [
+    "meta-llama/Meta-Llama-3-8B-Instruct",
+    "mistralai/Mistral-7B-Instruct-v0.2",
+    "HuggingFaceH4/zephyr-7b-beta"
+]
 
-def query_ai(prompt):
+def query_ai(prompt, model_url, token):
+    headers = {"Authorization": f"Bearer {token}"}
     payload = {
         "inputs": prompt,
-        "parameters": {"max_new_tokens": 2000, "temperature": 0.7}
+        "parameters": {"max_new_tokens": 1500, "temperature": 0.7, "return_full_text": False}
     }
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+    response = requests.post(f"https://api-inference.huggingface.co/models/{model_url}", headers=headers, json=payload)
+    return response
 
 # --- UI INTERFACE ---
 st.title("🎬 AI Video Scene Architect")
-st.markdown("Generator prompt berurutan (5 detik/scene) menggunakan Model Bahasa Open-Source.")
+
+# Ambil token dari Secrets atau Sidebar
+hf_token = os.environ.get("HF_TOKEN")
 
 with st.sidebar:
     st.header("Konfigurasi")
-    hf_token = st.text_input("Hugging Face Token", type="password", help="Dapatkan di huggingface.co/settings/tokens")
+    if not hf_token:
+        hf_token = st.text_input("Masukkan Hugging Face Token (hf_...)", type="password")
+        st.info("Tips: Simpan token di Settings > Secrets dengan nama HF_TOKEN agar otomatis terisi.")
+    
     num_scenes = st.slider("Jumlah Scene", 1, 20, 5)
-    ratio = st.selectbox("Aspect Ratio", ["16:9 (YouTube)", "9:16 (TikTok/Reels)", "1:1 (Instagram)"])
-
-    if hf_token:
-        headers["Authorization"] = f"Bearer {hf_token}"
+    style = st.selectbox("Gaya Visual", ["Cinematic", "Anime", "3D Render", "Cyberpunk", "Nature Documentary"])
 
 # Input User
 col1, col2 = st.columns(2)
 with col1:
-    idea = st.text_area("Ide Utama Video", "Seorang astronot menemukan taman bunga di Mars", height=150)
+    idea = st.text_area("Ide Utama Video", "Astronot menemukan hutan di bulan", height=100)
 with col2:
-    details = st.text_area("Detail Visual & Mood", "Cinematic, photorealistic, sunset lighting, lofi vibes", height=150)
+    details = st.text_area("Detail Visual", "Neon plants, dark blue sky, cinematic lighting", height=100)
 
 if st.button("🚀 Generate Sequence Prompts"):
     if not hf_token:
-        st.error("Silakan masukkan Hugging Face Token di sidebar terlebih dahulu!")
-    elif idea:
-        with st.spinner(f"Sedang merancang {num_scenes} scene berurutan..."):
-
-            # Engineering Prompt yang memaksa output terstruktur
-            master_prompt = f"""
-            [INST] Act as a professional Video Director.
-            Create a sequential storyboard for a video about: "{idea}".
-            Details: {details}.
-
-            Strict Requirements:
-            1. Total scenes: {num_scenes}.
-            2. Each scene duration: exactly 5 seconds.
-            3. Maintain visual continuity (characters and settings must stay consistent).
-            4. Format each scene as:
-               Scene [Number]: [5-second visual description for AI Video Generator]
-            [/INST]
-            """
-
-            result = query_ai(master_prompt)
-
-            if isinstance(result, list) and len(result) > 0:
-                output_text = result[0].get('generated_text', '')
-                # Membersihkan output dari prompt asli
-                clean_output = output_text.split("[/INST]")[-1].strip()
-
-                st.success("Rangkaian Scene Berhasil Dibuat!")
-
-                # Menampilkan hasil dalam bentuk kartu
-                scenes_raw = clean_output.split("Scene")
-                for s in scenes_raw:
-                    if s.strip():
-                        with st.container():
-                            st.info(f"🎬 Scene {s.strip()}")
-                            st.button(f"Copy Scene", key=s[:20])
-            else:
-                st.error("Gagal menghubungi AI. Pastikan Token benar atau coba lagi nanti.")
+        st.error("❌ Token tidak ditemukan! Masukkan token di sidebar atau di Settings Space.")
+    elif not idea:
+        st.warning("⚠️ Masukkan ide video terlebih dahulu.")
+    else:
+        success = False
+        with st.spinner("Menghubungi otak AI... (mencoba beberapa model jika perlu)"):
+            for model in MODELS:
+                try:
+                    res = query_ai(f"[INST] Create {num_scenes} video scenes for: {idea}. Style: {style}. Details: {details}. Each scene 5 seconds. Format: Scene X: [description] [/INST]", model, hf_token)
+                    
+                    if res.status_code == 200:
+                        result_json = res.json()
+                        # Tergantung model, format output bisa berbeda
+                        text = result_json[0]['generated_text'] if isinstance(result_json, list) else result_json.get('generated_text', '')
+                        
+                        st.success(f"✅ Berhasil menggunakan model: {model}")
+                        st.markdown("---")
+                        st.write(text)
+                        success = True
+                        break
+                    elif res.status_code == 503:
+                        st.info(f"🔄 Model {model} sedang loading (start-up). Mencoba model lain...")
+                    else:
+                        st.error(f"⚠️ Model {model} error: {res.status_code}")
+                except Exception as e:
+                    st.error(f"❌ Kesalahan teknis: {str(e)}")
+            
+            if not success:
+                st.error("💥 Semua model gagal merespon. Pastikan Token kamu valid (tipe 'Read') dan internet stabil.")