Update app.py

app.py

@@ -25,7 +25,7 @@ MODEL_STRATEGY = {
     "gpt": [{"provider": "openrouter", "model": "openai/gpt-4o-mini"}],
     "gemini": [{"provider": "openrouter", "model": "google/gemini-2.0-flash"}]
 }
-TIMEOUT = 300
+TIMEOUT = 600
 QUEUE = asyncio.Queue()
 
 # Load Whisper Base
@@ -38,45 +38,54 @@ def remove_file(path: str):
 
 def clean_json_output(text: str):
     """
-    Membersihkan output AI agar menjadi JSON murni yang valid untuk Clipper.
-    Menangani error 'Extra data' dengan memotong teks setelah objek JSON pertama.
+    Membersihkan output AI dan memastikan formatnya adalah LIST [ {...} ].
+    Mencegah error 'string indices must be integers' di Clipper.
     """
-    if not text: return "{}"
+    if not text: return "[]"
     
-    # 1. Hapus tag <think> (khusus DeepSeek)
+    # 1. Hapus tag <think>
     text = re.sub(r'<think>.*?</think>', '', text, flags=re.DOTALL)
     
     # 2. Hapus markdown code blocks
     text = re.sub(r'```json\s*', '', text)
     text = re.sub(r'```\s*', '', text)
     
-    # 3. Cari kurung kurawal pertama '{' dan rfind '}'
-    start_idx = text.find('{')
-    end_idx = text.rfind('}')
+    # 3. Ekstrak pola JSON (mencari yang diawali [ atau {)
+    match = re.search(r'(\[.*\]|\{.*\})', text, re.DOTALL)
+    if not match: return "[]"
     
-    if start_idx != -1 and end_idx != -1 and end_idx > start_idx:
-        candidate = text[start_idx:end_idx+1].strip()
-        try:
-            # Coba validasi JSON
-            json.loads(candidate)
-            return candidate
-        except json.JSONDecodeError as e:
-            # Jika ada "Extra data", potong string tepat di posisi error
-            if "Extra data" in str(e):
-                return candidate[:e.pos].strip()
-            return candidate
+    candidate = match.group(1).strip()
+    
+    try:
+        data = json.loads(candidate)
+        # SINKRONISASI: Jika AI mengirim satu objek {}, bungkus jadi daftar [{}]
+        # Ini adalah kunci untuk memperbaiki error 'TypeError' Anda
+        if isinstance(data, dict):
+            return json.dumps([data])
+        
+        # Jika data sudah dalam bentuk list [], kembalikan apa adanya
+        if isinstance(data, list):
+            return json.dumps(data)
             
-    return text.strip()
+        return "[]"
+    except json.JSONDecodeError as e:
+        # Jika ada 'Extra data', ambil bagian yang valid saja
+        valid_match = re.search(r'(\[.*?\]|\{.*?\})', candidate, re.DOTALL)
+        if valid_match:
+            return clean_json_output(valid_match.group(1))
+        return "[]"
 
 # --- PROVIDERS ---
 async def call_ollama(messages, model):
+    print(f"DEBUG: Calling Ollama ({model})...")
     try:
-        async with httpx.AsyncClient(timeout=120) as client:
+        async with httpx.AsyncClient(timeout=600.0) as client:
             res = await client.post("http://127.0.0.1:11434/api/chat", 
                                     json={"model": model, "messages": messages, "stream": False})
             if res.status_code == 200:
                 return res.json()["message"]["content"]
-    except: pass
+    except Exception as e: 
+        print(f"DEBUG: Ollama Error: {str(e)}")
     return None
 
 async def call_openrouter(messages, model):
@@ -102,7 +111,7 @@ async def smart_router(messages, alias, is_json_request=False):
             if is_json_request:
                 return clean_json_output(raw_res)
             return raw_res
-    return "{}" if is_json_request else "Maaf, AI gagal merespon. Cek koneksi Ollama atau API Key."
+    return "[]" if is_json_request else "Gagal merespon."
 
 async def worker():
     while True:
@@ -122,167 +131,9 @@ async def lifespan(app: FastAPI):
 app = FastAPI(title="AI Gateway", lifespan=lifespan)
 app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"], allow_credentials=True)
 
-# --- STARTUP UI PLAYGROUND ---
-@app.get("/", response_class=HTMLResponse)
+@app.get("/")
 async def home():
