Update app.py

app.py

@@ -96,6 +96,10 @@ if selected_page == "Order Management":
                 'status': [status]
             })
             orders = pd.concat([orders, new_order], ignore_index=True)
+            # Insert into the database
+            insert_data('orders', new_order.to_dict(orient='records')[0])
+            # Re-fetch data to reflect the changes
+            orders = fetch_data('orders')
 
     # Display order list
     st.write(orders)
@@ -120,6 +124,10 @@ elif selected_page == "Driver Management":
                 'status': [status]
             })
             drivers = pd.concat([drivers, new_driver], ignore_index=True)
+            # Insert into the database
+            insert_data('drivers', new_driver.to_dict(orient='records')[0])
+            # Re-fetch data to reflect the changes
+            drivers = fetch_data('drivers')
 
     # Display driver list
     st.write(drivers)
@@ -140,6 +148,10 @@ elif selected_page == "Zone Pressure":
                 'pressure_level': [pressure_level]
             })
             zone_pressure = pd.concat([zone_pressure, new_zone_data], ignore_index=True)
+            # Insert into the database
+            insert_data('zone_pressure', new_zone_data.to_dict(orient='records')[0])
+            # Re-fetch data to reflect the changes
+            zone_pressure = fetch_data('zone_pressure')
 
     # Display zone pressure data
     st.write(zone_pressure)
@@ -166,47 +178,19 @@ elif selected_page == "Analytics":
                 'customer_satisfaction': [customer_satisfaction]
             })
             analytics = pd.concat([analytics, new_analytics_data], ignore_index=True)
+            # Insert into the database
+            insert_data('analytics', new_analytics_data.to_dict(orient='records')[0])
+            # Re-fetch data to reflect the changes
+            analytics = fetch_data('analytics')
 
     # Display analytics information
     st.write(analytics)
 
-
 # Fetch initial data from the database
 drivers = fetch_data('drivers')
 orders = fetch_data('orders')
 zone_pressure = fetch_data('zone_pressure')
 analytics = fetch_data('analytics')
 
-# ... (within your form submission logic)
-
-        # Instead of appending to DataFrames, insert into the database
-        insert_data('orders', new_order.to_dict(orient='records')[0])
-
-        # Re-fetch data to reflect the changes
-        orders = fetch_data('orders')
-
-
-        # Instead of appending to DataFrames, insert into the database
-        insert_data('drivers', new_order.to_dict(orient='records')[0])
-
-        # Re-fetch data to reflect the changes
-        orders = fetch_data('drivers')
-
-
-        # Instead of appending to DataFrames, insert into the database
-        insert_data('zone_pressure', new_order.to_dict(orient='records')[0])
-
-        # Re-fetch data to reflect the changes
-        orders = fetch_data('zone_pressure')
-
-
-        # Instead of appending to DataFrames, insert into the database
-        insert_data('analytics', new_order.to_dict(orient='records')[0])
-
-        # Re-fetch data to reflect the changes
-        orders = fetch_data('analytics')
-
-# ... (similarly for other forms)
-
 # Close the database connection when the app is done
-st.session_state.on_session_end = conn.close
+st.session_state.on_session_end = conn.close