Updated app.py with new features/plots

app.py

@@ -2,6 +2,7 @@ import sympy as sp
 import numpy as np
 import pandas as pd
 import gradio as gr
+import matplotlib.pyplot as plt
 
 def solve_beam(l1_val, l2_val, q1_val, q2_val):
     # Define the variables
@@ -67,22 +68,46 @@ def solve_beam(l1_val, l2_val, q1_val, q2_val):
     
     return beam
 
+def plot_beam(beam_df):
+    fig, axs = plt.subplots(2, 1, figsize=(8, 8))
+
+    # Plot bending moment
+    axs[0].plot(beam_df['x'], beam_df['M'], label='Bending Moment (M)')
+    axs[0].invert_yaxis()
+    axs[0].set_title('Bending Moment Diagram')
+    axs[0].set_xlabel('Position along the beam (x)')
+    axs[0].set_ylabel('Bending Moment (M)')
+    axs[0].legend()
+
+    # Plot shear force
+    axs[1].plot(beam_df['x'], beam_df['V'], label='Shear Force (V)')
+    axs[1].invert_yaxis()
+    axs[1].set_title('Shear Force Diagram')
+    axs[1].set_xlabel('Position along the beam (x)')
+    axs[1].set_ylabel('Shear Force (V)')
+    axs[1].legend()
+
+    plt.tight_layout()
+    return fig
+
 def main(l1, l2, q1, q2):
     beam_df = solve_beam(l1, l2, q1, q2)
     csv_file = 'beam_analysis.csv'
     beam_df.to_csv(csv_file, index=False)
-    return beam_df, csv_file
+    fig = plot_beam(beam_df)
+    return beam_df, csv_file, fig
 
 inputs = [
-    gr.components.Number(label="Length of span 1 (l1[m])"),
-    gr.components.Number(label="Length of span 2 (l2[m])"),
-    gr.components.Number(label="Load on span 1 (q1[MN/m])"),
-    gr.components.Number(label="Load on span 2 (q2[MN/m])")
+    gr.components.Number(label="Length of span 1 (l1)"),
+    gr.components.Number(label="Length of span 2 (l2)"),
+    gr.components.Number(label="Load on span 1 (q1)"),
+    gr.components.Number(label="Load on span 2 (q2)")
 ]
 
 outputs = [
     gr.components.Dataframe(label="Beam Analysis Data"),
-    gr.components.File(label="Download CSV")
+    gr.components.File(label="Download CSV"),
+    gr.components.Plot(label="Bending Moment and Shear Force Diagrams")
 ]
 
-gr.Interface(fn=main, inputs=inputs, outputs=outputs, title="Beam Analysis Tool").launch()
+gr.Interface(fn=main, inputs=inputs, outputs=outputs, title="Beam Analysis Tool").launch()