Update app.py

app.py

@@ -1,6 +1,15 @@
 import streamlit as st
 import joblib
 from models.optimizer import optimize_design
+
+# Correct usage of the 'with' statement
+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
+st.write(f"Tool dimensions: {tool_width}mm x {tool_height}mm")
+
 # Load pre-trained model
 model = joblib.load('models/defect_model.pkl')
 st.title("AI-Driven Press Tool Defect Predictor & Optimizer")