Fix broken gifs with default

app.py

@@ -10,33 +10,54 @@ global stored_frames
 def load_and_store_frames(image_file, grid_x, grid_y):
     global stored_frames
 
-    # Make sure file exists
-    if image_file is None:
-        return "File not found", ""
-
-    print("Loading frames for {}".format(image_file.name))
-
-    if image_file.name.endswith('.mp4'):
-        frames = extract_frames_from_video(image_file.name)
-        video_path = image_file.name
-    else:  # it's a gif
-        img = Image.open(image_file.name)
-        frames = []
-        for i in range(0, img.n_frames):
-            img.seek(i)
-            frames.append(img.copy())
-        # Convert GIF to MP4 for preview
-        fourcc = cv2.VideoWriter_fourcc(*'mp4v') 
-        tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-        video_path = tmp_file.name
-        convert_gif_to_video(image_file.name, tmp_file.name, 1 / (img.info.get('duration', 100) / 1000.0))
-
-    stored_frames = frames  # Store the frames for later use
-    total_frames = len(frames)
-    selected_frames_count = grid_x * grid_y
-    details = f"**Total Frames:** {len(frames)}\n\n"
-    output_info = f"Grid size: {grid_x} x {grid_y}\n\nSelected Frames: {selected_frames_count} / {total_frames} ({selected_frames_count / total_frames * 100:.2f}%)"
-    return f"Frames loaded successfully\n\n{details}\n\n{output_info}", video_path
+    try:
+        # Make sure file exists
+        if image_file is None:
+            return "File not found", ""
+
+        print(f"Loading frames for {image_file.name}")
+
+        if image_file.name.endswith('.mp4'):
+            frames = extract_frames_from_video(image_file.name)
+            video_path = image_file.name
+        else:  # it's a gif
+            try:
+                img = Image.open(image_file.name)
+            except Exception as e:
+                print(f"Could not open GIF file: {e}")
+                return "Could not open GIF file", ""
+
+            frames = []
+            for i in range(0, img.n_frames):
+                try:
+                    img.seek(i)
+                    frames.append(img.copy())
+                except Exception as e:
+                    print(f"Could not seek to frame {i}: {e}")
+
+            # Convert GIF to MP4 for preview
+            fourcc = cv2.VideoWriter_fourcc(*'mp4v') 
+            tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+            video_path = tmp_file.name
+
+            try:
+                duration = img.info.get('duration', 100)
+                # default to reasonable framerate if duration is 0
+                framerate = 1 / (duration / 1000.0) if duration > 0 else 10
+                print(f"frame count: {len(frames)} framerate: {duration} {img.info}")
+                convert_gif_to_video(image_file.name, tmp_file.name, framerate)
+            except Exception as e:
+                print(f"Could not convert GIF to MP4: {e}")
+
+        stored_frames = frames  # Store the frames for later use
+        total_frames = len(frames)
+        selected_frames_count = grid_x * grid_y
+        details = f"**Total Frames:** {len(frames)}\n\n"
+        output_info = f"Grid size: {grid_x} x {grid_y}\n\nSelected Frames: {selected_frames_count} / {total_frames} ({selected_frames_count / total_frames * 100:.2f}%)"
+        return f"Frames loaded successfully\n\n{details}\n\n{output_info}", video_path
+    except Exception as e:
+        print(f"An error occurred while loading and storing frames: {e}")
+        return f"An error occurred: {e}", ""
 
 def generate_grid(grid_x, grid_y, font_size, font_color, position, border_size, border_color):
     global stored_frames
@@ -115,21 +136,33 @@ def extract_frames_from_video(video_file):
 
 
 def convert_gif_to_video(gif_path, output_video_path, frame_rate):
-    # Load the gif
-    gif = Image.open(gif_path)
-    # Define the codec and create VideoWriter object
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    out = cv2.VideoWriter(output_video_path, fourcc, frame_rate, (gif.width, gif.height))
-    
-    # Iterate over the frames of the gif
-    for frame_index in range(gif.n_frames):
-        gif.seek(frame_index)
-        # Convert the PIL Image to an array
-        frame_arr = np.array(gif.convert("RGB"))
-        # Convert RGB to BGR format
-        frame_bgr = cv2.cvtColor(frame_arr, cv2.COLOR_RGB2BGR)
-        # Write the frame to the video
-        out.write(frame_bgr)
+    try:
+        # Load the gif
+        gif = Image.open(gif_path)
+    except Exception as e:
+        print(f"Could not open GIF file: {e}")
+        return
+
+    try:
+        # Define the codec and create VideoWriter object
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        out = cv2.VideoWriter(output_video_path, fourcc, frame_rate, (gif.width, gif.height))
+    except Exception as e:
+        print(f"Could not create VideoWriter object: {e}")
+        return
+
+    try:
+        # Iterate over the frames of the gif
+        for frame_index in range(gif.n_frames):
+            gif.seek(frame_index)
+            # Convert the PIL Image to an array
+            frame_arr = np.array(gif.convert("RGB"))
+            # Convert RGB to BGR format
+            frame_bgr = cv2.cvtColor(frame_arr, cv2.COLOR_RGB2BGR)
+            # Write the frame to the video
+            out.write(frame_bgr)
+    except Exception as e:
+        print(f"Could not write frame to video: {e}")
     
     out.release()