app.py

import gradio as gr
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()