Update app.py

app.py

@@ -1,13 +1,32 @@
 import gradio as gr
 import os
 
-# Function to serve the HTML file
-def serve_html():
-    with open("index.html", "r") as file:
-        return file.read()
+# Function to handle image upload and layout setting
+def process_images(layout, image_files):
+    # Save uploaded images
+    img_paths = []
+    for img_file in image_files:
+        img_path = f"temp_images/{img_file.name}"
+        img_file.save(img_path)
+        img_paths.append(img_path)
+    
+    # Return the layout and image paths to the front-end (HTML and JS will take care of it)
+    return layout, img_paths
 
-# Create a Gradio Interface
-iface = gr.Interface(fn=serve_html, inputs=None, outputs="html", live=True)
+# Gradio Interface setup
+iface = gr.Interface(
+    fn=process_images, 
+    inputs=[
+        gr.Dropdown(
+            label="Choose Grid Layout", 
+            choices=["1x2", "1x3", "1x4", "2x2", "2x3", "random", "jagged4", "split6", "chaos5"],
+            default="3x3"
+        ),
+        gr.File(label="Upload Images", type="file", multiple=True)  # Multiple image upload
+    ], 
+    outputs=["text", "json"],  # We'll output a layout and list of image paths
+    live=True  # Updates in real-time
+)
 
-# Launch the Gradio app
-iface.launch()
+# Run Gradio app
+iface.launch(share=True)