Update app.py

app.py

@@ -1,125 +1,60 @@
 import gradio as gr
-import graphviz
-import requests
-import json
-import re
-
-# ======================================================
-# GOOGLE GEMINI: MULTI-MODEL SELECTOR
-# ======================================================
-# Kode ini akan mencoba berbagai versi model Google
-# sampai menemukan yang aktif untuk akun Anda.
-
-# 1. SETUP API KEY (Teknik Kamuflase agar aman)
-# Token asli: AIzaSyAY4KmX9Xf1xZ4bnme7c1mw61QWn4nNLDI
-k1 = "AIzaSyAY4KmX9Xf1xZ4bnme"
-k2 = "7c1mw61QWn4nNLDI"
-MY_KEY = k1 + k2
-
-def query_google_gemini(prompt_text):
-    # DAFTAR ALAMAT YANG AKAN DICOBA (Sapu Jagat)
-    # Kita urutkan dari yang paling cepat & baru
-    models = [
-        "gemini-1.5-flash",
-        "gemini-1.5-flash-latest",
-        "gemini-1.5-pro",
-        "gemini-1.5-pro-latest",
-        "gemini-1.0-pro"
-    ]
-
-    headers = {'Content-Type': 'application/json'}
-    payload = {
-        "contents": [{"parts": [{"text": prompt_text}]}]
-    }
-
-    last_error = ""
-
-    # LOOPING: Coba satu per satu
-    for model_name in models:
-        try:
-            # Coba Tembak URL Model Tersebut
-            url = f"https://generativelanguage.googleapis.com/v1beta/models/{model_name}:generateContent?key={MY_KEY}"
-            
-            response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=30)
-            
-            # Jika Sukses (Kode 200)
-            if response.status_code == 200:
-                data = response.json()
-                # Ambil teks jawaban
-                return data['candidates'][0]['content']['parts'][0]['text']
-            
-            # Jika Gagal, simpan errornya dan lanjut ke model berikutnya
-            else:
-                last_error = f"Model {model_name} Gagal: {response.text}"
-                continue # Coba model bawahnya
-
-        except Exception as e:
-            last_error = str(e)
-            continue
-
-    # Jika semua model gagal
-    return f"❌ SEMUA MODEL GOOGLE GAGAL.\nDetail Error Terakhir: {last_error}"
-
-# ======================================================
-# LOGIKA INSINYUR
-# ======================================================
-def engineer_mode(deskripsi_user):
-    if not deskripsi_user:
-        return "⚠️ Mohon isi deskripsi mesin dulu.", None
-
-    prompt_system = f"""
-    ROLE: Senior Industrial Automation Engineer.
-    TASK: Design a complete system for: "{deskripsi_user}"
+from llama_cpp import Llama
+from huggingface_hub import hf_hub_download
+
+# 1. Download Model Koding Terbaik (Qwen2.5-Coder 7B)
+# Kita pakai versi GGUF Q4_K_M agar muat di RAM gratisan dan cepat
+print("Sedang mendownload model... Mohon tunggu.")
+model_path = hf_hub_download(
+    repo_id="Qwen/Qwen2.5-Coder-7B-Instruct-GGUF",
+    filename="qwen2.5-coder-7b-instruct-q4_k_m.gguf"
+)
+
+# 2. Load Model ke Memory (CPU)
+llm = Llama(
+    model_path=model_path,
+    n_ctx=8192,      # Context window besar untuk kode panjang
+    n_threads=2,     # Menggunakan 2 core CPU
+    verbose=False
+)
+
+# 3. Fungsi Chat
+def chat_response(message, history):
+    # Format prompt agar sesuai standar Qwen
+    system_prompt = "You are an expert coding assistant. Write clean, professional code."
+    prompt_str = f"<|im_start|>system\n{system_prompt}<|im_end|>\n"
     
-    OUTPUT FORMAT (BAHASA INDONESIA):
-    1. BOM (Komponen Nyata & Tipe).
-    2. CARA KERJA (Step-by-step + Safety).
-    3. WIRING (Pin-to-Pin).
-    4. KODE PROGRAM (Arduino/PLC).
-    5. DIAGRAM BLOK (DOT CODE ONLY):
-       - Start: digraph G {{
-       - End: }}
-       - Style: rankdir=LR; node [shape=box style="filled,rounded" fillcolor="#e1f5fe"];
-    """
-
-    # Panggil Fungsi Sapu Jagat
-    hasil_teks = query_google_gemini(prompt_system)
-
-    # Cek jika hasil adalah pesan error
-    if "❌" in hasil_teks:
-        return hasil_teks, None
-
-    # --- PROSES GAMBAR DIAGRAM ---
-    dot_code = ""
-    if "digraph G {" in hasil_teks:
-        s = hasil_teks.find("digraph G {")
-        e = hasil_teks.rfind("}") + 1
-        dot_code = hasil_teks[s:e]
-    else:
-        dot_code = 'digraph G { Error -> "Diagram Tidak Tergenerate"; }'
-
-    try:
-        src = graphviz.Source(dot_code)
-        output_path = src.render("diagram_output", format="png")
-        return hasil_teks, output_path
-    except:
-        return hasil_teks, None
-
-# ======================================================
-# TAMPILAN UI
-# ======================================================
-with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🏭 AI Insinyur (Google Multi-Model)")
-    gr.Markdown("Sistem ini otomatis mencari jalur Google Gemini yang aktif (Flash/Pro/Latest).")
+    for user_input, ai_output in history:
+        prompt_str += f"<|im_start|>user\n{user_input}<|im_end|>\n<|im_start|>assistant\n{ai_output}<|im_end|>\n"
     
-    with gr.Row():
-        inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Jelaskan kebutuhan mesin Anda...")
-        btn = gr.Button("🚀 RANCANG (AUTO-CONNECT)", variant="primary")
+    prompt_str += f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
     
-    with gr.Row():
-        out_txt = gr.Markdown(label="Dokumen Teknis")
-        out_img = gr.Image(label="Diagram Sistem")
-            
-    btn.click(engineer_mode, inputs=inp, outputs=[out_txt, out_img])
-
-app.launch()
+    # Generate jawaban stream (agar teks muncul satu-satu tidak nge-lag)
+    output_stream = llm(
+        prompt_str,
+        max_tokens=2048,
+        stop=["<|im_end|>"],
+        stream=True
+    )
+    
+    partial_message = ""
+    for chunk in output_stream:
+        delta = chunk['choices'][0]['text']
+        partial_message += delta
+        yield partial_message
+
+# 4. Tampilan Web (Mobile Friendly)
+custom_css = """
+footer {visibility: hidden} 
+"""
+
+demo = gr.ChatInterface(
+    fn=chat_response, 
+    title="🤖 Qwen Android Coder",
+    description="Asisten Koding Profesional untuk Android. Copy kode & jalankan di Termux/Acode.",
+    theme=gr.themes.Soft(),
+    css=custom_css
+)
+
+if __name__ == "__main__":
+    demo.launch()