Update app.py

app.py

@@ -61,6 +61,7 @@ def process_image(input_image):
     # Create temporary directory for processed images
     temp_dir = tempfile.mkdtemp()
     zip_path = os.path.join(temp_dir, "responsive_images.zip")
+    processed_paths = []  # Initialize the list outside try block
     
     try:
         # Open and process image
@@ -71,7 +72,6 @@ def process_image(input_image):
             img = img.convert('RGB')
         
         # Process each size
-        processed_paths = []
         for width, height in SIZES:
             resized = resize_image(img, (width, height))
             output_path = os.path.join(temp_dir, f"image-{width}x{height}.jpg")
@@ -89,6 +89,11 @@ def process_image(input_image):
         return zip_path, html_snippet, "Processing completed successfully!"
         
     except Exception as e:
+        if os.path.exists(zip_path):
+            try:
+                os.remove(zip_path)
+            except:
+                pass
         return None, None, f"Error processing image: {str(e)}"
     
     finally:
@@ -98,7 +103,12 @@ def process_image(input_image):
                 os.remove(path)
             except:
                 pass
-
+        # Try to remove the temp directory if it's empty
+        try:
+            os.rmdir(temp_dir)
+        except:
+            pass
+            
 # Create Gradio interface
 with gr.Blocks(title="Responsive Image Generator") as app:
     gr.Markdown("""