Update app.py

app.py

@@ -1,101 +1,82 @@
-import os
-import re
-import google.generativeai as genai
 import gradio as gr
 import graphviz
+import re
+from huggingface_hub import InferenceClient
 
-# --- SETUP API KEY ---
-MY_API_KEY = "AIzaSyAY4KmX9Xf1xZ4bnme7c1mw61QWn4nNLDI"
-genai.configure(api_key=MY_API_KEY)
-
-# --- FUNGSI AUTO-DETECT MODEL (Solusi Error 404) ---
-def get_working_model():
-    # Daftar semua kemungkinan nama model dari Google
-    daftar_model = [
-        'gemini-1.5-flash',
-        'gemini-1.5-pro',
-        'gemini-1.0-pro',
-        'gemini-pro',
-        'models/gemini-1.5-flash',
-        'models/gemini-pro'
-    ]
-    
-    print("Sedang mencari model yang aktif...")
-    for nama in daftar_model:
-        try:
-            # Test koneksi ringan
-            model = genai.GenerativeModel(nama)
-            model.generate_content("test")
-            print(f"✅ BERHASIL! Menggunakan model: {nama}")
-            return model
-        except Exception as e:
-            print(f"❌ {nama} gagal, mencoba yang lain...")
-            continue
-            
-    return None
-
-# Inisialisasi Model saat aplikasi nyala
-model_aktif = get_working_model()
+# --- SETUP MODEL OPEN SOURCE (GRATIS & TANPA KUNCI) ---
+# Kita menggunakan Qwen 2.5 - 72 Miliar Parameter (Sangat Cerdas)
+# Model ini berjalan di server publik Hugging Face
+client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
 
-# --- LOGIKA INSINYUR ---
 def engineer_mode(deskripsi_user):
-    # Cek lagi jika model belum ketemu
-    global model_aktif
-    if model_aktif is None:
-        model_aktif = get_working_model()
-        if model_aktif is None:
-            return "⚠️ SEMUA MODEL GAGAL DIHUBUNGI. Cek API Key atau Server Google sedang down.", None
-
     if not deskripsi_user:
         return "⚠️ Mohon isi deskripsi mesin dulu.", None
 
-    final_prompt = f"""
-    BERTINDAKLAH SEBAGAI: Senior Industrial Automation Engineer.
-    TUGAS: Buatkan rancangan teknis detail untuk: "{deskripsi_user}"
+    # Prompt Insinyur yang sama, tapi dioptimalkan untuk Qwen
+    prompt_system = """
+    Anda adalah Senior Industrial Automation Engineer (20+ Tahun Pengalaman).
+    Tugas Anda adalah merancang sistem mesin pabrik secara presisi, logis, dan aman.
     
-    OUTPUT WAJIB (BAHASA INDONESIA):
-    1. DAFTAR KOMPONEN (Part Nyata & Merk)
-    2. CARA KERJA (Step-by-step Logika Safety)
-    3. WIRING GUIDE (Panduan Kabel Pin-to-Pin)
-    4. PROGRAM CODE (Arduino C++ / PLC)
-    5. DIAGRAM BLOK (HANYA KODE GRAPHVIZ DOT)
-       - Mulai dengan: digraph G {{
-       - Akhiri dengan: }}
+    OUTPUT WAJIB (DALAM BAHASA INDONESIA):
+    1. DAFTAR KOMPONEN (PART LIST): Sebutkan merk nyata (Omron, Siemens, dll) dan tipe spesifik.
+    2. CARA KERJA: Jelaskan logika urutan dan fitur Safety (Interlock/Emergency).
+    3. WIRING GUIDE: Panduan koneksi kabel (Pin-to-Pin).
+    4. PROGRAM CODE: Berikan kode C++ (Arduino) atau Structured Text (PLC) yang lengkap dengan komentar.
+    5. DIAGRAM BLOK (HANYA KODE GRAPHVIZ DOT):
+       - Mulai dengan: digraph G {
+       - Akhiri dengan: }
        - Gunakan: rankdir=LR; node [shape=box, style=filled, fillcolor=lightblue];
+       - Pastikan sintaks valid.
     """
-    
+
     try:
-        response = model_aktif.generate_content(final_prompt)
-        text = response.text
+        # Mengirim pesan ke Model Open Source
+        messages = [
+            {"role": "system", "content": prompt_system},
+            {"role": "user", "content": f"Rancang mesin ini: {deskripsi_user}"}
+        ]
         
-        # Ekstraksi Diagram
+        # Streaming respons (agar tidak timeout)
+        response_text = ""
+        for message in client.chat_completion(messages, max_tokens=2000, stream=True):
+            token = message.choices[0].delta.content
+            response_text += token
+            
+        # --- PROSES DIAGRAM (VISUALISASI) ---
         dot_code = ""
-        if "digraph G {" in text:
-            s = text.find("digraph G {")
-            e = text.rfind("}") + 1
-            dot_code = text[s:e]
+        # Mencari pola kode DOT di dalam teks
+        match = re.search(r'digraph G \{.*\}', response_text, re.DOTALL)
+        
+        if match:
+            dot_code = match.group(0)
         else:
-            dot_code = 'digraph G { Error -> "Diagram Gagal"; }'
+            # Fallback manual jika regex gagal
+            if "digraph G {" in response_text:
+                s = response_text.find("digraph G {")
+                e = response_text.rfind("}") + 1
+                dot_code = response_text[s:e]
+            else:
+                dot_code = 'digraph G { Error -> "Diagram Gagal Dibuat"; }'
 
         # Render Gambar
         try:
             src = graphviz.Source(dot_code)
             output_path = src.render("diagram_output", format="png")
-            return text, output_path
+            return response_text, output_path
         except:
-            return text, None
+            return response_text, None
 
     except Exception as e:
-        return f"Error saat generate: {str(e)}", None
+        return f"⚠️ Terjadi Kepadatan Server (Busy). Coba tekan tombol Rancang lagi dalam beberapa detik.\nError: {str(e)}", None
 
-# --- TAMPILAN (UI) ---
+# --- TAMPILAN APLIKASI ---
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🏭 AI Perancang Mesin (Auto-Fix Version)")
-    gr.Markdown("Versi ini otomatis mencari jalur server Google yang aktif.")
+    gr.Markdown("# 🏭 AI Insinyur (Open Source Qwen 2.5)")
+    gr.Markdown("Menggunakan model **Qwen-72B** yang gratis, tanpa API Key, dan presisi tinggi.")
     
     with gr.Row():
-        inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Jelaskan mesin yang mau dibuat...")
-        btn = gr.Button("🚀 RANCANG SEKARANG", variant="primary")
+        inp = gr.Textbox(lines=5, label="Deskripsi Mesin", placeholder="Contoh: Alat pres hidrolik otomatis...")
+        btn = gr.Button("🚀 RANCANG (GRATIS)", variant="primary")
     
     with gr.Row():
         out_txt = gr.Markdown(label="Dokumen Teknis")