Create app.py

app.py

@@ -0,0 +1,122 @@
+import gradio as gr
+import moviepy.editor as mp
+import os
+from PIL import Image
+
+def process_video(video_file, start_time, end_time, resolution, fps, playback_speed, loop_count):
+    # 로그 출력: 파일 업로드 확인
+    print("파일 업로드 완료:", video_file.name)
+    
+    # 동영상 불러오기
+    video = mp.VideoFileClip(video_file.name)
+    duration = video.duration
+    print("동영상 길이:", duration)
+
+    # 시작/끝 시간 처리
+    start_time = float(start_time) if start_time else 0
+    end_time = float(end_time) if end_time else duration
+    print(f"시작 시간: {start_time}, 끝 시간: {end_time}")
+
+    # 영상 클립 추출
+    trimmed_clip = video.subclip(start_time, end_time)
+
+    # 해상도 조정
+    if resolution != "원본":
+        width, height = [int(i) for i in resolution.split("x")]
+        trimmed_clip = trimmed_clip.resize(height=height)
+        print("해상도 조정:", resolution)
+    else:
+        print("해상도 유지: 원본")
+
+    # 프레임 속도 조정
+    trimmed_clip = trimmed_clip.set_fps(fps)
+    print("프레임 속도 조정:", fps)
+
+    # 재생 속도 조정
+    trimmed_clip = trimmed_clip.fx(mp.vfx.speedx, playback_speed)
+    print("재생 속도 조정:", playback_speed)
+
+    # GIF 반복 횟수
+    loop_count = "infinite" if loop_count == 0 else loop_count
+    print("GIF 반복 횟수 설정:", loop_count)
+
+    # 썸네일 생성
+    start_thumb = trimmed_clip.get_frame(0)
+    end_thumb = trimmed_clip.get_frame(end_time - start_time)
+    start_thumbnail = Image.fromarray(start_thumb)
+    end_thumbnail = Image.fromarray(end_thumb)
+
+    # GIF 저장
+    gif_path = "output.gif"
+    trimmed_clip.write_gif(gif_path, loop=loop_count)
+    print("GIF 생성 완료:", gif_path)
+
+    return start_thumbnail, end_thumbnail, gif_path
+
+
+def preview_gif(gif_path):
+    return gif_path
+
+
+# Gradio 인터페이스 구성
+with gr.Blocks() as app:
+    gr.Markdown("## 동영상을 GIF로 변환하는 툴")
+
+    # 동영상 업로드
+    video_input = gr.Video(label="동영상 업로드")
+
+    # 동영상 재생 및 정보 출력
+    video_output = gr.Video(label="업로드된 동영상 미리보기")
+    video_info = gr.Label(label="동영상 정보")
+
+    # 입력창: 시작 시간, 끝 시간
+    start_time = gr.Textbox(label="시작 시간 (초 단위)", placeholder="예: 5.0")
+    end_time = gr.Textbox(label="끝 시간 (초 단위)", placeholder="예: 10.0")
+
+    # 썸네일 출력
+    start_thumbnail_output = gr.Image(label="시작 지점 썸네일")
+    end_thumbnail_output = gr.Image(label="끝 지점 썸네일")
+
+    # 해상도 조절 슬라이더
+    resolution_slider = gr.Dropdown(
+        choices=["원본", "1920x1080", "1280x720", "640x480"],
+        value="원본",
+        label="해상도 선택"
+    )
+
+    # 프레임 속도 조절 슬라이더
+    fps_slider = gr.Slider(1, 60, step=1, value=30, label="프레임 속도 (기본 30fps)")
+
+    # 재생 속도 조절 슬라이더
+    playback_speed_slider = gr.Slider(0.1, 4.0, step=0.1, value=1.0, label="재생 속도 (기본 1배속)")
+
+    # GIF 반복 횟수 조절 슬라이더
+    loop_count_slider = gr.Slider(0, 10, step=1, value=0, label="GIF 반복 횟수 (0은 무한 반복)")
+
+    # GIF 생성 버튼
+    generate_button = gr.Button("GIF 생성")
+
+    # GIF 미리보기 및 다운로드 출력
+    gif_preview_output = gr.Image(label="GIF 미리보기")
+    gif_download_output = gr.File(label="GIF 다운로드")
+
+    # 동영상 처리 함수 연결
+    video_input.change(
+        lambda video_file: (video_file, {"동영상 길이(초)": mp.VideoFileClip(video_file.name).duration}),
+        inputs=[video_input],
+        outputs=[video_output, video_info]
+    )
+
+    generate_button.click(
+        process_video,
+        inputs=[
+            video_input, start_time, end_time, resolution_slider,
+            fps_slider, playback_speed_slider, loop_count_slider
+        ],
+        outputs=[start_thumbnail_output, end_thumbnail_output, gif_download_output]
+    )
+
+    gif_preview_output.change(preview_gif, inputs=[gif_download_output], outputs=[gif_preview_output])
+
+# 앱 실행
+app.launch()