Update app.py

app.py

@@ -15,30 +15,27 @@ timeout = 100
 
 
 
-# Function to split the image into nine 341x341 sections (3x3 grid)
-def split_image(input_image, num_splits=9):
+def split_image(input_image, num_splits=4):
+    # Create a list to store the output images
     output_images = []
 
-    # Image size is 1024x1024, dividing into 3x3 grid
-    box_size = 341  # Size of each box in the grid
+    # Get image dimensions
+    img_width, img_height = input_image.size
 
-    # Define the coordinates for the 3x3 grid (left-to-right, top-to-bottom)
-    for i in range(3):  # Iterate rows
-        for j in range(3):  # Iterate columns
-            left = j * box_size
-            upper = i * box_size
-            right = left + box_size
-            lower = upper + box_size
+    # Calculate the width of each split
+    split_width = img_width // num_splits
 
-            # Crop the input image based on the calculated coordinates
-            box = (left, upper, right, lower)
-            output_images.append(input_image.crop(box))
+    # Split the image into equal sections horizontally
+    for i in range(num_splits):
+        left = i * split_width
+        right = (i + 1) * split_width
+        box = (left, 0, right, img_height)
+        output_images.append(input_image.crop(box))
 
     return output_images
 
-
 # Function to export split images to GIF
-def export_to_gif(images, output_path, fps=9):
+def export_to_gif(images, output_path, fps=4):
     # Calculate duration per frame in milliseconds based on fps
     duration = int(1000 / fps)
 
@@ -53,7 +50,7 @@ def export_to_gif(images, output_path, fps=9):
 
 
 def predict(prompt, seed=-1, randomize_seed=True, guidance_scale=3.5, num_inference_steps=28, lora_id="black-forest-labs/FLUX.1-dev", progress=gr.Progress(track_tqdm=True)):
-    prompt_template = f"""a 3x3 total 9 grid of frames, showing consecutive stills from a looped gif of {prompt}"""
+    prompt_template = f"""a 2x2 total 4 grid of frames, showing consecutive stills from a looped gif of {prompt}"""
 
     if lora_id.strip() == "" or lora_id == None:
         lora_id = "black-forest-labs/FLUX.1-dev"
@@ -94,7 +91,7 @@ def predict(prompt, seed=-1, randomize_seed=True, guidance_scale=3.5, num_infere
         gif_path = "output.gif"
     
         # Export the split images to GIF
-        export_to_gif(split_images, gif_path, fps=9)
+        export_to_gif(split_images, gif_path, fps=4)
         return gif_path, image, seed
     except Exception as e:
         print(f"Error when trying to open the image: {e}")