Update app.py

app.py

@@ -1,79 +1,79 @@
-from moviepy.editor import *
-
 from PIL import Image
-
+from moviepy.editor import *
 import tempfile
 import gradio as gr
 
 def get_video_detail(video_path):
-  info = VideoFileClip(video_path)
-  return 0, info.duration, info.fps, max(info.size)
+    info = VideoFileClip(video_path)
+    return 0, info.duration, info.fps, max(info.size)
 
 def convert_video_to_gif(
-  input: str,
-  start: float = 0,
-  end: float | None = None,
-  fps: int | None = None,
-  quality: int | None = None
+    input: str,
+    start: float = 0,
+    end: float | None = None,
+    fps: int | None = None,
+    quality: int | None = None
 ):
-  """
-  Convert video to GIF image
+    """
+    Convert video to GIF image
+
+    Args:
+        input (str): Path to video file.
+        start (float, optional): Start time in seconds. Defaults to 0.
+        end (float, optional): End time in seconds. Defaults to None.
+        fps (int, optional): Frames per second. Defaults to None (max fps).
+        quality (int, optional): Image quality. Defaults to None (max quality).
 
-  Args:
-      input (str): Path to video file.
-      start (float, optional): Start time in seconds. Defaults to 0.
-      end (float, optional): End time in seconds. Defaults to None.
-      fps (int, optional): Frames per second. Defaults to None (max fps).
-      quality (int, optional): Image quality. Defaults to None (max quality).
+    Returns:
+        str: Path to the GIF file
 
-  Returns:
-      Image.Image: GIF image
+    Examples:
+        >>> convert_video_to_gif("input.mp4")
 
-  Examples:
-      >>> convert_video_to_gif("input.mp4")
-      
-  """
-  # Get video input & info
-  clip = VideoFileClip(input)
+    """
+    # Get video input & info
+    clip = VideoFileClip(input)
 
-  max_fps = clip.fps
-  max_duration = clip.duration
-  max_res = max(clip.size)
+    max_fps = clip.fps
+    max_duration = clip.duration
+    max_res = max(clip.size)
 
-  if end is None:
-    end = max_duration
- 
-  if end > max_duration:
-    raise ValueError(f'End time {end} is longer than video duration {max_duration}')
+    if end is None:
+        end = max_duration
 
-  if fps > max_fps:
-    raise ValueError(f'FPS {fps} is greater than video FPS {max_fps}')
+    if end > max_duration:
+        raise ValueError(f"End time {end} is longer than video duration {max_duration}")
 
-  if quality is None:
-    quality = max_res
+    if fps > max_fps:
+        raise ValueError(f"FPS {fps} is greater than video FPS {max_fps}")
 
-  if quality > max_res:
-    raise ValueError(f'Quality must be less than video max resolution {max_res}, but is {quality}')
+    if quality is None:
+        quality = max_res
 
+    if quality > max_res:
+        raise ValueError(
+            f"Quality must be less than video max resolution {max_res}, but is {quality}"
+        )
 
-  clip = clip.subclip(start, end)
-  target_height = quality
-  aspect_ratio = clip.size[0] / clip.size[1]
-  target_width = int(target_height * aspect_ratio)
-  clip = clip.resize((target_width, target_height))
+    clip = clip.subclip(start, end)
+    target_height = quality
+    aspect_ratio = clip.size[0] / clip.size[1]
+    target_width = int(target_height * aspect_ratio)
+    clip = clip.resize((target_width, target_height))
 
-  clip = clip.set_fps(fps)
+    clip = clip.set_fps(fps)
 
-  # Create a temporary file
-  with tempfile.NamedTemporaryFile(suffix='.gif') as temp_file:
-    # Write the GIF to the temporary file
-    clip.write_gif(temp_file.name)
+    # Create a temporary file
+    with tempfile.NamedTemporaryFile(suffix=".gif") as temp_file:
+        # Write the GIF to the temporary file
+        clip.write_gif(temp_file.name)
 
-    # Read the GIF from the temporary file
-    gif_image = Image.open(temp_file.name)
+        # Return the path to the GIF file
+        gif_file_path = temp_file.name
 
-  return gif_image
+    return gif_file_path
 
+# Gradio code...
 
 with gr.Blocks(
   theme = gr.themes.Base(
@@ -108,10 +108,8 @@ with gr.Blocks(
     )
 
   run_btn = gr.Button("Convert")
-  output = gr.Image(
-    label="GIF",
-    format="gif",
-    type="pil"
+  output = gr.File(
+    label="GIF"
   )
   
   input.change(