Update app.py

app.py

@@ -35,8 +35,9 @@ def generate_thumbnail(video_clip, time_point):
         frame = video_clip.get_frame(time_point)
         thumbnail_img = Image.fromarray(frame)
         return thumbnail_img
-    except:
-        # 만약 time_point가 재생시간 범위를 벗어나면 첫 프레임을 썸네일로
+    except Exception as e:
+        add_log(f"[ERROR] 썸네일 생성 실패: {e}")
+        # time_point가 범위를 벗어나면 첫 프레임을 반환
         frame = video_clip.get_frame(0)
         thumbnail_img = Image.fromarray(frame)
         return thumbnail_img
@@ -52,25 +53,28 @@ def process_video(video,
     전체 GIF 변환 과정을 수행하는 함수
     """
     global global_logs
-    global_logs = []  # 호출 시마다 로그를 초기화
+    global_logs = []  # 호출 시마다 로그 초기화
 
-    # 로그 1: 업로드된 파일 확인
     add_log("[LOG 1] 비디오 업로드 및 처리 시작")
 
-    # 비디오 불러오기
+    # video 파라미터에서 파일 경로 추출
+    video_path = video if isinstance(video, str) else video.name
+
     add_log("[LOG 2] 비디오 로드 중...")
-    input_video = mp.VideoFileClip(video.name)
+    try:
+        input_video = mp.VideoFileClip(video_path)
+    except Exception as e:
+        add_log(f"[ERROR] 비디오 로드 실패: {e}")
+        return None, None, "\n".join(global_logs)
 
-    # 재생시간 출력
     duration = input_video.duration
     add_log(f"[LOG 3] 업로드된 영상의 재생시간: {duration:.2f}초")
 
-    # 시작/끝 시간 parse
     add_log("[LOG 4] 시작/끝 시간 파싱 중...")
     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:
@@ -81,44 +85,43 @@ def process_video(video,
 
     add_log(f"[LOG 5] 적용된 시작 시간: {start_sec}초, 종료 시간: {end_sec}초")
 
-    # 해당 구간 자르기(subclip)
     add_log("[LOG 6] 영상 자르기 작업 진행...")
     clip = input_video.subclip(start_sec, end_sec)
 
-    # 속도 조절
     if abs(speed_factor - 1.0) > 1e-3:
         add_log(f"[LOG 7] 속도 {speed_factor}배로 조절 중...")
         clip = clip.fx(mp.vfx.speedx, speed_factor)
 
-    # 해상도 조절 (기본 1.0 = 원본)
     if abs(resolution_factor - 1.0) > 1e-3:
         add_log(f"[LOG 8] 해상도 {resolution_factor*100:.1f}%로 조절 중...")
         clip = clip.resize(resolution_factor)
 
-    # 프레임 레이트 조절 (기본 원본 fps)
     original_fps = clip.fps
     target_fps = original_fps * frame_rate_factor
     add_log(f"[LOG 9] 프레임 레이트를 {target_fps:.2f}로 조절 중... (원본 {original_fps} FPS)")
     clip = clip.set_fps(target_fps)
 
-    # 반복 회수 (최대 10회)
     repeat_count = min(max(int(repeat_count), 1), 10)
     add_log(f"[LOG 10] 반복 회수: {repeat_count}회")
 
     repeated_clips = [clip] * repeat_count
     final_clip = mp.concatenate_videoclips(repeated_clips)
 
-    # GIF 생성
     add_log("[LOG 11] GIF 생성 중...")
-    # 임시 파일 이름 생성
     output_filename = f"temp_{uuid.uuid4().hex}.gif"
 
-    final_clip.write_gif(output_filename, fps=target_fps)
-
-    add_log("[LOG 12] GIF 생성 완료! 파일명: " + output_filename)
+    try:
+        final_clip.write_gif(output_filename, fps=target_fps)
+        add_log("[LOG 12] GIF 생성 완료! 파일명: " + output_filename)
+    except Exception as e:
+        add_log(f"[ERROR] GIF 생성 실패: {e}")
+        return None, None, "\n".join(global_logs)
 
-    # GIF 미리보기용 이미지 반환 (첫 프레임)
-    gif_preview = Image.open(output_filename)
+    try:
+        gif_preview = Image.open(output_filename)
+    except Exception as e:
+        add_log(f"[ERROR] GIF 미리보기 이미지 생성 실패: {e}")
+        gif_preview = None
 
     return gif_preview, output_filename, "\n".join(global_logs)
 
@@ -126,12 +129,18 @@ def update_thumbnails(video, start_time_str, end_time_str):
     """
     시작/끝 썸네일을 업데이트하기 위한 함수
     """
-    input_video = mp.VideoFileClip(video.name)
+    video_path = video if isinstance(video, str) else video.name
+
+    try:
+        input_video = mp.VideoFileClip(video_path)
+    except Exception as e:
+        add_log(f"[ERROR] 비디오 로드 실패: {e}")
+        return None, None
+
     duration = input_video.duration
 
     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:
@@ -152,76 +161,46 @@ with gr.Blocks() as demo:
     gr.Markdown("## 동영상을 GIF로 변환하기 데모")
 
     with gr.Tab("GIF 변환"):
-        # 1. 업로드 입력창 (수정: type 인수를 제거)
+        # 동영상 업로드
         video_input = gr.Video(label="동영상 업로드")
-
-        # 4. 시작/끝 부분 시간 입력
+        # 시작/끝 시간 입력
         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="종료 썸네일 미리보기")
-
-        # 6. 해상도 조절 슬라이더(기본 1.0)
+        # 해상도, 프레임 레이트, 재생 속도, 반복 횟수 조절
         resolution_slider = gr.Slider(label="해상도 비율 조절(0.1 ~ 1.0)", minimum=0.1, maximum=1.0, step=0.1, value=1.0)
-
-        # 7. 프레임 속도 조절 슬라이더(기본 1.0)
         fps_slider = gr.Slider(label="프레임 레이트 배율 조절(0.1 ~ 2.0)", minimum=0.1, maximum=2.0, step=0.1, value=1.0)
-
-        # 8. 속도 조절 슬라이더(기본 1배)
         speed_slider = gr.Slider(label="재생 속도 조절(0.5 ~ 2.0)", minimum=0.5, maximum=2.0, step=0.1, value=1.0)
-
-        # 9. 반복 횟수
         repeat_slider = gr.Slider(label="GIF 반복 횟수(1 ~ 10)", minimum=1, maximum=10, step=1, value=1)
-
-        # 10. GIF 생성 버튼
+        # GIF 생성 버튼 및 결과 출력
         generate_button = gr.Button("GIF 생성하기")
-
-        # 11, 12. 결과 미리보기(이미지), 다운로드(링크)
         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]
-        )
-
-        generate_button.click(
-            fn=process_video,
-            inputs=[
-                video_input,
-                start_time,
-                end_time,
-                resolution_slider,
-                fps_slider,
-                speed_slider,
-                repeat_slider
-            ],
-            outputs=[
-                gif_preview_output,
-                download_output,
-                logs_output
-            ]
-        )
-
-    gr.Markdown("### [사용 가이드]\n1. 동영상을 업로드하세요.\n2. 시작/끝 시간을 적절히 입력하세요.\n3. 해상도, 프레임 속도, 재생 속도, 반복 횟수를 조절한 뒤 `GIF 생성하기` 버튼을 누르면 GIF가 생성됩니다.\n4. 결과 미리보기와 다운로드 링크를 통해 GIF를 확인할 수 있습니다.")
+        # 썸네일 업데이트 이벤트 (입력 시간이나 동영상이 변경될 때)
+        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,
+                                      resolution_slider, fps_slider, speed_slider, repeat_slider],
+                              outputs=[gif_preview_output, download_output, logs_output])
+
+    gr.Markdown("### [사용 가이드]\n"
+                "1. 동영상을 업로드하세요.\n"
+                "2. 시작/끝 시간을 적절히 입력하세요.\n"
+                "3. 해상도, 프레임 속도, 재생 속도, 반복 횟수를 조절한 뒤 `GIF 생성하기` 버튼을 누르면 GIF가 생성됩니다.\n"
+                "4. 결과 미리보기와 다운로드 링크를 통해 GIF를 확인할 수 있습니다.")
 
 if __name__ == "__main__":
     demo.launch()