Update app.py

app.py

@@ -2,6 +2,32 @@ import streamlit as st
 import pandas as pd
 import pickle
 from models.optimizer import optimize_design
+import pandas as pd
+import os
+import sys
+
+# Ensure the project root is in the Python path
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from models.optimizer import optimize_design
+
+def main():
+    print("Welcome to the Press Tool AI application!")
+
+    # Load sample data
+    data_path = "data/sample_data.csv"
+    data = pd.read_csv(data_path)
+    print("Sample Data Loaded:")
+    print(data.head())
+
+    # Call optimization logic
+    optimized_result = optimize_design(data)
+    print("Optimization Complete. Results:")
+    print(optimized_result)
+
+if __name__ == "__main__":
+    main()
+
 
 # Load pre-trained model
 def load_model():