Update app.py

app.py

@@ -26,11 +26,6 @@ def resize_image(img, target_size):
     img_copy = img.copy()
     return img_copy.resize(target_size, PIL.Image.Resampling.LANCZOS)
 
-def get_base_filename(original_path):
-    """Extract original filename without extension"""
-    base_name = os.path.basename(original_path)
-    return os.path.splitext(base_name)[0]
-
 def generate_html_snippet(image_path_prefix):
     """Generate HTML code snippet with srcset"""
     srcset_items = [f"{image_path_prefix}-{w}x{h}.jpg {w}w" for w, h in SIZES]
@@ -59,9 +54,9 @@ def generate_html_snippet(image_path_prefix):
     
     return html
 
-def process_image(input_image, original_filename):
+def process_image(input_img):
     """Main processing function"""
-    if input_image is None:
+    if input_img is None:
         return None, None, "Please upload an image"
     
     # Create temporary directory for processed images
@@ -70,16 +65,28 @@ def process_image(input_image, original_filename):
     processed_paths = []  # Initialize the list outside try block
     
     try:
-        # Convert numpy array to PIL Image
-        img = Image.fromarray(input_image)
+        # Get original filename
+        base_filename = "image"  # Default filename
+        if isinstance(input_img, dict) and "name" in input_img:
+            original_name = input_img["name"]
+            base_filename = os.path.splitext(original_name)[0]
         
+        # Get image data
+        if isinstance(input_img, dict) and "image" in input_img:
+            img_data = input_img["image"]
+        else:
+            img_data = input_img
+            
+        # Convert to PIL Image if needed
+        if isinstance(img_data, np.ndarray):
+            img = Image.fromarray(img_data)
+        else:
+            img = img_data
+            
         # Convert to RGB if necessary
         if img.mode != 'RGB':
             img = img.convert('RGB')
         
-        # Get base filename without extension
-        base_filename = os.path.splitext(original_filename)[0]
-        
         # Process each size
         for width, height in SIZES:
             resized = resize_image(img, (width, height))
@@ -133,7 +140,7 @@ with gr.Blocks(title="Responsive Image Generator") as app:
         with gr.Column():
             input_image = gr.Image(
                 label="Upload Original Image",
-                type="numpy"  # Using numpy type for better compatibility
+                type="pil"  # Changed to PIL type
             )
             process_btn = gr.Button("Process Image")
         
@@ -147,7 +154,7 @@ with gr.Blocks(title="Responsive Image Generator") as app:
     
     process_btn.click(
         fn=process_image,
-        inputs=[input_image, input_image],  # Pass the image twice to get both data and filename
+        inputs=[input_image],  # Now only passing single input
         outputs=[output_zip, output_html, output_message]
     )