Update app.py

app.py

@@ -1,62 +1,72 @@
+import os
 import gradio as gr
 import graphviz
 import re
 from huggingface_hub import InferenceClient
 
-# --- 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")
+# ======================================================
+# SETUP AMAN (MENGAMBIL DARI SETTINGS)
+# ======================================================
+# Kita mengambil token dari "Secrets" yang baru Anda buat
+# Jadi tidak ada kode rahasia yang terlihat di sini.
+hf_token = os.getenv("HF_TOKEN")
+
+# Inisialisasi Model Qwen 2.5 (72B)
+client = InferenceClient("Qwen/Qwen2.5-72B-Instruct", token=hf_token)
 
 def engineer_mode(deskripsi_user):
     if not deskripsi_user:
         return "⚠️ Mohon isi deskripsi mesin dulu.", None
 
-    # Prompt Insinyur yang sama, tapi dioptimalkan untuk Qwen
+    # Prompt Insinyur
     prompt_system = """
-    Anda adalah Senior Industrial Automation Engineer (20+ Tahun Pengalaman).
-    Tugas Anda adalah merancang sistem mesin pabrik secara presisi, logis, dan aman.
+    PERAN: Senior Industrial Automation Engineer (20+ Tahun).
+    TUGAS: Terjemahkan ide user menjadi Dokumen Teknis Siap Rakit.
     
-    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.
+    OUTPUT WAJIB (BAHASA INDONESIA):
+    1. BILL OF MATERIALS (BOM):
+       - Komponen NYATA (Omron, Siemens, Schneider).
+       - Tipe spesifik (Misal: MY2N-J Relay, E3Z-D61 Sensor).
+       
+    2. LOGIKA OPERASIONAL (SEQUENCE):
+       - Step-by-step dari Start sampai Stop.
+       - WAJIB ADA: Logika Safety (Emergency Stop).
+       
+    3. WIRING DIAGRAM GUIDE:
+       - Koneksi Pin-to-Pin yang detail.
+       - Pisahkan Power (24V) dan Signal (I/O).
+       
+    4. PROGRAM CODE:
+       - Arduino C++ atau PLC Structured Text.
+       - Lengkap dengan komentar penjelasan.
+       
+    5. VISUALISASI BLOK (GRAPHVIZ DOT):
+       - Hanya kode DOT.
+       - Mulai: digraph G {
+       - Akhiri: }
+       - Gunakan: rankdir=LR; node [shape=box, style="filled,rounded", fontname="Arial"];
+       - Warna: Input (Hijau Muda #c8e6c9), Proses (Kuning #fff9c4), Output (Merah #ffccbc).
     """
 
     try:
-        # Mengirim pesan ke Model Open Source
         messages = [
             {"role": "system", "content": prompt_system},
-            {"role": "user", "content": f"Rancang mesin ini: {deskripsi_user}"}
+            {"role": "user", "content": f"Rancang detail untuk: {deskripsi_user}"}
         ]
         
-        # Streaming respons (agar tidak timeout)
         response_text = ""
-        for message in client.chat_completion(messages, max_tokens=2000, stream=True):
+        for message in client.chat_completion(messages, max_tokens=3000, stream=True):
             token = message.choices[0].delta.content
             response_text += token
             
-        # --- PROSES DIAGRAM (VISUALISASI) ---
+        # Ekstraksi Diagram
         dot_code = ""
-        # Mencari pola kode DOT di dalam teks
-        match = re.search(r'digraph G \{.*\}', response_text, re.DOTALL)
-        
-        if match:
-            dot_code = match.group(0)
+        if "digraph G {" in response_text:
+            s = response_text.find("digraph G {")
+            e = response_text.rfind("}") + 1
+            dot_code = response_text[s:e]
         else:
-            # 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"; }'
+            dot_code = 'digraph G { Error -> "Diagram Tidak Tergenerate"; }'
 
         # Render Gambar
         try:
@@ -67,19 +77,19 @@ def engineer_mode(deskripsi_user):
             return response_text, None
 
     except Exception as e:
-        return f"⚠️ Terjadi Kepadatan Server (Busy). Coba tekan tombol Rancang lagi dalam beberapa detik.\nError: {str(e)}", None
+        return f"⚠️ ERROR TOKEN: Cek Settings > Secrets Anda.\nDetail: {str(e)}", None
 
-# --- TAMPILAN APLIKASI ---
+# Tampilan UI
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🏭 AI Insinyur (Open Source Qwen 2.5)")
-    gr.Markdown("Menggunakan model **Qwen-72B** yang gratis, tanpa API Key, dan presisi tinggi.")
+    gr.Markdown("# 🏭 AI Insinyur (Secure Version)")
+    gr.Markdown("Model: **Qwen 2.5 (72B)** | Status: **Secure Token**")
     
     with gr.Row():
-        inp = gr.Textbox(lines=5, label="Deskripsi Mesin", placeholder="Contoh: Alat pres hidrolik otomatis...")
-        btn = gr.Button("🚀 RANCANG (GRATIS)", variant="primary")
+        inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Jelaskan kebutuhan mesin Anda...")
+        btn = gr.Button("🚀 RANCANG TEKNIS", variant="primary")
     
     with gr.Row():
-        out_txt = gr.Markdown(label="Dokumen Teknis")
+        out_txt = gr.Markdown(label="Dokumen Spesifikasi")
         out_img = gr.Image(label="Diagram Sistem")
             
     btn.click(engineer_mode, inputs=inp, outputs=[out_txt, out_img])