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_nutrition_in_city(city, target_calories, target_protein, max_cost):
     # Extracting cost, calories, and protein data for the selected city
     costs = [nutritional_data[dish][city] for dish in nutritional_data]
@@ -112,17 +128,27 @@ 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")
         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 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)
         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(optimize_button, optimization_output)
+        gr.Row(display_dishes_button, optimize_button)
+        gr.Row(dishes_output)
+        gr.Row(optimization_output)
     
     return demo