-    return """
-    <!DOCTYPE html>
-    <html lang="id">
-    <head>
-        <meta charset="UTF-8">
-        <meta name="viewport" content="width=device-width, initial-scale=1.0">
-        <title>Imbecome AI Infrastructure</title>
-        <script src="https://cdn.tailwindcss.com"></script>
-        <link href="https://fonts.googleapis.com/css2?family=Plus+Jakarta+Sans:wght@400;600;800&display=swap" rel="stylesheet">
-        <style>
-            body { background-color: #020617; color: #f8fafc; font-family: 'Plus Jakarta Sans', sans-serif; background-image: radial-gradient(circle at 50% 0%, #1e1b4b 0%, #020617 70%); }
-            .glass { background: rgba(30, 41, 59, 0.5); backdrop-filter: blur(12px); border: 1px solid rgba(255, 255, 255, 0.1); }
-            .neon-border:focus { border-color: #38bdf8; box-shadow: 0 0 15px rgba(56, 189, 248, 0.3); }
-            .btn-gradient { background: linear-gradient(135deg, #38bdf8 0%, #818cf8 100%); transition: all 0.2s; }
-            .btn-gradient:hover { transform: translateY(-2px); filter: brightness(1.1); }
-            .status-dot { width: 8px; height: 8px; background: #4ade80; border-radius: 50%; box-shadow: 0 0 10px #4ade80; }
-        </style>
-    </head>
-    <body class="min-h-screen p-6">
-        <div class="max-w-6xl mx-auto">
-            <nav class="flex justify-between items-center mb-12 glass p-4 rounded-2xl">
-                <div class="flex items-center gap-2">
-                    <div class="bg-sky-500 w-8 h-8 rounded-lg flex items-center justify-center font-bold">I</div>
-                    <span class="text-xl font-extrabold tracking-tight text-white">IMBECOME <span class="text-sky-400">AI</span></span>
-                </div>
-                <div class="flex items-center gap-4 text-sm text-slate-400">
-                    <div class="status-dot"></div> Gateway Online
-                </div>
-            </nav>
-
-            <div class="grid grid-cols-1 lg:grid-cols-3 gap-8">
-                <div class="lg:col-span-2 glass rounded-3xl p-8 flex flex-col">
-                    <div class="flex justify-between items-center mb-6">
-                        <h3 class="text-xl font-bold flex items-center gap-2 text-white">Smart AI Chat</h3>
-                        <select id="chatModel" class="bg-slate-900 border border-slate-700 text-sm rounded-xl px-3 py-1 outline-none neon-border text-white">
-                            <option value="deepseek">DeepSeek R1 (7b)</option>
-                            <option value="qwen">Qwen 2.5 (7b)</option>
-                            <option value="gpt">GPT-4o Mini</option>
-                            <option value="gemini">Gemini 2.0</option>
-                        </select>
-                    </div>
-                    <div id="chatBox" class="bg-slate-900/50 border border-slate-800 rounded-2xl p-4 mb-4 overflow-y-auto h-[350px] text-sm space-y-4 text-white"></div>
-                    <form onsubmit="testChat(event)" class="relative">
-                        <input id="chatInput" type="text" placeholder="Tanya sesuatu..." class="w-full bg-slate-900 border border-slate-700 rounded-2xl px-5 py-4 pr-16 outline-none neon-border text-white">
-                        <button type="submit" id="btnChat" class="absolute right-2 top-2 bottom-2 bg-sky-500 px-4 rounded-xl font-bold btn-gradient text-white">Kirim</button>
-                    </form>
-                </div>
-
-                <div class="space-y-8">
-                    <div class="glass rounded-3xl p-6">
-                        <h3 class="text-lg font-bold mb-4 text-white">Voice Engine (TTS)</h3>
-                        <textarea id="ttsInput" rows="2" class="w-full bg-slate-900 border border-slate-700 rounded-xl p-3 text-sm outline-none mb-3 text-white" placeholder="Teks ke suara..."></textarea>
-                        <button onclick="testTTS()" id="btnTTS" class="w-full bg-emerald-500 py-3 rounded-xl font-bold btn-gradient text-white">Generate Suara</button>
-                        <div id="ttsPlayer" class="hidden mt-4">
-                            <audio id="audioElement" controls class="w-full h-8"></audio>
-                        </div>
-                    </div>
-
-                    <div class="glass rounded-3xl p-6">
-                        <h3 class="text-lg font-bold mb-4 text-white">Whisper AI (STT)</h3>
-                        <input type="file" id="whisperFile" class="text-xs text-slate-400 mb-4 w-full">
-                        <button onclick="testWhisper()" id="btnWhisper" class="w-full bg-purple-600 py-3 rounded-xl font-bold btn-gradient text-white">Transkripsi (Base)</button>
-                        <div id="whisperResult" class="hidden mt-4 p-3 bg-slate-900 rounded-xl border border-slate-800 text-xs text-slate-300 max-h-32 overflow-y-auto"></div>
-                    </div>
-                </div>
-            </div>
-        </div>
-
-        <script>
-            async function testChat(e) {
-                if(e) e.preventDefault();
-                const input = document.getElementById('chatInput');
-                const box = document.getElementById('chatBox');
-                const btn = document.getElementById('btnChat');
-                if(!input.value || btn.disabled) return;
-                const userMsg = input.value;
-                input.value = '';
-                btn.disabled = true;
-                btn.innerText = '...';
-                
-                const userDiv = document.createElement('div');
-                userDiv.className = 'ml-auto bg-sky-500/20 p-3 rounded-2xl max-w-[80%] text-right border border-sky-500/30 text-white';
-                userDiv.innerText = userMsg;
-                box.appendChild(userDiv);
-                box.scrollTop = box.scrollHeight;
-
-                const botDiv = document.createElement('div');
-                botDiv.className = 'mr-auto bg-slate-800 p-3 rounded-2xl max-w-[80%] border border-slate-700 text-white animate-pulse';
-                botDiv.innerText = 'Sedang berpikir...';
-                box.appendChild(botDiv);
-                box.scrollTop = box.scrollHeight;
-
-                try {
-                    const res = await fetch('/v1/chat/completions', {
-                        method: 'POST',
-                        headers: {'Content-Type': 'application/json'},
-                        body: JSON.stringify({ 
-                            model: document.getElementById('chatModel').value, 
-                            messages: [{role: 'user', content: userMsg}] 
-                        })
-                    });
-                    if (!res.ok) throw new Error('API Error ' + res.status);
-                    const data = await res.json();
-                    botDiv.classList.remove('animate-pulse');
-                    botDiv.innerText = data.choices[0].message.content;
-                } catch (err) {
-                    botDiv.classList.remove('animate-pulse');
-                    botDiv.className = 'mr-auto bg-red-900/20 p-3 rounded-2xl max-w-[80%] border border-red-500/30 text-red-200';
-                    botDiv.innerText = 'Kesalahan: ' + err.message;
-                } finally {
-                    btn.disabled = false;
-                    btn.innerText = 'Kirim';
-                    box.scrollTop = box.scrollHeight;
-                }
-            }
-
-            async function testTTS() {
-                const res = await fetch('/v1/audio/speech', {
-                    method: 'POST',
-                    headers: {'Content-Type': 'application/json'},
-                    body: JSON.stringify({ input: document.getElementById('ttsInput').value, voice: 'id' })
-                });
-                const blob = await res.blob();
-                document.getElementById('audioElement').src = URL.createObjectURL(blob);
-                document.getElementById('ttsPlayer').classList.remove('hidden');
-            }
-
-            async function testWhisper() {
-                const file = document.getElementById('whisperFile').files[0];
-                if(!file) return;
-                document.getElementById('whisperResult').classList.remove('hidden');
-                document.getElementById('whisperResult').innerText = "Sedang memproses...";
-                const fd = new FormData();
-                fd.append('file', file);
-                const res = await fetch('/v1/audio/transcriptions', { method: 'POST', body: fd });
-                const data = await res.json();
-                document.getElementById('whisperResult').innerText = data.text || "Gagal transkripsi.";
-            }
-        </script>
-    </body>
-    </html>
-    """
-
-# --- API ENDPOINTS ---
-
-@app.get("/v1/models")
-async def list_models():
-    return {
-        "object": "list",
-        "data": [
-            {"id": "deepseek", "object": "model", "owned_by": "ollama"},
-            {"id": "qwen", "object": "model", "owned_by": "ollama"},
-            {"id": "gpt", "object": "model", "owned_by": "openrouter"},
-            {"id": "gemini", "object": "model", "owned_by": "openrouter"},
-            {"id": "whisper-1", "object": "model", "owned_by": "openai-style"},
-            {"id": "tts-1", "object": "model", "owned_by": "openai-style"}   
-        ]
-    }
+    return HTMLResponse("<h1>AI Gateway Active - Dual Core Optimized</h1>")
 
 @app.post("/v1/chat/completions")
 async def chat(request: Request):
