Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,26 +2,51 @@ import streamlit as st
 import requests
 import os
 
-# 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}",
-        "Content-Type": "application/json"
-    }
-    payload = {
-        "inputs": f"<s>[INST] {prompt} [/INST]",
-        "parameters": {
-            "max_new_tokens": 1500,
-            "temperature": 0.7,
-            # 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
+# 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)
+                
+                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}")