Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,12 +1,3 @@
-# -*- coding: utf-8 -*-
-"""Untitled0.ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1XzCraWM_cfHaZvcAoEw56ClMXCmJG3TN
-"""
-
 import streamlit as st
 import requests
 import os
@@ -14,79 +5,93 @@ import os
 # Konfigurasi Halaman
 st.set_page_config(page_title="AI Video Scene Architect", page_icon="🎬", layout="wide")
 
-# Model yang lebih stabil (V0.3 lebih baru dan jarang error 410)
-MODEL_URL = "https://router.huggingface.co/models/Mistralai/Mistral-7B-Instruct-v0.2"
+# Menggunakan Model yang mendukung API Router terbaru
+# Kita gunakan Mistral 7B v0.3 yang sangat stabil
+MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.3"
+API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
 
 def query_ai(prompt, token):
     headers = {"Authorization": f"Bearer {token}"}
     payload = {
         "inputs": f"<s>[INST] {prompt} [/INST]",
-        "parameters": {"max_new_tokens": 2000, "temperature": 0.7}
+        "parameters": {
+            "max_new_tokens": 1500,
+            "temperature": 0.7,
+            "wait_for_model": True  # Penting: Menunggu jika model sedang loading
+        }
     }
-    response = requests.post(MODEL_URL, headers=headers, json=payload)
+    response = requests.post(API_URL, headers=headers, json=payload)
     return response
 
 # --- UI INTERFACE ---
 st.title("🎬 AI Video Scene Architect")
 
-# Ambil token dari Secrets
+# 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)
     
-    # Fitur Baru: Aspect Ratio
     aspect_ratio = st.selectbox(
-        "Aspect Ratio (Ukuran Video)", 
-        ["16:9 (Landscape/YouTube)", "9:16 (Portrait/TikTok)", "2:3 (Vertical)", "4:5 (Instagram)"]
+        "Ukuran Video (Aspect Ratio)", 
+        ["16:9 (YouTube)", "9:16 (TikTok/Reels)", "2:3 (Vertical)", "4:5 (Instagram)"]
     )
     
-    style = st.selectbox("Gaya Visual", ["Cinematic", "Realistic", "Anime", "3D Animation"])
+    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: Pemuda di rumah makan nasi kandar...", height=120)
+    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="Siang hari, terang, cuaca cerah...", height=120)
+    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 belum diisi! Masukkan di sidebar atau Settings > Secrets.")
+        st.error("❌ Token tidak ditemukan! Masukkan di sidebar atau di Settings Space.")
     elif not idea:
-        st.warning("⚠️ Masukkan ide video dulu.")
+        st.warning("⚠️ Silakan tulis ide video Anda.")
     else:
-        with st.spinner("Sedang merancang rangkaian scene..."):
-            # Prompt Engineering yang lebih ketat
-            prompt_instruction = f"""
-            Task: Create {num_scenes} sequential video prompts.
-            Topic: {idea}
-            Details: {details}
+        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}
-            Target Aspect Ratio: {aspect_ratio}
-            Duration per scene: 5 seconds.
+            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 output as:
-            Scene [Number] ([Duration]): [Detailed visual prompt including style and aspect ratio]
+            Format:
+            Scene [Number] (5s): [Visual Prompt]
             """
             
-            res = query_ai(prompt_instruction, hf_token)
+            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', '')
                 
-                # Menampilkan hasil
                 st.success("✅ Rangkaian Scene Berhasil Dibuat!")
                 st.markdown("---")
                 
-                # Membersihkan teks dari instruksi awal jika terbawa
-                clean_text = output_text.split("[/INST]")[-1]
-                st.write(clean_text)
+                # 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"❌ Error {res.status_code}: {res.text}")
-                st.info("Tips: Jika error 401, token salah. Jika 503, tunggu 1 menit lalu klik lagi.")
+                st.error(f"❌ Terjadi kesalahan (Error {res.status_code}).")
+                st.info(f"Pesan: {res.text}")