Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from components.menu import generate_menu
 from components.popup import show_popup
-from components.cart import display_cart, cart_data
+from components.cart import add_to_cart, cart_data
 import pandas as pd
 
 # Load menu data
@@ -28,10 +28,19 @@ def main():
         # Cart Section
         gr.Markdown("## Your Cart")
         cart_display = gr.Dataframe(
-            value=cart_data(),
+            value=pd.DataFrame(cart_data(), columns=["Dish", "Spice Level", "Extras", "Instructions", "Quantity", "Price"]),
             headers=["Dish", "Spice Level", "Extras", "Instructions", "Quantity", "Price"]
         )
 
+        # Keep updating the cart display
+        def update_cart_display():
+            return pd.DataFrame(cart_data(), columns=["Dish", "Spice Level", "Extras", "Instructions", "Quantity", "Price"])
+
+        # Update cart when new item is added
+        gr.Button("Update Cart").click(
+            update_cart_display, inputs=[], outputs=cart_display
+        )
+
         app.launch()
 
 if __name__ == "__main__":