Update app.py

app.py

@@ -6,6 +6,7 @@ import tempfile
 import zipfile
 import shutil
 from pathlib import Path
+import numpy as np
 
 # Define target sizes for different viewports
 SIZES = [
@@ -64,8 +65,8 @@ def process_image(input_image):
     processed_paths = []  # Initialize the list outside try block
     
     try:
-        # Open the image directly since Gradio provides the image path
-        img = Image.open(input_image)
+        # Convert numpy array to PIL Image
+        img = Image.fromarray(input_image)
         
         # Convert to RGB if necessary
         if img.mode != 'RGB':
@@ -108,7 +109,7 @@ def process_image(input_image):
             os.rmdir(temp_dir)
         except:
             pass
-            
+
 # Create Gradio interface
 with gr.Blocks(title="Responsive Image Generator") as app:
     gr.Markdown("""
@@ -124,7 +125,7 @@ with gr.Blocks(title="Responsive Image Generator") as app:
         with gr.Column():
             input_image = gr.Image(
                 label="Upload Original Image",
-                type="filepath"
+                type="numpy"  # Changed to numpy type for better compatibility
             )
             process_btn = gr.Button("Process Image")
         
@@ -142,5 +143,4 @@ with gr.Blocks(title="Responsive Image Generator") as app:
         outputs=[output_zip, output_html, output_message]
     )
 
-# Launch the app
 app.launch()