Update app.py

app.py

@@ -1,59 +1,55 @@
 import gradio as gr
 
-def display_portion_image(dish, portion):
-    # Define a mapping of images for each portion size
-    images = {
-        "Pizza": {
-            "Small": "https://example.com/pizza_small.jpg",
-            "Medium": "https://example.com/pizza_medium.jpg",
-            "Large": "https://example.com/pizza_large.jpg",
-        },
-        "Burger": {
-            "Small": "https://example.com/burger_small.jpg",
-            "Medium": "https://example.com/burger_medium.jpg",
-            "Large": "https://example.com/burger_large.jpg",
-        },
-    }
-
-    # Fetch the corresponding image based on dish and portion size
-    if dish in images and portion in images[dish]:
-        image_url = images[dish][portion]
-    else:
-        image_url = "https://example.com/default_image.jpg"  # Default image if no match found
-
-    return image_url
-
-# Define the interface
-def main():
-    with gr.Blocks() as app:
-        gr.Markdown("""
-        ### Food Selection with Portion Sizes
-        Select a dish and a portion size to view the corresponding image.
-        """)
-
-        dish = gr.Dropdown(
-            label="Select Dish",
-            choices=["Pizza", "Burger"],
-            value="Pizza"
-        )
-
-        portion = gr.Radio(
-            label="Select Portion Size",
-            choices=["Small", "Medium", "Large"],
-            value="Medium"
-        )
-
-        image_output = gr.Image(type="filepath", label="Dish Image")
-
-
-        # Link inputs and outputs
-        gr.Button("Show Image").click(
-            display_portion_image,
-            inputs=[dish, portion],
-            outputs=image_output
-        )
-
-    app.launch()
-
-if __name__ == "__main__":
-    main()
+# Dictionary to store images for each dish and portion size
+dish_data = {
+    "Pizza": {
+        "main": "images/pizza_main.jpg",
+        "small": "images/pizza_small.jpg",
+        "medium": "images/pizza_medium.jpg",
+        "large": "images/pizza_large.jpg",
+    },
+    "Burger": {
+        "main": "images/burger_main.jpg",
+        "small": "images/burger_small.jpg",
+        "medium": "images/burger_medium.jpg",
+        "large": "images/burger_large.jpg",
+    },
+    "Pasta": {
+        "main": "images/pasta_main.jpg",
+        "small": "images/pasta_small.jpg",
+        "medium": "images/pasta_medium.jpg",
+        "large": "images/pasta_large.jpg",
+    },
+}
+
+# Function to display the main image and portion size image
+def display_images(dish, portion):
+    if dish in dish_data:
+        main_image = dish_data[dish]["main"]
+        portion_image = dish_data[dish].get(portion, main_image)  # Default to main image if portion not found
+        return main_image, portion_image
+    return "", ""
+
+# Gradio interface
+dish_list = list(dish_data.keys())
+portion_sizes = ["small", "medium", "large"]
+
+with gr.Blocks() as demo:
+    gr.Markdown("### Select a Dish and Portion Size")
+
+    with gr.Row():
+        dish_dropdown = gr.Dropdown(choices=dish_list, label="Select Dish")
+        portion_dropdown = gr.Dropdown(choices=portion_sizes, label="Select Portion Size")
+
+    with gr.Row():
+        main_image_display = gr.Image(label="Main Image")
+        portion_image_display = gr.Image(label="Portion Image")
+
+    # Button to trigger image display
+    submit_button = gr.Button("Show Images")
+
+    # Update images based on selections
+    submit_button.click(display_images, inputs=[dish_dropdown, portion_dropdown], outputs=[main_image_display, portion_image_display])
+
+# Launch the application
+demo.launch()