Update app.py

app.py

@@ -56,6 +56,22 @@ nutritional_data = {
     }
 }
 
+def display_dishes_in_city(city):
+    """Displays all dishes available in the selected city with their nutritional information and cost."""
+    result_str = f"### Available Dishes in {city}:\n"
+    for dish, info in nutritional_data.items():
+        result_str += f"- **{dish}**\n"
+        result_str += f"  - Cost: ₹{info[city]}\n"
+        result_str += f"  - Energy: {info['Energy (kcal)']} kcal\n"
+        result_str += f"  - Protein: {info['Protein (g)']} g\n"
+        result_str += f"  - Fat: {info['Fat (g)']} g\n"
+        result_str += f"  - Carbohydrate: {info['Carbohydrate (g)']} g\n"
+        result_str += f"  - Fiber: {info['Fiber (g)']} g\n"
+        result_str += f"  - Calcium: {info['Calcium (mg)']} mg\n"
+        result_str += f"  - Iron: {info['Iron (mg)']} mg\n"
+        result_str += f"  - Vitamin C: {info['Vitamin C (mg)']} mg\n\n"
+    return result_str
+
 def optimize_dishes_for_budget(city, daily_budget):
     # Extracting cost, calories, and protein data for the selected city
     costs = [nutritional_data[dish][city] for dish in nutritional_data]
@@ -106,18 +122,28 @@ def create_interface():
         city_selector = gr.Dropdown(choices=cities, label="Select City")
         budget_input = gr.Number(label="Daily Budget (₹)", value=300)
         
+        show_all_dishes_button = gr.Button("Show All Available Dishes")
         optimize_button = gr.Button("Optimize Nutrition")
+        
+        all_dishes_output = gr.Markdown(label="All Available Dishes")
         optimization_output = gr.Markdown(label="Optimization Results")
         
+        # Function to handle showing all dishes
+        def show_all_dishes(city):
+            return display_dishes_in_city(city)
+        
         # Function to handle optimization
         def run_optimization(city, daily_budget):
             return optimize_dishes_for_budget(city, daily_budget)
         
+        show_all_dishes_button.click(fn=show_all_dishes, inputs=[city_selector], outputs=all_dishes_output)
         optimize_button.click(fn=run_optimization, inputs=[city_selector, budget_input], outputs=optimization_output)
         
         gr.Row([city_selector, budget_input])
         gr.Row(optimize_button)
         gr.Row(optimization_output)
+        gr.Row(show_all_dishes_button)
+        gr.Row(all_dishes_output)
     
     return demo