Update app.py

app.py

@@ -3,15 +3,17 @@ import math
 import itertools
 import ezdxf
 import os
-from groq import Groq  # ✅ Correct class
+import groq
 from ezdxf import zoom
 from ezdxf.enums import TextEntityAlignment
 
-# ✅ Initialize Groq client correctly
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+# 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(
@@ -37,13 +39,15 @@ def calculate_power_parameters(voltage, current, power_factor):
             "reactive_power": round(reactive_power, 2),
             "calculated_pf": round(calculated_pf, 2)
         }
-    return None
+    else:
+        return None
 
 def design_capacitor_bank(reactive_power, num_caps):
     if reactive_power > 0 and num_caps > 0:
         best_combo = None
         min_error = float('inf')
 
+        # Allow repetition freely to match reactive power
         combos = itertools.combinations_with_replacement(available_capacitors, num_caps)
         for combo in combos:
             total = sum(combo)
@@ -64,37 +68,47 @@ def design_capacitor_bank(reactive_power, num_caps):
                 "message": message,
                 "combo": best_combo
             }
-    return None
+        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
     y = 0
-    row_width = 15
+    row_width = 15  # Distance between capacitors in a row
     row_height = 20
     max_in_row = 5
     
     for idx, cap in enumerate(capacitors):
         label = f"{cap} kVAR"
+        # Draw rectangle for capacitor
         points = [(x, y), (x + 10, y), (x + 10, y + 10), (x, y + 10), (x, y)]
-        msp.add_lwpolyline(points, close=True, dxfattribs={'color': 3})
+        msp.add_lwpolyline(points, close=True, dxfattribs={'color': 3})  # Color 3 = Green
+        
+        # Add Text with more control
         text = msp.add_text(label, dxfattribs={
             'height': 2.5,
-            'color': 4,
-            'style': 'STANDARD',
-            'halign': TextEntityAlignment.CENTER,
-            'valign': TextEntityAlignment.BOTTOM,
+            'color': 4, # color Cyan
+            'style': 'STANDARD',  # You can define text styles in DXF
+            'halign': TextEntityAlignment.CENTER,  # Horizontal alignment
+            'valign': TextEntityAlignment.BOTTOM if hasattr(TextEntityAlignment, 'BOTTOM') else 2,
         })
-        text.dxf.insert = (x + 5, y + 5)
+        text.dxf.insert = (x + 5, y + 5)  # Position at center of rectangle
+        
         x += row_width
-        if (idx + 1) % max_in_row == 0:
+        if (idx + 1) % max_in_row == 0:  # Move to the next row
             x = 0
             y += row_height
 
+    # Add a title
     title_text = msp.add_text("Capacitor Bank Layout", dxfattribs={'height': 5, 'color': 1})
     title_text.dxf.insert = (0, y + 30)
-    zoom.extents(msp, factor=1.1)
+    
+    # Zoom to extents
+    zoom.extents(msp, factor=1.1)  # Add a small padding
     
     output_path = "capacitor_bank_layout.dxf"
     doc.saveas(output_path)
@@ -103,14 +117,19 @@ def create_dxf_capacitor_bank(capacitors):
 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']}**"
         return (
-            f"Apparent Power: **{power_results['apparent_power']} VA**",
-            f"Real Power: **{power_results['real_power']} kW**",
-            f"Reactive Power: **{power_results['reactive_power']} kVAR**",
-            f"Calculated Power Factor: **{power_results['calculated_pf']}**",
+            apparent_power_out,
+            real_power_out,
+            reactive_power_out,
+            calculated_pf_out,
             power_results['reactive_power']
         )
-    return ("⚠️ Please enter valid Voltage and Current!", "", "", "", 0)
+    else:
+        return ("⚠️ Please enter valid Voltage and Current!", "", "", "", 0)
 
 def finalize_capacitor_bank(reactive_power, num_caps):
     cap_bank_design = design_capacitor_bank(reactive_power, num_caps)
@@ -120,7 +139,8 @@ def finalize_capacitor_bank(reactive_power, num_caps):
         )
         dxf_path = create_dxf_capacitor_bank(cap_bank_design["combo"])
         return suggested_capacitors_text, cap_bank_design['message'], dxf_path
-    return "Could not find a suitable combination.", "", None
+    else:
+        return "Could not find a suitable combination.", "", None
 
 with gr.Blocks() as iface:
     gr.Markdown("# ⚡ Three-Phase Power Calculator - Reactive Power Compensation")
@@ -137,6 +157,7 @@ with gr.Blocks() as iface:
         frequency = gr.Radio(label="Select Frequency", choices=[50, 60], value=50)
 
     calc_btn = gr.Button("🔍 Calculate Power Parameters")
+
     apparent_power_out = gr.HTML()
     real_power_out = gr.HTML()
     reactive_power_out = gr.HTML()
@@ -146,12 +167,19 @@ with gr.Blocks() as iface:
     calc_btn.click(
         fn=reactive_power_first,
         inputs=[voltage, current, power_factor],
-        outputs=[apparent_power_out, real_power_out, reactive_power_out, calculated_pf_out, reactive_value]
+        outputs=[
+            apparent_power_out,
+            real_power_out,
+            reactive_power_out,
+            calculated_pf_out,
+            reactive_value
+        ]
     )
 
     gr.Markdown("### ➕ Enter number of capacitors to compensate reactive power:")
     num_caps_input = gr.Number(label="Number of Capacitors", precision=0)
     finalize_btn = gr.Button("⚙️ Generate Capacitor Bank")
+
     capacitor_out = gr.HTML()
     total_comp_out = gr.HTML()
     dxf_file = gr.File(label="📥 Download AutoCAD File")