Update app.py

app.py

@@ -4,20 +4,29 @@ import zipfile
 from PIL import Image
 import tempfile
 
-def flip_image(image):
-    image = Image.fromarray(np.uint8(image))
-    # Create a temporary file
-    with tempfile.NamedTemporaryFile(suffix='.jpg') as temp:
-        # Save the image to the temporary file
-        image.save(temp.name)
-        # Create zip file
-        with zipfile.ZipFile("image.zip", 'w') as zip:
-            zip.write(temp.name)
+def split_image(image):
+    im = Image.fromarray(np.uint8(image))
+
+    # Obtiene las dimensiones de la imagen
+    width, height = im.size
+
+    # Divide la imagen en 9 imágenes más pequeñas
+    for i in range(3):
+        for j in range(3):
+            box = (j*width/3, i*height/3, (j+1)*width/3, (i+1)*height/3)
+            region = im.crop(box)
+            region = region.convert("RGB")
+            region = region.resize((512, 512), Image.ANTIALIAS)
+
+            with tempfile.NamedTemporaryFile(suffix='.jpg') as temp:
+                region.save(temp.name)
+                with zipfile.ZipFile("image.zip", 'w') as zip:
+                    zip.write(temp.name)
     return "image.zip"
-    
-gr.Interface(fn=flip_image,
+
+gr.Interface(fn=split_image,
              inputs="image",
              outputs="file",
-             title="Flip An Image Upside Down",
+             title="Convert collage to images",
              examples = ["example.jpg"]  
             ).launch();