Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,52 +1,67 @@
 import streamlit as st
-import requests
+from huggingface_hub import InferenceClient
 import os
 
-# Konfigurasi halaman diletakkan paling atas
-st.set_page_config(page_title="Video Prompt Gen", layout="wide")
-
-try:
-    # Model ID & URL
-    MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.3"
-    API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
-
-    st.title("🎬 Video Scene Prompt Generator")
-
-    # Ambil token
-    hf_token = os.environ.get("HF_TOKEN")
-
-    with st.sidebar:
-        st.header("⚙️ Settings")
-        if not hf_token:
-            hf_token = st.text_input("HF Token", type="password")
-        
-        num_scenes = st.slider("Jumlah Scene", 1, 20, 5)
-        aspect_ratio = st.selectbox("Ratio", ["16:9", "9:16", "2:3", "4:5"])
-        style = st.selectbox("Style", ["Cinematic", "Anime", "3D", "Realistic"])
-
-    col1, col2 = st.columns(2)
-    with col1:
-        idea = st.text_area("Ide Video", "Seorang pemuda makan nasi kandar")
-    with col2:
-        details = st.text_area("Detail", "Siang hari, cerah")
-
-    if st.button("🚀 Generate"):
-        if not hf_token:
-            st.error("Masukkan Token di Sidebar/Settings!")
-        else:
-            with st.spinner("AI sedang bekerja..."):
-                prompt = f"Create {num_scenes} scenes for: {idea}. Details: {details}. Style: {style}. Ratio: {aspect_ratio}. Each scene 5s."
-                headers = {"Authorization": f"Bearer {hf_token}"}
-                payload = {"inputs": f"<s>[INST] {prompt} [/INST]", "parameters": {"wait_for_model": True}}
-                
-                response = requests.post(API_URL, headers=headers, json=payload)
+# Konfigurasi Halaman
+st.set_page_config(page_title="AI Video Scene Architect", page_icon="🎬", layout="wide")
+
+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", "9:16", "2:3", "4:5"]
+    )
+    
+    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: Astronot memasak di bulan...", height=120)
+with col2:
+    details = st.text_area("Detail Visual", placeholder="Pencahayaan dramatis, warna cerah...", height=120)
+
+if st.button("🚀 Generate Sequence Prompts"):
+    if not hf_token:
+        st.error("❌ Token tidak ditemukan! Isi di sidebar atau Settings > Secrets.")
+    elif not idea:
+        st.warning("⚠️ Silakan tulis ide video Anda.")
+    else:
+        with st.spinner("Menghubungi AI melalui Router Baru..."):
+            try:
+                # Inisialisasi Client (Otomatis menangani routing & error 410)
+                client = InferenceClient(
+                    model="mistralai/Mistral-7B-Instruct-v0.3",
+                    token=hf_token
+                )
+
+                # Format Prompt
+                system_instruction = f"Create {num_scenes} sequential video prompts for: {idea}. Details: {details}. Style: {style}. Ratio: {aspect_ratio}. Duration: 5s per scene. Format: Scene [X] (5s): [Description]"
                 
-                if response.status_code == 200:
-                    res_json = response.json()
-                    st.success("Selesai!")
-                    st.write(res_json[0]['generated_text'].split("[/INST]")[-1])
-                else:
-                    st.error(f"Error {response.status_code}: {response.text}")
-
-except Exception as e:
-    st.error(f"Aplikasi Error saat Start: {e}")
+                # Memanggil AI
+                response = ""
+                for message in client.chat_completion(
+                    messages=[{"role": "user", "content": system_instruction}],
+                    max_tokens=2000,
+                    stream=True,
+                ):
+                    response += message.choices[0].delta.content
+
+                st.success("✅ Rangkaian Scene Berhasil Dibuat!")
+                st.markdown("---")
+                st.write(response)
+
+            except Exception as e:
+                st.error(f"❌ Terjadi kesalahan: {e}")
+                st.info("Jika error 503, tunggu 1 menit lalu klik Generate lagi karena model sedang 'warming up'.")