Update app.py

app.py

@@ -5,17 +5,18 @@ import gradio as gr
 import graphviz
 
 # --- SETUP API KEY ---
-# Ini API Key Anda yang sudah saya tanam langsung
+# Menggunakan API Key Anda
 MY_API_KEY = "AIzaSyAY4KmX9Xf1xZ4bnme7c1mw61QWn4nNLDI"
 
+# --- PERBAIKAN DI SINI ---
+# Kita gunakan 'gemini-pro' karena lebih stabil dan pasti ada.
 try:
     genai.configure(api_key=MY_API_KEY)
-    # Kita pakai model Flash agar cepat & gratis
-    model = genai.GenerativeModel('gemini-1.5-flash')
+    model = genai.GenerativeModel('gemini-pro') 
 except Exception as e:
     print(f"Error Setup: {e}")
 
-# --- LOGIKA INSINYUR (BRAIN) ---
+# --- LOGIKA INSINYUR ---
 def engineer_mode(deskripsi_user):
     if not deskripsi_user:
         return "⚠️ Mohon isi deskripsi mesin dulu.", None
@@ -36,26 +37,23 @@ def engineer_mode(deskripsi_user):
     """
     
     try:
+        # Generate konten
         response = model.generate_content(final_prompt)
         text = response.text
         
         # Ekstraksi Diagram
         dot_code = ""
-        match = re.search(r'digraph G \{.*\}', text, re.DOTALL)
-        if match:
-            dot_code = match.group(0)
+        # Mencari pola digraph G { ... }
+        if "digraph G {" in text:
+            s = text.find("digraph G {")
+            e = text.rfind("}") + 1
+            dot_code = text[s:e]
         else:
-            if "digraph" in text:
-                s = text.find("digraph")
-                e = text.rfind("}") + 1
-                dot_code = text[s:e]
-            else:
-                dot_code = 'digraph G { Error -> "Diagram Gagal"; }'
+            dot_code = 'digraph G { Error -> "Diagram Gagal"; }'
 
         # Render Gambar
         try:
             src = graphviz.Source(dot_code)
-            # Di HuggingFace kita render ke file sementara
             output_path = src.render("diagram_output", format="png")
             return text, output_path
         except:
@@ -66,12 +64,11 @@ def engineer_mode(deskripsi_user):
 
 # --- TAMPILAN (UI) ---
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🏭 AI Perancang Mesin (Fast Version)")
-    gr.Markdown("Masukkan ide mesin, AI akan buatkan Part List, Wiring, Program & Diagram.")
+    gr.Markdown("# 🏭 AI Perancang Mesin (Versi Stabil)")
     
     with gr.Row():
-        inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Contoh: Alat sortir otomatis...")
-        btn = gr.Button("🚀 RANCANG SEKARANG", variant="primary")
+        inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Jelaskan mesin yang mau dibuat...")
+        btn = gr.Button("🚀 RANCANG (GEMINI PRO)", variant="primary")
     
     with gr.Row():
         out_txt = gr.Markdown(label="Dokumen Teknis")