Update app.py

app.py

@@ -7,30 +7,34 @@ Created on Tue Sep 17 15:03:39 2024
 
 import gradio as gr
 import os
+from PIL import Image
 
-# Paths to the images folder (use relative paths)
-RAW_PATH = "/images/raw"
-EMBEDDINGS_PATH = "/images/embeddings"
+# Paths to the images folder
+RAW_PATH = os.path.join("images", "raw")
+EMBEDDINGS_PATH = os.path.join("images", "embeddings")
 
-# Function to display images based on user selection
+# Function to load and display images based on user selection
 def display_images(percentage, complexity):
-    # Format the filenames based on user selections
-    raw_image_path = f"{RAW_PATH}/percentage_{percentage}_complexity_{complexity}.png"
-    embeddings_image_path = f"{EMBEDDINGS_PATH}/percentage_{percentage}_complexity_{complexity}.png"
+    # Generate the paths to the images
+    raw_image_path = os.path.join(RAW_PATH, f"percentage_{percentage}_complexity_{complexity}.png")
+    embeddings_image_path = os.path.join(EMBEDDINGS_PATH, f"percentage_{percentage}_complexity_{complexity}.png")
     
-    # Return the corresponding images for raw and embeddings
-    return raw_image_path, embeddings_image_path
+    # Load images using PIL
+    raw_image = Image.open(raw_image_path)
+    embeddings_image = Image.open(embeddings_image_path)
+    
+    # Return the loaded images
+    return raw_image, embeddings_image
 
 # Define the Gradio interface
-# Step 2: Pass arrays of values for data percentage and task complexity
 data_percentage_options = [10, 30, 50, 70, 100]
 task_complexity_options = [16, 32]
 
 demo = gr.Interface(
     fn=display_images,
     inputs=[
-        gr.Dropdown(data_percentage_options, label="Percentage of Data for Training"),  # Dropdown for data percentage
-        gr.Radio(task_complexity_options, label="Task Complexity")  # Radio for task complexity
+        gr.Dropdown(data_percentage_options, label="Percentage of Data for Training"),
+        gr.Radio(task_complexity_options, label="Task Complexity")
     ],
     outputs=[
         gr.Image(label="Raw Channels"),
@@ -41,4 +45,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()