Update app.py

app.py

@@ -5,19 +5,47 @@ import gradio as gr
 import graphviz
 
 # --- SETUP API KEY ---
-# Menggunakan API Key Anda
 MY_API_KEY = "AIzaSyAY4KmX9Xf1xZ4bnme7c1mw61QWn4nNLDI"
+genai.configure(api_key=MY_API_KEY)
 
-# --- PERBAIKAN DI SINI ---
-# Kita gunakan 'gemini-pro' karena lebih stabil dan pasti ada.
-try:
-    genai.configure(api_key=MY_API_KEY)
-    model = genai.GenerativeModel('gemini-pro') 
-except Exception as e:
-    print(f"Error Setup: {e}")
+# --- 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()
 
 # --- 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
 
@@ -37,13 +65,11 @@ def engineer_mode(deskripsi_user):
     """
     
     try:
-        # Generate konten
-        response = model.generate_content(final_prompt)
+        response = model_aktif.generate_content(final_prompt)
         text = response.text
         
         # Ekstraksi Diagram
         dot_code = ""
-        # Mencari pola digraph G { ... }
         if "digraph G {" in text:
             s = text.find("digraph G {")
             e = text.rfind("}") + 1
@@ -60,15 +86,16 @@ def engineer_mode(deskripsi_user):
             return text, None
 
     except Exception as e:
-        return f"Error AI: {str(e)}", None
+        return f"Error saat generate: {str(e)}", None
 
 # --- TAMPILAN (UI) ---
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🏭 AI Perancang Mesin (Versi Stabil)")
+    gr.Markdown("# 🏭 AI Perancang Mesin (Auto-Fix Version)")
+    gr.Markdown("Versi ini otomatis mencari jalur server Google yang aktif.")
     
     with gr.Row():
         inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Jelaskan mesin yang mau dibuat...")
-        btn = gr.Button("🚀 RANCANG (GEMINI PRO)", variant="primary")
+        btn = gr.Button("🚀 RANCANG SEKARANG", variant="primary")
     
     with gr.Row():
         out_txt = gr.Markdown(label="Dokumen Teknis")