Update app.py

app.py

@@ -1,31 +1,74 @@
 import gradio as gr
+import pandas as pd
+import matplotlib.pyplot as plt
 
-# Function to calculate load, breaker size, and cable size
-def calculate_load(load_rating, num_of_loads, voltage, load_type):
-    # Convert total load to watts
-    total_load_kw = load_rating * num_of_loads
-    total_load_w = total_load_kw * 1000  # Convert kW to W
-    
-    # Calculate breaker size
-    if load_type == "Resistive":
-        breaker_size = total_load_w / (voltage * 0.8)
-    elif load_type == "Inductive":
-        breaker_size = total_load_w / (voltage * 0.9)
-    
-    # Calculate cable size (simplified estimation)
-    cable_size = total_load_w / (voltage * 2)
-    
-    # Prepare the output
-    result = f"Total Load: {total_load_kw} kW\nBreaker Size: {breaker_size:.2f} A\nCable Size: {cable_size:.2f} mm²"
-    return result
-
-# Set up the Gradio interface
-interface = gr.Interface(fn=calculate_load,
-                         inputs=[gr.Number(label="Load Rating per Device (kW)"),
-                                 gr.Number(label="Number of Loads"),
-                                 gr.Number(label="Voltage (V)"),
-                                 gr.Dropdown(choices=["Resistive", "Inductive"], label="Load Type")],
-                         outputs="text")
-
-if __name__ == "__main__":
-    interface.launch()
+# Updated Calculation Function
+def load_calculator(load_rating, num_of_loads, voltage, power_factor, safety_factor, phase_type):
+    total_load = load_rating * num_of_loads  # Total Load in kW
+    total_current = (total_load * 1000) / (voltage * power_factor)  # Current in Amperes
+
+    if phase_type == "Three-Phase":
+        total_current /= (3 ** 0.5)
+
+    # Adjust breaker and cable sizes using safety factor
+    breaker_size = total_current * safety_factor
+    cable_size = total_current * safety_factor * 0.75  # Example adjustment for cable size
+
+    apparent_power = total_load / power_factor  # Apparent Power in kVA
+
+    # Save results to DataFrame
+    results = pd.DataFrame({
+        "Metric": ["Total Load (kW)", "Total Current (A)", "Breaker Size (A)", "Cable Size (mm²)", "Apparent Power (kVA)"],
+        "Value": [total_load, total_current, breaker_size, cable_size, apparent_power],
+    })
+
+    # Generate bar chart
+    plt.figure(figsize=(6, 4))
+    plt.bar(results["Metric"], results["Value"], color="skyblue")
+    plt.title("Load Calculation Results")
+    plt.xticks(rotation=45, ha='right')
+    plt.tight_layout()
+    plt.savefig("results_chart.png")
+    plt.close()
+
+    # Save results to Excel
+    results.to_excel("load_results.xlsx", index=False)
+
+    return (
+        round(total_load, 2),
+        round(total_current, 2),
+        round(breaker_size, 2),
+        round(cable_size, 2),
+        round(apparent_power, 2),
+        "results_chart.png",
+    )
+
+# Gradio Interface
+def interface():
+    load_rating = gr.Number(label="Load Rating per Device (kW)")
+    num_of_loads = gr.Number(label="Number of Loads")
+    voltage = gr.Number(label="Voltage (230V for single-phase, 400V for three-phase)")
+    power_factor = gr.Number(label="Power Factor (0.8 for inductive loads, 1 for resistive)")
+    safety_factor = gr.Number(label="Safety Factor (e.g., 1.25)", value=1.25)
+    phase_type = gr.Radio(["Single-Phase", "Three-Phase"], label="Phase Type")
+
+    # Outputs
+    outputs = [
+        gr.Textbox(label="Total Load (kW)"),
+        gr.Textbox(label="Total Current (A)"),
+        gr.Textbox(label="Recommended Breaker Size (A)"),
+        gr.Textbox(label="Recommended Cable Size (mm²)"),
+        gr.Textbox(label="Total Apparent Power (kVA)"),
+        gr.Image(label="Graphical Visualization"),  # Graph output
+    ]
+
+    # Launch Interface
+    gr.Interface(
+        fn=load_calculator,
+        inputs=[load_rating, num_of_loads, voltage, power_factor, safety_factor, phase_type],
+        outputs=outputs,
+        title="Enhanced Load Calculation Assistant",
+        description="Calculate electrical loads with safety factors, export results, and visualize data.",
+    ).launch()
+
+interface()