Hugging Face's logo

Spaces:
aliceblue11
/
Gif_test_01
Sleeping

App Files Community
Gif_test_01
File size: 4,182 Bytes 
db3a622
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
 
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()