Update app.py

app.py

@@ -1,47 +1,57 @@
 import gradio as gr
 import pandas as pd
 import matplotlib.pyplot as plt
+from io import BytesIO
 
 # 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
+    try:
+        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)
+        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
+        # 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
 
-    apparent_power = total_load / power_factor  # Apparent Power in kVA
+        # Prepare results
+        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],
+        })
 
-    # 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()
+        
+        # Save the plot to a BytesIO object
+        buf = BytesIO()
+        plt.savefig(buf, format='png')
+        buf.seek(0)
+        plt.close()
 
-    # 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 Excel data to a BytesIO object
+        excel_buf = BytesIO()
+        results.to_excel(excel_buf, index=False)
+        excel_buf.seek(0)
 
-    # 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",
-    )
+        return (
+            round(total_load, 2),
+            round(total_current, 2),
+            round(breaker_size, 2),
+            round(cable_size, 2),
+            round(apparent_power, 2),
+            buf,
+            excel_buf,
+        )
+    except Exception as e:
+        return f"An error occurred: {e}"
 
 # Gradio Interface
 def interface():
@@ -60,6 +70,7 @@ def interface():
         gr.Textbox(label="Recommended Cable Size (mm²)"),
         gr.Textbox(label="Total Apparent Power (kVA)"),
         gr.Image(label="Graphical Visualization"),  # Graph output
+        gr.File(label="Download Excel Results"),  # Excel file output
     ]
 
     # Launch Interface