Update app.py

app.py

@@ -2,10 +2,10 @@ import streamlit as st
 import pandas as pd
 import pickle
 import os
-from database import create_connection, insert_data
+from database import create_connection, insert_data, fetch_data
 
 # Database setup
-conn = create_connection('example_db')  # Assuming this function is correctly defined in database.py
+conn = create_connection('example_db')  # Adjust the name 'example_db' as necessary
 
 # Load the trained models
 current_dir = os.path.dirname(os.path.abspath(__file__))
@@ -172,21 +172,23 @@ with tabs[4]:
         
         # Insert supplier data
         supplier_data = pd.DataFrame({
-            'Part No': [part_no],
-            'Process Type': [process_type],
-            'Part OD': [part_od],
-            'Part ID': [part_id],
-            'Part Width': [part_width],
-            'Finish Wt': [finish_wt],
-            'Grade Type': [grade_type],
-            'Material ID': [material_id],
-            'Predicted Input Weight': [predicted_input_weight],
-            'Raw Material Cost': [raw_material_cost],
-            'Process Cost': [process_cost],
-            'Machining Cost': [machining_cost],
-            'Scrap Recovery': [scrap_recovery],
-            'Inspection Cost': [inspection_cost],
-            'Total Manufacturing Cost': [total_mg_cost]
+            'part_no': [part_no],
+            'process_type': [process_type],
+            'part_od': [part_od],
+            'part_id': [part_id],
+            'part_width': [part_width],
+            'finish_wt': [finish_wt],
+            'grade_type': [grade_type],
+            'material_id': [material_id],
+            'input_weight': [predicted_input_weight],
+            'raw_material_cost': [raw_material_cost],
+            'process_cost': [process_cost],
+            'machining_time': [predicted_machining_time],
+            'machining_cost': [machining_cost],
+            'scrap_recovery': [scrap_recovery],
+            'inspection_time': [predicted_inspection_time],
+            'inspection_cost': [inspection_cost],
+            'total_mg_cost': [total_mg_cost]
         })
         insert_data(conn, 'supplier_data', supplier_data)
         st.success("Supplier data added successfully")