@@ -294,10 +145,10 @@ async def chat(request: Request):
     is_json = "json" in prompt_str or "highlights" in prompt_str
 
     if is_json:
-        # Tambahkan instruksi sistem yang SANGAT KETAT untuk Qwen/DeepSeek
-        messages.append({
+        # Perintah sistem yang sangat spesifik agar AI selalu memberikan LIST
+        messages.insert(0, {
             "role": "system", 
-            "content": "You are a JSON generator. Respond ONLY with a valid JSON object. Do not include any text before or after the JSON. Do not include thinking tags. If you cannot fulfill the request, return {}."
+            "content": "You are a JSON generator. Respond ONLY with a LIST of objects inside square brackets: [{\"title\":...}]. No explanation."
         })
 
     alias = "deepseek"
@@ -323,7 +174,6 @@ async def transcribe(background_tasks: BackgroundTasks, file: UploadFile = File(
     with open(tmp_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
     try:
-        # Optimasi VAD filter untuk suara pelan (min_speech_duration_ms=500)
         segments, _ = whisper_model.transcribe(
             tmp_path, 
             beam_size=5, 
@@ -333,8 +183,6 @@ async def transcribe(background_tasks: BackgroundTasks, file: UploadFile = File(
         )
         text = " ".join([s.text for s in segments])
         background_tasks.add_task(remove_file, tmp_path)
-        
-        # Placeholder jika hasil tetap kosong agar Clipper tidak melempar Exception
         final_text = text.strip() if text.strip() else "[Suara tidak terdeteksi]"
         return {"text": final_text}
     except Exception as e: