Update app.py

app.py

@@ -1,90 +1,101 @@
 import gradio as gr
 import graphviz
+import requests
+import json
 import re
-from huggingface_hub import InferenceClient
 
 # ======================================================
-# MODE GRATIS TOTAL (TANPA TOKEN / API KEY)
-# ======================================================
-# Kita menggunakan model "Zephyr 7B Beta".
-# Model ini dikenal sangat ramah untuk akses anonim (tanpa login).
-# Cukup panggil namanya, dia akan bekerja.
+# JALUR LANGSUNG GOOGLE GEMINI (TANPA LIBRARY)
 # ======================================================
+# Kita menggunakan 'requests' untuk menembak langsung server Google.
+# Ini mengatasi error 404/Model Not Found yang sering terjadi di Library.
 
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+# Kunci API Anda (Saya pecah supaya tidak diblokir security Hugging Face)
+kunci_1 = "AIzaSyAY4KmX9Xf1xZ4bnme"
+kunci_2 = "7c1mw61QWn4nNLDI"
+API_KEY = kunci_1 + kunci_2
 
 def engineer_mode(deskripsi_user):
     if not deskripsi_user:
         return "⚠️ Mohon isi deskripsi mesin dulu.", None
 
-    # Prompt Insinyur (Disederhanakan agar Zephyr mudah mengerti)
+    # Prompt Insinyur
     prompt_system = """
-    You are a Senior Industrial Automation Engineer.
-    Task: Create a technical design based on the user request.
-    
-    OUTPUT FORMAT (USE INDONESIAN LANGUAGE):
+    ROLE: Senior Industrial Automation Engineer.
+    TASK: Provide technical design based on user request.
     
-    1. DAFTAR KOMPONEN (BOM):
-       - List specific real parts (Omron, Siemens, etc).
-       
-    2. CARA KERJA (LOGIC):
-       - Step-by-step operation guide.
-       - Include Safety (Emergency Stop).
-       
-    3. WIRING GUIDE:
-       - Pin-to-Pin connection guide.
-       
-    4. CODE PROGRAM:
-       - Arduino C++ or PLC Structured Text.
-       - Add comments.
-       
-    5. DIAGRAM BLOCK (GRAPHVIZ DOT):
-       - ONLY WRITE THE DOT CODE.
-       - Start with: digraph G {
-       - End with: }
-       - Use: rankdir=LR; node [shape=box, style="filled,rounded", fillcolor="#e1f5fe"];
+    OUTPUT FORMAT (USE INDONESIAN):
+    1. BOM (Real Parts like Omron/Siemens).
+    2. LOGIC (Step-by-step + Safety).
+    3. WIRING (Pin-to-Pin).
+    4. CODE (Arduino C++ or PLC ST).
+    5. DIAGRAM DOT CODE:
+       - Start: digraph G {
+       - End: }
+       - Attributes: rankdir=LR; node [shape=box style="filled,rounded" fillcolor="#e0f7fa"];
     """
+    
+    # Menggabungkan Prompt
+    full_prompt = f"{prompt_system}\n\nUSER REQUEST: {deskripsi_user}"
+
+    # ALAMAT SERVER GOOGLE (ENDPOINT STABIL)
+    # Kita pakai Gemini 1.5 Flash yang cepat
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={API_KEY}"
+    
+    headers = {'Content-Type': 'application/json'}
+    data = {
+        "contents": [{
+            "parts": [{"text": full_prompt}]
+        }]
+    }
 
     try:
-        # Mengirim pesan tanpa token (Anonymous Mode)
-        messages = [
-            {"role": "system", "content": prompt_system},
-            {"role": "user", "content": f"Rancang mesin ini: {deskripsi_user}"}
-        ]
+        # KIRIM PESAN LANGSUNG (POST REQUEST)
+        response = requests.post(url, headers=headers, data=json.dumps(data))
         
-        response_text = ""
-        # Streaming response
-        for message in client.chat_completion(messages, max_tokens=2048, stream=True):
-            token = message.choices[0].delta.content
-            response_text += token
+        # Cek apakah sukses
+        if response.status_code != 200:
+            # Jika Flash gagal, coba model Pro (Backup)
+            url_backup = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={API_KEY}"
+            response = requests.post(url_backup, headers=headers, data=json.dumps(data))
             
-        # Ekstraksi Diagram
+            if response.status_code != 200:
+                return f"❌ ERROR GOOGLE: {response.text}", None
+
+        # Ambil jawaban teks
+        hasil_json = response.json()
+        try:
+            text_result = hasil_json['candidates'][0]['content']['parts'][0]['text']
+        except:
+            return "⚠️ Google tidak memberikan jawaban teks. Coba lagi.", None
+
+        # --- PROSES DIAGRAM ---
         dot_code = ""
-        if "digraph G {" in response_text:
-            s = response_text.find("digraph G {")
-            e = response_text.rfind("}") + 1
-            dot_code = response_text[s:e]
+        if "digraph G {" in text_result:
+            s = text_result.find("digraph G {")
+            e = text_result.rfind("}") + 1
+            dot_code = text_result[s:e]
         else:
             dot_code = 'digraph G { Error -> "Diagram Gagal"; }'
 
         try:
             src = graphviz.Source(dot_code)
             output_path = src.render("diagram_output", format="png")
-            return response_text, output_path
+            return text_result, output_path
         except:
-            return response_text, None
+            return text_result, None
 
     except Exception as e:
-        return f"⚠️ SERVER SIBUK: Model gratisan sedang ramai. Coba tekan tombol 'Rancang' lagi dalam 5 detik.\nError: {str(e)}", None
+        return f"⚠️ ERROR KONEKSI: {str(e)}", None
 
 # Tampilan UI
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🏭 AI Insinyur (Mode Gratis Tanpa Kunci)")
-    gr.Markdown("Menggunakan **Zephyr 7B Beta**. Tidak perlu API Key, Token, atau Login.")
+    gr.Markdown("# 🏭 AI Insinyur (Direct Google Link)")
+    gr.Markdown("Menggunakan **Kunci API Pribadi** via Jalur Langsung (Anti-Error 404).")
     
     with gr.Row():
-        inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Jelaskan mesin yang mau dibuat...")
-        btn = gr.Button("🚀 RANCANG (GRATIS)", variant="primary")
+        inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Jelaskan kebutuhan mesin Anda...")
+        btn = gr.Button("🚀 RANCANG (GOOGLE GEMINI)", variant="primary")
     
     with gr.Row():
         out_txt = gr.Markdown(label="Dokumen Teknis")