Update app.py

app.py

@@ -26,6 +26,11 @@ 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]
@@ -54,7 +59,7 @@ def generate_html_snippet(image_path_prefix):
     
     return html
 
-def process_image(input_image):
+def process_image(input_image, original_filename):
     """Main processing function"""
     if input_image is None:
         return None, None, "Please upload an image"
@@ -72,10 +77,13 @@ def process_image(input_image):
         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))
-            output_path = os.path.join(temp_dir, f"image-{width}x{height}.jpg")
+            output_path = os.path.join(temp_dir, f"{base_filename}-{width}x{height}.jpg")
             resized.save(output_path, "JPEG", quality=90)
             processed_paths.append(output_path)
         
@@ -84,8 +92,8 @@ def process_image(input_image):
             for path in processed_paths:
                 zf.write(path, os.path.basename(path))
         
-        # Generate HTML snippet
-        html_snippet = generate_html_snippet("images/image")
+        # Generate HTML snippet using original filename
+        html_snippet = generate_html_snippet(f"images/{base_filename}")
         
         return zip_path, html_snippet, "Processing completed successfully!"
         
@@ -125,7 +133,8 @@ with gr.Blocks(title="Responsive Image Generator") as app:
         with gr.Column():
             input_image = gr.Image(
                 label="Upload Original Image",
-                type="numpy"  # Changed to numpy type for better compatibility
+                type="numpy",  # Using numpy type for better compatibility
+                source="upload"  # Ensure we only accept uploads
             )
             process_btn = gr.Button("Process Image")
         
@@ -139,8 +148,9 @@ with gr.Blocks(title="Responsive Image Generator") as app:
     
     process_btn.click(
         fn=process_image,
-        inputs=[input_image],
+        inputs=[input_image, input_image],  # Pass the image twice to get both data and filename
         outputs=[output_zip, output_html, output_message]
     )
 
+# Launch the app with share=True for public access
 app.launch()