Update app.py

app.py

@@ -1,10 +1,27 @@
 import gradio as gr
 import math
 import itertools
+import ezdxf
+import os
+import groq
+
+# Initialize Groq client
+client = groq.Client(api_key=os.getenv("GROQ_API_KEY"))
 
 # Available Capacitor Units (kVAR)
 available_capacitors = [25, 20, 15, 10, 5, 2.5, 1.5, 1]
 
+# Prompt Groq for explanation (optional)
+def ask_groq(prompt):
+    try:
+        response = client.chat.completions.create(
+            model="llama3-8b-8192",
+            messages=[{"role": "user", "content": prompt}]
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Groq Error: {str(e)}"
+
 def calculate_power_parameters(voltage, current, power_factor):
     if voltage > 0 and current > 0:
         apparent_power = math.sqrt(3) * voltage * current
@@ -28,14 +45,16 @@ def design_capacitor_bank(reactive_power, num_caps):
         best_combo = None
         min_error = float('inf')
 
-        for combo in itertools.combinations_with_replacement(available_capacitors, num_caps):
+        # Allow repetition only when needed
+        combos = itertools.combinations_with_replacement(available_capacitors, num_caps)
+        for combo in combos:
+            if len(set(combo)) < len(combo) and len(set(combo)) != 1:
+                continue
             total = sum(combo)
             error = abs(total - reactive_power)
-
             if error < min_error:
                 min_error = error
                 best_combo = combo
-
             if error == 0:
                 break
 
@@ -47,19 +66,33 @@ def design_capacitor_bank(reactive_power, num_caps):
                 "suggested_capacitors": suggested_capacitors,
                 "total_kvar": round(total_kvar, 2),
                 "message": message,
+                "combo": best_combo
             }
         else:
             return {"message": "Could not find a suitable combination."}
     else:
         return None
 
+def create_dxf_capacitor_bank(capacitors):
+    doc = ezdxf.new()
+    msp = doc.modelspace()
+    x = 0
+    for idx, cap in enumerate(capacitors):
+        label = f"{cap} kVAR"
+        msp.add_lwpolyline([(x, 0), (x+10, 0), (x+10, 10), (x, 10), (x, 0)])
+        msp.add_text(label, dxfattribs={'height': 2.5}).set_pos((x, 12))
+        x += 15
+    output_path = "/mnt/data/capacitor_bank.dxf"
+    doc.saveas(output_path)
+    return output_path
+
 def reactive_power_first(voltage, current, power_factor):
     power_results = calculate_power_parameters(voltage, current, power_factor)
     if power_results:
-        apparent_power_out = f"Apparent Power: {power_results['apparent_power']} VA"
-        real_power_out = f"Real Power: {power_results['real_power']} kW"
-        reactive_power_out = f"Reactive Power: {power_results['reactive_power']} kVAR"
-        calculated_pf_out = f"Calculated Power Factor: {power_results['calculated_pf']}"
+        apparent_power_out = f"Apparent Power: **{power_results['apparent_power']} VA**"
+        real_power_out = f"Real Power: **{power_results['real_power']} kW**"
+        reactive_power_out = f"Reactive Power: **{power_results['reactive_power']} kVAR**"
+        calculated_pf_out = f"Calculated Power Factor: **{power_results['calculated_pf']}**"
         return (
             apparent_power_out,
             real_power_out,
@@ -74,17 +107,19 @@ def finalize_capacitor_bank(reactive_power, num_caps):
     cap_bank_design = design_capacitor_bank(reactive_power, num_caps)
     if cap_bank_design and cap_bank_design.get("suggested_capacitors"):
         suggested_capacitors_text = "<br>".join(
-            [f"🔹 Capacitor {idx + 1}: {cap}" for idx, cap in enumerate(cap_bank_design['suggested_capacitors'])]
+            [f"🔹 Capacitor {idx + 1}: **{cap}**" for idx, cap in enumerate(cap_bank_design['suggested_capacitors'])]
         )
-        return suggested_capacitors_text, cap_bank_design['message']
+        dxf_path = create_dxf_capacitor_bank(cap_bank_design["combo"])
+        return suggested_capacitors_text, cap_bank_design['message'], dxf_path
     else:
-        return "Could not find a suitable combination.", ""
+        return "Could not find a suitable combination.", "", None
 
 with gr.Blocks() as iface:
     gr.Markdown("# ⚡ Three-Phase Power Calculator - Reactive Power Compensation")
     gr.Markdown("""
-    Step 1: Enter system parameters to calculate apparent and reactive power.
-    Step 2: Input number of capacitors to compute optimal configuration.
+    Step 1: Enter system parameters to calculate apparent and reactive power.<br>
+    Step 2: Input number of capacitors to compute optimal configuration.<br>
+    Step 3: Download AutoCAD (.dxf) layout.
     """)
 
     with gr.Row():
@@ -99,7 +134,7 @@ with gr.Blocks() as iface:
     real_power_out = gr.HTML()
     reactive_power_out = gr.HTML()
     calculated_pf_out = gr.HTML()
-    reactive_value = gr.State()  # Store the reactive power for reuse
+    reactive_value = gr.Number(visible=False)
 
     calc_btn.click(
         fn=reactive_power_first,
@@ -119,11 +154,12 @@ with gr.Blocks() as iface:
 
     capacitor_out = gr.HTML()
     total_comp_out = gr.HTML()
+    dxf_file = gr.File(label="📥 Download AutoCAD File", visible=False)
 
     finalize_btn.click(
         fn=finalize_capacitor_bank,
         inputs=[reactive_value, num_caps_input],
-        outputs=[capacitor_out, total_comp_out]
+        outputs=[capacitor_out, total_comp_out, dxf_file]
     )
 
 if __name__ == "__main__":