Update app.py

app.py

@@ -2,43 +2,43 @@ import streamlit as st
 import joblib
 from models.optimizer import optimize_design
 
-# Correct usage of the 'with' statement
+# Sidebar for user inputs
 with st.sidebar:
     tool_width = st.number_input("Tool Width (mm)", value=50)
     tool_height = st.number_input("Tool Height (mm)", value=20)
 
-# Additional code outside the 'with' block
+# Display tool dimensions
 st.write(f"Tool dimensions: {tool_width}mm x {tool_height}mm")
 
 # Load pre-trained model
 model = joblib.load('models/defect_model.pkl')
+
+# Title
 st.title("AI-Driven Press Tool Defect Predictor & Optimizer")
-# Input Form
-import streamlit as st
 
-# Example of a correct 'with' statement usage
-with st.sidebar:
-    tool_width = st.number_input("Tool Width (mm)", value=50)  # Indented properly
-    tool_height = st.number_input("Tool Height (mm)", value=20)  # Also indented properly
+# Input Form
+with st.form("Input Parameters"):
+    # Form input fields
+    tool_width = st.number_input("Tool Width (mm)", value=50)
+    tool_height = st.number_input("Tool Height (mm)", value=20)
+    material_strength = st.number_input("Material Strength (MPa)", value=300)
+    press_force = st.number_input("Press Force (kN)", value=100)
 
-# Code outside the 'with' block
-st.write(f"Tool dimensions: {tool_width}mm x {tool_height}mm")
+    # Submit button
+    submit = st.form_submit_button("Predict & Optimize")
 
-with st.form("Input Parameters"):
-tool_width = st.number_input("Tool Width (mm)", value=50)
-tool_height = st.number_input("Tool Height (mm)", value=20)
-material_strength = st.number_input("Material Strength (MPa)", value=300)
-press_force = st.number_input("Press Force (kN)", value=100)
-submit = st.form_submit_button("Predict & Optimize")
+# Handle form submission
 if submit:
-# Predict defect
-input_data = [[tool_width, tool_height, material_strength, press_force]]
-defect = model.predict(input_data)[0]
-st.subheader("Predicted Defect")
-st.write(f"Predicted Defect Type: **{defect}**")
-# Optimize design
-optimized_params = optimize_design(tool_width, tool_height, material_strength,
-press_force)
-st.subheader("Optimized Design Parameters")
-st.write(optimized_params)
-st.success("Optimization and Prediction Complete!")
+    # Predict defect
+    input_data = [[tool_width, tool_height, material_strength, press_force]]
+    defect = model.predict(input_data)[0]
+    st.subheader("Predicted Defect")
+    st.write(f"Predicted Defect Type: **{defect}**")
+
+    # Optimize design
+    optimized_params = optimize_design(tool_width, tool_height, material_strength, press_force)
+    st.subheader("Optimized Design Parameters")
+    st.write(optimized_params)
+
+    # Completion message
+    st.success("Optimization and Prediction Complete!")