Arif571 commited on
Commit
75a7093
·
verified ·
1 Parent(s): 374b848

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +82 -0
app.py ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import re
3
+ import google.generativeai as genai
4
+ import gradio as gr
5
+ import graphviz
6
+
7
+ # --- SETUP API KEY ---
8
+ # Ini API Key Anda yang sudah saya tanam langsung
9
+ MY_API_KEY = "AIzaSyAY4KmX9Xf1xZ4bnme7c1mw61QWn4nNLDI"
10
+
11
+ try:
12
+ genai.configure(api_key=MY_API_KEY)
13
+ # Kita pakai model Flash agar cepat & gratis
14
+ model = genai.GenerativeModel('gemini-1.5-flash')
15
+ except Exception as e:
16
+ print(f"Error Setup: {e}")
17
+
18
+ # --- LOGIKA INSINYUR (BRAIN) ---
19
+ def engineer_mode(deskripsi_user):
20
+ if not deskripsi_user:
21
+ return "⚠️ Mohon isi deskripsi mesin dulu.", None
22
+
23
+ final_prompt = f"""
24
+ BERTINDAKLAH SEBAGAI: Senior Industrial Automation Engineer.
25
+ TUGAS: Buatkan rancangan teknis detail untuk: "{deskripsi_user}"
26
+
27
+ OUTPUT WAJIB (BAHASA INDONESIA):
28
+ 1. DAFTAR KOMPONEN (Part Nyata & Merk)
29
+ 2. CARA KERJA (Step-by-step Logika Safety)
30
+ 3. WIRING GUIDE (Panduan Kabel Pin-to-Pin)
31
+ 4. PROGRAM CODE (Arduino C++ / PLC)
32
+ 5. DIAGRAM BLOK (HANYA KODE GRAPHVIZ DOT)
33
+ - Mulai dengan: digraph G {{
34
+ - Akhiri dengan: }}
35
+ - Gunakan: rankdir=LR; node [shape=box, style=filled, fillcolor=lightblue];
36
+ """
37
+
38
+ try:
39
+ response = model.generate_content(final_prompt)
40
+ text = response.text
41
+
42
+ # Ekstraksi Diagram
43
+ dot_code = ""
44
+ match = re.search(r'digraph G \{.*\}', text, re.DOTALL)
45
+ if match:
46
+ dot_code = match.group(0)
47
+ else:
48
+ if "digraph" in text:
49
+ s = text.find("digraph")
50
+ e = text.rfind("}") + 1
51
+ dot_code = text[s:e]
52
+ else:
53
+ dot_code = 'digraph G { Error -> "Diagram Gagal"; }'
54
+
55
+ # Render Gambar
56
+ try:
57
+ src = graphviz.Source(dot_code)
58
+ # Di HuggingFace kita render ke file sementara
59
+ output_path = src.render("diagram_output", format="png")
60
+ return text, output_path
61
+ except:
62
+ return text, None
63
+
64
+ except Exception as e:
65
+ return f"Error AI: {str(e)}", None
66
+
67
+ # --- TAMPILAN (UI) ---
68
+ with gr.Blocks(theme=gr.themes.Soft()) as app:
69
+ gr.Markdown("# 🏭 AI Perancang Mesin (Fast Version)")
70
+ gr.Markdown("Masukkan ide mesin, AI akan buatkan Part List, Wiring, Program & Diagram.")
71
+
72
+ with gr.Row():
73
+ inp = gr.Textbox(lines=4, label="Deskripsi Mesin", placeholder="Contoh: Alat sortir otomatis...")
74
+ btn = gr.Button("🚀 RANCANG SEKARANG", variant="primary")
75
+
76
+ with gr.Row():
77
+ out_txt = gr.Markdown(label="Dokumen Teknis")
78
+ out_img = gr.Image(label="Diagram Sistem")
79
+
80
+ btn.click(engineer_mode, inputs=inp, outputs=[out_txt, out_img])
81
+
82
+ app.launch()