Update app.py

app.py

@@ -54,7 +54,7 @@ def generate_html_snippet(image_path_prefix):
     
     return html
 
-def process_image(input_img):
+def process_image(input_img, input_name):
     """Main processing function"""
     if input_img is None:
         return None, None, "Please upload an image"
@@ -65,23 +65,14 @@ def process_image(input_img):
     processed_paths = []  # Initialize the list outside try block
     
     try:
-        # 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 original filename without extension
+        base_filename = os.path.splitext(input_name)[0] if input_name else "image"
         
-        # 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)
+        if isinstance(input_img, np.ndarray):
+            img = Image.fromarray(input_img)
         else:
-            img = img_data
+            img = input_img
             
         # Convert to RGB if necessary
         if img.mode != 'RGB':
@@ -138,10 +129,17 @@ with gr.Blocks(title="Responsive Image Generator") as app:
     
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(
-                label="Upload Original Image",
-                type="pil"  # Changed to PIL type
-            )
+            with gr.Row():
+                input_image = gr.Image(
+                    label="Upload Original Image",
+                    type="pil",
+                    show_label=True
+                )
+            with gr.Row():
+                input_name = gr.Textbox(
+                    label="Original Filename (optional)",
+                    placeholder="Enter filename or leave empty to use uploaded filename"
+                )
             process_btn = gr.Button("Process Image")
         
         with gr.Column():
@@ -152,9 +150,23 @@ with gr.Blocks(title="Responsive Image Generator") as app:
             )
             output_message = gr.Textbox(label="Status")
     
+    def handle_upload(img):
+        """Handle image upload to get filename"""
+        if isinstance(img, dict) and 'name' in img:
+            return img['name']
+        return ""
+    
+    # Update filename when image is uploaded
+    input_image.upload(
+        fn=handle_upload,
+        inputs=[input_image],
+        outputs=[input_name]
+    )
+    
+    # Process button click
     process_btn.click(
         fn=process_image,
-        inputs=[input_image],  # Now only passing single input
+        inputs=[input_image, input_name],
         outputs=[output_zip, output_html, output_message]
     )