Update app.py

app.py

@@ -13,7 +13,6 @@ def add_log(message: str):
     """
     로그 메시지를 저장하는 함수
     """
-    global global_logs
     global_logs.append(message)
     print(message)
 
@@ -38,7 +37,6 @@ def generate_thumbnail(video_clip, time_point):
         return thumbnail_img
     except Exception as e:
         add_log(f"[ERROR] 썸네일 생성 실패: {e}")
-        # 범위를 벗어나면 첫 프레임
         frame = video_clip.get_frame(0)
         thumbnail_img = Image.fromarray(frame)
         return thumbnail_img
@@ -66,7 +64,7 @@ def adjust_aspect_ratio(clip, option):
     width, height = clip.size
     current_ratio = width / height
 
-    # 영상이 너무 넓으면 좌우 크롭, 너무 높으면 상하 크롭
+    # 영상이 너무 넓다면 가로를, 너무 높다면 세로를 크롭
     if current_ratio > target_ratio:
         new_width = int(height * target_ratio)
         new_height = height
@@ -74,23 +72,16 @@ def adjust_aspect_ratio(clip, option):
         new_width = width
         new_height = int(width / target_ratio)
 
-    return clip.crop(
-        x_center=width / 2,
-        y_center=height / 2,
-        width=new_width,
-        height=new_height
-    )
+    return clip.crop(x_center=width / 2, y_center=height / 2, width=new_width, height=new_height)
 
-def process_video(
-    video,
-    start_time_str,
-    end_time_str,
-    platform_option,
-    frame_rate_factor,
-    speed_factor,
-    repeat_count,
-    resolution_scale
-):
+def process_video(video,
+                  start_time_str,
+                  end_time_str,
+                  platform_option,    # 플랫폼별 비율 선택 (Radio)
+                  frame_rate_factor,  # 프레임 레이트 축소 배율 (1.0이면 원본)
+                  speed_factor,
+                  repeat_count,
+                  resolution_scale):  # 출력 해상도 축소 비율 (1.0이면 원본)
     """
     동영상을 GIF로 변환하는 함수
     """
