Update app.py

app.py

@@ -1,54 +1,31 @@
-
-# Import necessary libraries
 import gradio as gr
 
-# Function to calculate load distribution
-def load_calculation(load_rating, num_of_loads, voltage, load_type):
-    # Constants
-    safety_factor = 1.25  # 25% safety margin
-    cable_size_factor = 1.5  # Approximation factor for cable size
-
-    # Calculate total load (kW to kVA conversion for inductive loads)
-    if load_type == "Inductive":
-        total_load = load_rating * num_of_loads * 1.2  # Assuming 0.8 power factor
-    else:
-        total_load = load_rating * num_of_loads
-
-    # Calculate current
-    current = (total_load * 1000) / (voltage * (3**0.5) if voltage > 240 else voltage)
-
+# 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
-    breaker_size = round(current * safety_factor, 2)
-
-    # Calculate cable size (approximation)
-    cable_size = round(current * cable_size_factor, 2)
-
-    # Return the results
-    return f"""
-    Total Load: {round(total_load, 2)} kW
-    Recommended Breaker Size: {breaker_size} A
-    Suggested Cable Size: {cable_size} mm²
-    """
-
-# Gradio interface
-def interface():
-    # Inputs
-    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)")
-    load_type = gr.Dropdown(choices=["Resistive", "Inductive"], label="Load Type")
-
-    # Outputs
-    output = gr.Textbox(label="Load Calculation Results")
-
-    # Launch the Gradio app
-    gr.Interface(
-        fn=load_calculation,
-        inputs=[load_rating, num_of_loads, voltage, load_type],
-        outputs=output,
-        title="Load Calculation Assistant",
-        description="Calculate load, breaker size, and cable size for electrical systems."
-    ).launch()
-
-# Run the interface
-interface()
+    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()