Update app.py

app.py

@@ -1,99 +1,112 @@
 import gradio as gr
+from moviepy.editor import VideoFileClip
 import os
-import sys
-
-# Ensure moviepy is properly imported
-try:
-    sys.path.append("/usr/local/lib/python3.10/site-packages")
-    import moviepy.editor as mp
-except ModuleNotFoundError:
-    print("moviepy 패키지가 설치되어 있지 않습니다. 설치를 진행 중입니다...")
-    os.system(f"{sys.executable} -m pip install moviepy")
-    sys.path.append("/usr/local/lib/python3.10/site-packages")
-    try:
-        import moviepy.editor as mp
-    except ModuleNotFoundError:
-        print("moviepy 모듈을 로드할 수 없습니다. 설치 확인 후 다시 시도하세요.")
-        sys.exit(1)
-
-def process_video(video, start_time, end_time, resolution, fps, speed, loop):
-    try:
-        # Load video
-        clip = mp.VideoFileClip(video.name)
-
-        # Get video info
-        duration = clip.duration
-        width, height = clip.size
-        info = {
-            "duration": duration,
-            "resolution": f"{width}x{height}"
-        }
-
-        # Cut clip based on input times
-        start_time = float(start_time) if start_time else 0
-        end_time = float(end_time) if end_time else duration
-        clip = clip.subclip(start_time, end_time)
-
-        # Create thumbnails
-        start_thumbnail = clip.get_frame(start_time) if start_time < duration else None
-        end_thumbnail = clip.get_frame(end_time) if end_time <= duration else None
-
-        # Adjust resolution, fps, speed, and loop count
-        if resolution:
-            width, height = map(int, resolution.split('x'))
-            clip = clip.resize((width, height))
-
-        if fps:
-            clip = clip.set_fps(int(fps))
-
-        if speed:
-            clip = clip.fx(mp.vfx.speedx, float(speed))
-
-        gif_path = f"output.gif"
-        clip.write_gif(gif_path, loop=int(loop))
-
-        return start_thumbnail, end_thumbnail, gif_path, info
-    except Exception as e:
-        return None, None, None, {"error": str(e)}
-
-def main():
-    with gr.Blocks() as app:
-        gr.Markdown("## 동영상을 GIF로 변환하는 도구")
-
-        with gr.Row():
-            video_input = gr.Video(label="동영상 업로드")
-            video_info = gr.JSON(label="영상 정보")
-
-        with gr.Row():
-            start_time = gr.Textbox(label="시작 시간 (초)", placeholder="예: 0")
-            end_time = gr.Textbox(label="끝 시간 (초)", placeholder="예: 10")
-
-        with gr.Row():
-            resolution = gr.Textbox(label="해상도 (예: 640x480)", placeholder="예: 640x480")
-            fps = gr.Slider(label="프레임 속도 (FPS)", minimum=1, maximum=60, step=1, value=24)
-            speed = gr.Slider(label="재생 속도 (배속)", minimum=0.1, maximum=3.0, step=0.1, value=1.0)
-            loop = gr.Slider(label="GIF 반복 횟수", minimum=0, maximum=10, step=1, value=0)
-
-        with gr.Row():
-            start_thumbnail = gr.Image(label="시작 썸네일")
-            end_thumbnail = gr.Image(label="끝 썸네일")
-
-        gif_preview = gr.Image(label="GIF 미리보기")
-        gif_download = gr.File(label="GIF 다운로드")
-
-        generate_button = gr.Button("GIF 생성")
-
-        def generate(video, start, end, res, frame_rate, playback_speed, repeat):
-            return process_video(video, start, end, res, frame_rate, playback_speed, repeat)
-
-        generate_button.click(
-            generate,
-            inputs=[video_input, start_time, end_time, resolution, fps, speed, loop],
-            outputs=[start_thumbnail, end_thumbnail, gif_preview, gif_download]
-        )
-
-    return app
-
-if __name__ == "__main__":
-    app = main()
-    app.launch()
+from PIL import Image
+import tempfile
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+
+# Function to get video info
+def get_video_info(video):
+    logging.info("Getting video info")
+    if not video:
+        return "No video uploaded."
+    clip = VideoFileClip(video)
+    duration = clip.duration
+    clip.close()
+    return f"Duration: {duration:.2f} seconds"
+
+# Function to generate thumbnails
+def generate-thumbnails(video, start_time, end_time):
+    logging.info("Generating thumbnails")
+    if not video:
+        return "No video uploaded."
+    clip = VideoFileClip(video)
+    if start_time > clip.duration:
+        start_img = None
+    else:
+        start_frame = clip.get_frame(start_time)
+        start_img = Image.fromarray(start_frame)
+    if end_time > clip.duration:
+        end_img = None
+    else:
+        end_frame = clip.get_frame(end_time)
+        end_img = Image.fromarray(end_frame)
+    clip.close()
+    return start_img, end_img
+
+# Function to generate GIF
+def generate_gif(video, start_time, end_time, resolution, frame_rate, playback_speed, loop_count):
+    logging.info("Generating GIF")
+    if not video:
+        return None, None
+    clip = VideoFileClip(video)
+    # Apply start and end times
+    if start_time > clip.duration:
+        start_time = 0
+    if end_time > clip.duration:
+        end_time = clip.duration
+    subclip = clip.subclip(start_time, end_time)
+    # Apply playback speed
+    subclip = subclip.set_fps(subclip.fps * playback_speed)
+    # Resize if resolution is changed
+    if resolution != subclip.size[0]:
+        ratio = resolution / subclip.size[0]
+        new_height = int(subclip.size[1] * ratio)
+        subclip = subclip.resize((resolution, new_height))
+    # Write GIF
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".gif") as temp:
+        gif_path = temp.name
+        subclip.write_gif(gif_path, fps=frame_rate, loop=loop_count)
+    # Read GIF for preview
+    preview = Image.open(gif_path)
+    return preview, gif_path
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("<h1>Video to GIF Converter</h1>")
+    
+    # Upload video
+    video_input = gr.Video(label="Upload Video")
+    
+    # Button to get video info
+    with gr.Row():
+        get_info_button = gr.Button("Get Video Info")
+        video_info = gr.Textbox(label="Video Information")
+    
+    # Display uploaded video
+    video_output = gr.Video(label="Uploaded Video")
+    
+    # Inputs for start and end times
+    start_time = gr.Number(label="Start Time (seconds)", value=0)
+    end_time = gr.Number(label="End Time (seconds)", value=0)
+    
+    # Thumbnails
+    with gr.Row():
+        start_thumbnail = gr.Image(label="Start Thumbnail")
+        end_thumbnail = gr.Image(label="End Thumbnail")
+    
+    # Sliders
+    resolution = gr.Slider(100, 1920, step=100, value=1920, label="Resolution")
+    frame_rate = gr.Slider(1, 30, step=1, value=10, label="Frame Rate")
+    playback_speed = gr.Slider(0.1, 2.0, step=0.1, value=1.0, label="Playback Speed")
+    loop_count = gr.Slider(0, 10, step=1, value=0, label="GIF Loop Count (0 for infinite)")
+    
+    # GIF Generation button
+    generate_button = gr.Button("GIF 생성")
+    
+    # GIF Preview and Download
+    gif_preview = gr.Image(label="GIF Preview")
+    gif_download = gr.Download(label="Download GIF")
+    
+    # Button functions
+    get_info_button.click(get_video_info, inputs=video_input, outputs=video_info)
+    video_input.change(lambda x: x, inputs=video_input, outputs=video_output)
+    video_input.change(generate-thumbnails, inputs=[video_input, start_time, end_time], outputs=[start_thumbnail, end_thumbnail])
+    generate_button.click(generate_gif, 
+                         inputs=[video_input, start_time, end_time, resolution, frame_rate, playback_speed, loop_count], 
+                         outputs=[gif_preview, gif_download])
+
+demo.launch()