@@ -99,7 +90,7 @@ def process_video(
 
     add_log("[LOG 1] 비디오 업로드 및 처리 시작")
 
-    # video 파라미터에서 파일 경로 추출
+    # video 파라미터에서 파일 경로 추출 (Gradio에서 반환하는 값이 문자열일 수 있음)
     video_path = video if isinstance(video, str) else video.name
 
     add_log("[LOG 2] 비디오 로드 중...")
@@ -116,7 +107,7 @@ def process_video(
     start_sec = parse_time_to_seconds(start_time_str)
     end_sec = parse_time_to_seconds(end_time_str)
 
-    # 범위 보정
+    # 시간 범위 보정
     if start_sec < 0:
         start_sec = 0
     if end_sec <= 0 or end_sec > duration:
@@ -127,7 +118,6 @@ def process_video(
 
     add_log(f"[LOG 5] 적용된 시작 시간: {start_sec}초, 종료 시간: {end_sec}초")
 
-    # 영상 자르기
     add_log("[LOG 6] 영상 자르기 작업 진행...")
     clip = input_video.subclip(start_sec, end_sec)
 
@@ -135,44 +125,34 @@ def process_video(
     add_log(f"[LOG 7] 플랫폼 비율 조정: {platform_option}")
     clip = adjust_aspect_ratio(clip, platform_option)
 
-    # 출력 해상도 축소
+    # 출력 해상도 축소 옵션
     if abs(resolution_scale - 1.0) > 1e-3:
         add_log(f"[LOG 7-1] 출력 해상도 축소: {resolution_scale*100:.0f}%")
         clip = clip.resize(resolution_scale)
 
-    # 재생속도 조절
+    # 재생속도 조절: speedx를 적용하면 clip.duration은 줄어들지만 fps는 그대로 유지됨.
     if abs(speed_factor - 1.0) > 1e-3:
         add_log(f"[LOG 8] 재생속도 {speed_factor}배로 조절 중...")
         clip = clip.fx(mp.vfx.speedx, speed_factor)
 
-    # 프레임 레이트 축소
+    # FPS 조절: 최종 출력 FPS는 프레임 레이트 배율만 반영 (speed_factor는 이미 duration에 반영됨)
     original_fps = clip.fps
     target_fps = original_fps * frame_rate_factor
     add_log(f"[LOG 9] 최종 출력 FPS: {target_fps:.2f} (원본 FPS: {original_fps})")
     clip = clip.set_fps(target_fps)
 
-    # 반복 횟수
+    # 반복 횟수(0: 무한, 1~10: 지정 횟수)
     add_log(f"[LOG 10] GIF 반복 횟수 설정: {repeat_count} (0이면 무한반복)")
-    # MoviePy의 write_gif에서 loop=N 은 "N+1번 재생"을 의미합니다.
-    # 사용자가 입력한 repeat_count = 재생 횟수 로 맞추기 위해 아래와 같이 설정
-    if repeat_count == 0:
-        loop_param = 0  # 0 => infinite
-    else:
-        loop_param = max(repeat_count - 1, 0)
-
     final_clip = clip
 
-    # GIF 생성
     add_log("[LOG 11] GIF 생성 중...")
     output_filename = f"temp_{uuid.uuid4().hex}.gif"
     try:
-        final_clip.write_gif(
-            output_filename,
-            fps=target_fps,
-            program="ImageMagick",  # ffmpeg -> ImageMagick 로 변경
-            loop=loop_param
-        )
-        add_log("[LOG 12] GIF 생성 완료! 파일명: " + output_filename)
+        # ImageMagick으로 GIF 생성 (기본값), loop=0 => 무한
+        # loop=N => GIF가 총 N회 재생 (N=1이면 재생 후 정지)
+        loop_param = 0 if int(repeat_count) == 0 else int(repeat_count)
+        final_clip.write_gif(output_filename, fps=target_fps, loop=loop_param)
+        add_log(f"[LOG 12] GIF 생성 완료! 파일명: {output_filename}, loop={loop_param}")
     except Exception as e:
         add_log(f"[ERROR] GIF 생성 실패: {e}")
         return None, None, "\n".join(global_logs)
@@ -217,101 +197,61 @@ with gr.Blocks() as demo:
     with gr.Tab("GIF 변환"):
         # 동영상 업로드
         video_input = gr.Video(label="동영상 업로드")
-        
         # 시작/끝 시간 입력
         start_time = gr.Textbox(label="시작 시간 (예: 00:00:05)", value="00:00:00")
         end_time = gr.Textbox(label="종료 시간 (예: 00:00:10)", value="00:00:05")
-        
         # 썸네일 미리보기
         start_thumb_output = gr.Image(label="시작 썸네일 미리보기")
         end_thumb_output = gr.Image(label="종료 썸네일 미리보기")
-        
-        # 플랫폼별 해상도(비율) 선택 (Radio)
+        # 플랫폼별 해상도(비율) 선택 (Radio 버튼)
         platform_option = gr.Radio(
             label="해상도/비율 선택",
-            choices=[
-                "원본 유지",
-                "유튜브 (16:9)",
-                "쇼츠/릴스 (9:16)",
-                "정사각형 (1:1)",
-                "인스타그램 (4:5)",
-                "클래식 (4:3)"
-            ],
+            choices=["원본 유지", "유튜브 (16:9)", "쇼츠/릴스 (9:16)", "정사각형 (1:1)", "인스타그램 (4:5)", "클래식 (4:3)"],
             value="원본 유지"
         )
-        
-        # 출력 해상도 축소 (1.0이면 원본)
-        resolution_scale_slider = gr.Slider(
-            label="출력 해상도 축소 비율 (0.1 ~ 1.0)",
-            minimum=0.1, 
-            maximum=1.0, 
-            step=0.1, 
-            value=1.0
-        )
-        
+        # 출력 해상도 축소 옵션 (1.0이면 원본, 0.1 ~ 1.0 범위)
+        resolution_scale_slider = gr.Slider(label="출력 해상도 축소 비율 (0.1 ~ 1.0)", 
+                                            minimum=0.1, maximum=1.0, step=0.1, value=1.0)
         # 프레임 레이트, 재생 속도, 반복 횟수 조절
-        frame_rate_slider = gr.Slider(
-            label="프레임 레이트 배율 조절 (0.1 ~ 1.0)",
-            minimum=0.1, 
-            maximum=1.0, 
-            step=0.1, 
-            value=1.0
-        )
-        speed_slider = gr.Slider(
-            label="재생 속도 조절 (0.5 ~ 5.0)",
-            minimum=0.5, 
-            maximum=5.0, 
-            step=0.1, 
-            value=1.0
-        )
-        repeat_slider = gr.Slider(
-            label="GIF 반복 횟수 (0=무한, 1=1회, 2=2회, ...)",
-            minimum=0, 
-            maximum=10, 
-            step=1, 
-            value=0
-        )
-
+        frame_rate_slider = gr.Slider(label="프레임 레이트 배율 조절 (0.1 ~ 1.0)", 
+                                      minimum=0.1, maximum=1.0, step=0.1, value=1.0)
+        speed_slider = gr.Slider(label="재생 속도 조절 (0.5 ~ 5.0)", 
+                                 minimum=0.5, maximum=5.0, step=0.1, value=1.0)
+        repeat_slider = gr.Slider(label="GIF 반복 횟수 (0: 무한반복, 1~10: 반복 횟수)", 
+                                  minimum=0, maximum=10, step=1, value=0)
         # GIF 생성 버튼 및 결과 출력
         generate_button = gr.Button("GIF 생성하기")
         gif_preview_output = gr.Image(label="완성된 GIF 미리보기")
         download_output = gr.File(label="GIF 다운로드 링크")
         logs_output = gr.Textbox(label="로그 출력", lines=10)
 
-        # 썸네일 업데이트 이벤트
-        start_time.change(
-            fn=update_thumbnails,
-            inputs=[video_input, start_time, end_time],
-            outputs=[start_thumb_output, end_thumb_output]
-        )
-        end_time.change(
-            fn=update_thumbnails,
-            inputs=[video_input, start_time, end_time],
-            outputs=[start_thumb_output, end_thumb_output]
-        )
-        video_input.change(
-            fn=update_thumbnails,
-            inputs=[video_input, start_time, end_time],
-            outputs=[start_thumb_output, end_thumb_output]
-        )
-        
+        # 썸네일 업데이트 이벤트 (동영상 또는 시간 입력 변경 시)
+        start_time.change(fn=update_thumbnails,
+                          inputs=[video_input, start_time, end_time],
+                          outputs=[start_thumb_output, end_thumb_output])
+        end_time.change(fn=update_thumbnails,
+                        inputs=[video_input, start_time, end_time],
+                        outputs=[start_thumb_output, end_thumb_output])
+        video_input.change(fn=update_thumbnails,
+                           inputs=[video_input, start_time, end_time],
+                           outputs=[start_thumb_output, end_thumb_output])
         # GIF 생성 버튼 이벤트
         generate_button.click(
             fn=process_video,
             inputs=[
-                video_input,
-                start_time,
-                end_time,
-                platform_option,
-                frame_rate_slider,
-                speed_slider,
-                repeat_slider,
-                resolution_scale_slider
+                video_input,         # 동영상 업로드
+                start_time,          # 시작 시간
+                end_time,            # 종료 시간
+                platform_option,     # 플랫폼별 비율
+                frame_rate_slider,   # 프레임 레이트 배율
+                speed_slider,        # 재생 속도
+                repeat_slider,       # 반복 횟수
+                resolution_scale_slider  # 해상도 축소 비율
             ],
             outputs=[
-                gif_preview_output,
-                download_output,
-                logs_output
+                gif_preview_output,  # 완성된 GIF 미리보기
+                download_output,     # GIF 다운로드 링크
+                logs_output          # 로그 출력
             ]
         )
 
@@ -319,13 +259,11 @@ with gr.Blocks() as demo:
         "### [사용 가이드]\n"
         "1. 동영상을 업로드하세요.\n"
         "2. 시작/끝 시간을 입력하세요.\n"
-        "3. 플랫폼별 권장 해상도/비율, 프레임 레이트, 재생 속도, 반복 횟수 등을 조절한 뒤\n"
-        "   `GIF 생성하기` 버튼을 누르면 GIF가 생성됩니다.\n"
-        "   - **GIF 반복 횟수** 예시:\n"
-        "       - 0 → 무한반복\n"
-        "       - 1 → 1회 재생 후 종료\n"
-        "       - 2 → 2회 재생 후 종료\n"
-        "4. 결과 미리보기와 다운로드 링크를 통해 GIF를 확인할 수 있습니다."
+        "3. 플랫폼별 권장 해상도/비율(원본 유지, 유튜브, 쇼츠/릴스, 정사각형, 인스타그램, 클래식)과 출력 해상도 축소 비율을 선택하고,\n"
+        "   프레임 레이트 배율, 재생 속도, 반복 횟수를 조절한 후 `GIF 생성하기` 버튼을 누르면 GIF가 생성됩니다.\n"
+        "   - 반복 횟수(0: 무한반복, 1~10: 지정 횟수). 예) 1이면 1회 재생 후 멈춤, 2이면 2회 재생, 0이면 무한반복.\n"
+        "4. 결과 미리보기와 다운로드 링크를 통해 GIF를 확인할 수 있습니다.\n"
+        "※ 무한반복이 아닌 특정 반복 횟수를 원할 경우, ImageMagick이 설치되어 있어야 합니다."
     )
 
 if __name__ == "__main__":