Update app.py

app.py

@@ -54,7 +54,14 @@ def generate_html_snippet(image_path_prefix):
     
     return html
 
-def process_image(input_img, input_name):
+def get_path_components(filepath):
+    """Split filepath into directory and filename"""
+    dirpath = os.path.dirname(filepath)
+    filename = os.path.basename(filepath)
+    base_filename = os.path.splitext(filename)[0]
+    return dirpath, base_filename
+
+def process_image(input_img, input_path):
     """Main processing function"""
     if input_img is None:
         return None, None, "Please upload an image"
@@ -65,9 +72,13 @@ def process_image(input_img, input_name):
     processed_paths = []  # Initialize the list outside try block
     
     try:
-        # Get original filename without extension
-        base_filename = os.path.splitext(input_name)[0] if input_name else "image"
-        
+        # Get path components
+        if input_path and input_path.strip():
+            dirpath, base_filename = get_path_components(input_path.strip())
+        else:
+            # If no path provided, use default
+            dirpath, base_filename = "", "image"
+            
         # Convert to PIL Image if needed
         if isinstance(input_img, np.ndarray):
             img = Image.fromarray(input_img)
@@ -80,8 +91,9 @@ def process_image(input_img, input_name):
         
         # Process each size
         for width, height in SIZES:
-            resized = resize_image(img, (width, height))
+            # For zip file, only use filename without path
             output_path = os.path.join(temp_dir, f"{base_filename}-{width}x{height}.jpg")
+            resized = resize_image(img, (width, height))
             resized.save(output_path, "JPEG", quality=90)
             processed_paths.append(output_path)
         
@@ -90,8 +102,9 @@ def process_image(input_img, input_name):
             for path in processed_paths:
                 zf.write(path, os.path.basename(path))
         
-        # Generate HTML snippet using original filename
-        html_snippet = generate_html_snippet(f"images/{base_filename}")
+        # Generate HTML snippet using full path
+        full_path = dirpath + ("/" if dirpath else "") + base_filename
+        html_snippet = generate_html_snippet(full_path)
         
         return zip_path, html_snippet, "Processing completed successfully!"
         
@@ -125,6 +138,8 @@ with gr.Blocks(title="Responsive Image Generator") as app:
     You'll receive:
     1. A ZIP file containing all sized versions
     2. HTML code snippet with proper srcset attributes
+    
+    Optional: Specify the full path where images will be stored (e.g., 'assets/images/about/mock-up.png')
     """)
     
     with gr.Row():
@@ -136,9 +151,10 @@ with gr.Blocks(title="Responsive Image Generator") as app:
                     show_label=True
                 )
             with gr.Row():
-                input_name = gr.Textbox(
-                    label="Original Filename (optional)",
-                    placeholder="Enter filename or leave empty to use uploaded filename"
+                input_path = gr.Textbox(
+                    label="Image Path (optional)",
+                    placeholder="e.g., assets/images/about/mock-up.png",
+                    value=""
                 )
             process_btn = gr.Button("Process Image")
         
@@ -160,14 +176,15 @@ with gr.Blocks(title="Responsive Image Generator") as app:
     input_image.upload(
         fn=handle_upload,
         inputs=[input_image],
-        outputs=[input_name]
+        outputs=[input_path]
     )
     
     # Process button click
     process_btn.click(
         fn=process_image,
-        inputs=[input_image, input_name],
+        inputs=[input_image, input_path],
         outputs=[output_zip, output_html, output_message]
     )
 
-app.launch()
+# Launch the app with share=True for public access
+app.launch(share=True)