Update app.py

app.py

@@ -128,25 +128,34 @@ def create_interface():
         protein_target = gr.Number(label="Target Protein (g)", value=25)
         cost_limit = gr.Number(label="Maximum Cost (₹)", value=300)
         
-        display_dishes_button = gr.Button("Display Available Dishes")
+        toggle_dishes_button = gr.Button("Display Dishes")
         optimize_button = gr.Button("Optimize")
         
         dishes_output = gr.Markdown(label="Available Dishes")
         optimization_output = gr.Markdown(label="Optimization Results")
         
-        # Function to handle displaying dishes
-        def display_dishes(city):
-            return display_dishes_in_city(city)
+        # Function to toggle the display of dishes
+        dishes_visible = False
+        
+        def toggle_dishes(city):
+            nonlocal dishes_visible
+            dishes_visible = not dishes_visible
+            if dishes_visible:
+                toggle_dishes_button.label = "Hide Dishes"
+                return display_dishes_in_city(city)
+            else:
+                toggle_dishes_button.label = "Display Dishes"
+                return ""
         
         # Function to handle optimization
         def run_optimization(city, target_calories, target_protein, max_cost):
             return optimize_nutrition_in_city(city, target_calories, target_protein, max_cost)
         
-        display_dishes_button.click(fn=display_dishes, inputs=[city_selector], outputs=dishes_output)
+        toggle_dishes_button.click(fn=toggle_dishes, inputs=[city_selector], outputs=dishes_output)
         optimize_button.click(fn=run_optimization, inputs=[city_selector, energy_target, protein_target, cost_limit], outputs=optimization_output)
         
         gr.Row([city_selector, energy_target, protein_target, cost_limit])
-        gr.Row(display_dishes_button, optimize_button)
+        gr.Row(toggle_dishes_button, optimize_button)
         gr.Row(dishes_output)
         gr.Row(optimization_output)