Update app.py

app.py

@@ -3,17 +3,15 @@ import math
 import itertools
 import ezdxf
 import os
-import groq
+from groq import Groq  # ✅ Correct class
 from ezdxf import zoom
 from ezdxf.enums import TextEntityAlignment
 
-# Initialize Groq client
-client = groq.Client(api_key=os.getenv("GROQ_API_KEY"))
+# ✅ Initialize Groq client correctly
+client = Groq(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(
@@ -39,15 +37,13 @@ def calculate_power_parameters(voltage, current, power_factor):
             "reactive_power": round(reactive_power, 2),
             "calculated_pf": round(calculated_pf, 2)
         }
-    else:
-        return None
+    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)
@@ -68,47 +64,37 @@ def design_capacitor_bank(reactive_power, num_caps):
                 "message": message,
                 "combo": best_combo
             }
-        else:
-            return {"message": "Could not find a suitable combination."}
-    else:
-        return None
+    return None
 
 def create_dxf_capacitor_bank(capacitors):
     doc = ezdxf.new()
     msp = doc.modelspace()
     x = 0
     y = 0
-    row_width = 15  # Distance between capacitors in a row
+    row_width = 15
     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})  # Color 3 = Green
-        
-        # Add Text with more control
+        msp.add_lwpolyline(points, close=True, dxfattribs={'color': 3})
         text = msp.add_text(label, dxfattribs={
             'height': 2.5,
-            'color': 4, # color Cyan
-            'style': 'STANDARD',  # You can define text styles in DXF
-            'halign': TextEntityAlignment.CENTER,  # Horizontal alignment
-            'valign': TextEntityAlignment.BOTTOM,  # Vertical alignment
+            'color': 4,
+            'style': 'STANDARD',
+            'halign': TextEntityAlignment.CENTER,
+            'valign': TextEntityAlignment.BOTTOM,
         })
-        text.dxf.insert = (x + 5, y + 5)  # Position at center of rectangle
-        
+        text.dxf.insert = (x + 5, y + 5)
         x += row_width
-        if (idx + 1) % max_in_row == 0:  # Move to the next row
+        if (idx + 1) % max_in_row == 0:
             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 to extents
-    zoom.extents(msp, factor=1.1)  # Add a small padding
+    zoom.extents(msp, factor=1.1)
     
     output_path = "capacitor_bank_layout.dxf"
     doc.saveas(output_path)
@@ -117,19 +103,14 @@ 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 (
-            apparent_power_out,
-            real_power_out,
-            reactive_power_out,
-            calculated_pf_out,
+            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']}**",
             power_results['reactive_power']
         )
-    else:
-        return ("⚠️ Please enter valid Voltage and Current!", "", "", "", 0)
+    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)
@@ -139,8 +120,7 @@ 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
-    else:
-        return "Could not find a suitable combination.", "", None
+    return "Could not find a suitable combination.", "", None
 
 with gr.Blocks() as iface:
     gr.Markdown("# ⚡ Three-Phase Power Calculator - Reactive Power Compensation")
@@ -157,7 +137,6 @@ 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()
@@ -167,19 +146,12 @@ 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")