Update app.py

app.py

@@ -16,8 +16,9 @@ def augment_images(image_file, num_duplicates):
         horizontal_flip=True,
         fill_mode='nearest')
     
-    img = Image.open(image_file).convert('RGB')  # Convert to RGB
-    img = img.resize((256, 256))  # Resize image
+    # Convert to RGB and resize the image
+    img = Image.open(image_file).convert('RGB')
+    img = img.resize((256, 256))
     x = img_to_array(img)  # Convert image to numpy array
     x = x.reshape((1,) + x.shape)  # Reshape for data generator
     
@@ -29,7 +30,7 @@ def augment_images(image_file, num_duplicates):
                 break
 
         # Zip the augmented images
-        zip_name = tempfile.mktemp(suffix='.zip')
+        zip_name = os.path.join(tempfile.gettempdir(), 'augmented_images.zip')
         with zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED) as zipf:
             for root, _, files in os.walk(temp_dir):
                 for file in files:
@@ -39,7 +40,10 @@ def augment_images(image_file, num_duplicates):
 
 iface = gr.Interface(
     fn=augment_images,
-    inputs=[gr.Image(label="Upload Image", source="upload"), gr.Slider(minimum=1, maximum=20, default=5, label="Number of Augmented Samples")],
+    inputs=[
+        gr.Image(tool="editor", label="Upload Image"),
+        gr.Slider(minimum=1, maximum=20, default=5, label="Number of Augmented Samples")
+    ],
     outputs=gr.File(label="Download Augmented Images"),
     title="Image Augmentation App",
     description="Upload an image to generate augmented versions. Select the number of augmented duplicates you want for the image."