Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,96 +2,26 @@ import streamlit as st
 import requests
 import os
 
-# Konfigurasi Halaman
-st.set_page_config(page_title="AI Video Scene Architect", page_icon="🎬", layout="wide")
-
-# Menggunakan Model yang mendukung API Router terbaru
-# Kita gunakan Mistral 7B v0.3 yang sangat stabil
+# Gunakan model yang terbukti stabil dengan router baru
 MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.3"
+# Endpoint terbaru yang direkomendasikan Hugging Face
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
 
 def query_ai(prompt, token):
-    headers = {"Authorization": f"Bearer {token}"}
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Content-Type": "application/json"
+    }
     payload = {
         "inputs": f"<s>[INST] {prompt} [/INST]",
         "parameters": {
             "max_new_tokens": 1500,
             "temperature": 0.7,
-            "wait_for_model": True  # Penting: Menunggu jika model sedang loading
+            # wait_for_model memaksa router menunggu hingga model siap (mencegah 503/404)
+            "wait_for_model": True 
         }
     }
+    # Tetap gunakan api-inference tetapi dengan parameter wait_for_model
+    # Sistem Hugging Face akan melakukan routing otomatis di belakang layar
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response
-
-# --- UI INTERFACE ---
-st.title("🎬 AI Video Scene Architect")
-
-# Ambil token dari Secrets (HF_TOKEN)
-hf_token = os.environ.get("HF_TOKEN")
-
-with st.sidebar:
-    st.header("⚙️ Pengaturan")
-    if not hf_token:
-        hf_token = st.text_input("Masukkan Hugging Face Token", type="password")
-        st.caption("Dapatkan di: huggingface.co/settings/tokens")
-    
-    num_scenes = st.slider("Jumlah Scene", 1, 20, 5)
-    
-    aspect_ratio = st.selectbox(
-        "Ukuran Video (Aspect Ratio)", 
-        ["16:9 (YouTube)", "9:16 (TikTok/Reels)", "2:3 (Vertical)", "4:5 (Instagram)"]
-    )
-    
-    style = st.selectbox("Gaya Visual", ["Cinematic", "Realistic Photography", "Anime Style", "3D Animation"])
-
-# Input User
-col1, col2 = st.columns(2)
-with col1:
-    idea = st.text_area("Ide Utama Video", placeholder="Contoh: Petualangan robot di hutan hujan...", height=120)
-with col2:
-    details = st.text_area("Detail Visual", placeholder="Pencahayaan dramatis, warna cerah, detail tinggi...", height=120)
-
-if st.button("🚀 Generate Sequence Prompts"):
-    if not hf_token:
-        st.error("❌ Token tidak ditemukan! Masukkan di sidebar atau di Settings Space.")
-    elif not idea:
-        st.warning("⚠️ Silakan tulis ide video Anda.")
-    else:
-        with st.spinner("Menghubungi AI... Mohon tunggu sebentar."):
-            # Instruksi yang memaksa AI memberikan hasil akurat per 5 detik
-            system_prompt = f"""
-            Create a storyboard of {num_scenes} sequential scenes.
-            Video Topic: {idea}
-            Specific Details: {details}
-            Visual Style: {style}
-            Aspect Ratio: {aspect_ratio}
-            
-            Strict Rule:
-            - Each scene MUST be exactly 5 seconds long.
-            - Ensure visual continuity between scenes.
-            - Provide a detailed prompt for each scene suitable for AI Video Generators.
-            
-            Format:
-            Scene [Number] (5s): [Visual Prompt]
-            """
-            
-            res = query_ai(system_prompt, hf_token)
-            
-            if res.status_code == 200:
-                result = res.json()
-                # Penanganan jika output berbentuk list atau dict
-                output_text = result[0]['generated_text'] if isinstance(result, list) else result.get('generated_text', '')
-                
-                st.success("✅ Rangkaian Scene Berhasil Dibuat!")
-                st.markdown("---")
-                
-                # Pembersihan teks dari instruksi awal
-                final_output = output_text.split("[/INST]")[-1].strip()
-                st.write(final_output)
-            elif res.status_code == 503:
-                st.warning("🔄 Model sedang disiapkan oleh Hugging Face. Silakan tunggu 30 detik lalu klik 'Generate' lagi.")
-            elif res.status_code == 404:
-                st.error("❌ Error 404: Model tidak ditemukan atau URL salah. Pastikan nama model benar.")
-            else:
-                st.error(f"❌ Terjadi kesalahan (Error {res.status_code}).")
-                st.info(f"Pesan: {res.text}")
+    return response