Update app.py

app.py

@@ -41,10 +41,43 @@ def generate_thumbnail(video_clip, time_point):
         thumbnail_img = Image.fromarray(frame)
         return thumbnail_img
 
+def adjust_aspect_ratio(clip, option):
+    """
+    영상 클립을 선택한 플랫폼의 권장 비율로 중앙 크롭하여 조정합니다.
+    """
+    if option == "원본 유지":
+        return clip
+
+    if option == "유튜브 (16:9)":
+        target_ratio = 16 / 9
+    elif option == "쇼츠/릴스 (9:16)":
+        target_ratio = 9 / 16
+    elif option == "정사각형 (1:1)":
+        target_ratio = 1.0
+    elif option == "인스타그램 (4:5)":
+        target_ratio = 4 / 5
+    elif option == "클래식 (4:3)":
+        target_ratio = 4 / 3
+    else:
+        return clip
+
+    width, height = clip.size
+    current_ratio = width / height
+
+    # 영상이 너무 넓다면 가로를, 너무 높다면 세로를 크롭합니다.
+    if current_ratio > target_ratio:
+        new_width = int(height * target_ratio)
+        new_height = height
+    else:
+        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)
+
 def process_video(video,
                   start_time_str,
                   end_time_str,
-                  resolution_factor,
+                  platform_option,  # 해상도 비율 옵션 선택
                   frame_rate_factor,
                   speed_factor,
                   repeat_count):
@@ -87,12 +120,11 @@ def process_video(video,
     add_log("[LOG 6] 영상 자르기 작업 진행...")
     clip = input_video.subclip(start_sec, end_sec)
 
-    # 해상도 조절 (파일 크기 절감을 위해)
-    if abs(resolution_factor - 1.0) > 1e-3:
-        add_log(f"[LOG 7] 해상도 {resolution_factor*100:.1f}%로 조절 중...")
-        clip = clip.resize(resolution_factor)
+    # 플랫폼별 해상도(비율) 조정
+    add_log(f"[LOG 7] 플랫폼 비율 조정: {platform_option}")
+    clip = adjust_aspect_ratio(clip, platform_option)
 
-    # 재생속도 조절: speedx를 적용하면 clip.duration이 줄어들지만 fps는 그대로 유지됨.
+    # 재생속도 조절: 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)
@@ -110,7 +142,6 @@ def process_video(video,
     add_log("[LOG 11] GIF 생성 중...")
     output_filename = f"temp_{uuid.uuid4().hex}.gif"
     try:
-        # loop=0이면 무한반복, 그 외에는 해당 값 사용
         loop_param = 0 if int(repeat_count) == 0 else int(repeat_count)
         final_clip.write_gif(output_filename, fps=target_fps, program='ffmpeg', loop=loop_param)
         add_log("[LOG 12] GIF 생성 완료! 파일명: " + output_filename)
@@ -165,9 +196,14 @@ with gr.Blocks() as demo:
         # 썸네일 미리보기
         start_thumb_output = gr.Image(label="시작 썸네일 미리보기")
         end_thumb_output = gr.Image(label="종료 썸네일 미리보기")
-        # 해상도, 프레임 레이트, 재생 속도, 반복 횟수 조절
-        resolution_slider = gr.Slider(label="해상도 비율 조절 (0.1 ~ 1.0)", minimum=0.1, maximum=1.0, step=0.1, value=0.5)
-        fps_slider = gr.Slider(label="프레임 레이트 배율 조절 (0.1 ~ 2.0)", minimum=0.1, maximum=2.0, step=0.1, value=1.0)
+        # 플랫폼별 해상도(비율) 선택
+        platform_option = gr.Dropdown(
+            label="해상도/비율 선택",
+            choices=["원본 유지", "유튜브 (16:9)", "쇼츠/릴스 (9:16)", "정사각형 (1:1)", "인스타그램 (4:5)", "클래식 (4:3)"],
+            value="원본 유지"
+        )
+        # 프레임 레이트, 재생 속도, 반복 횟수 조절
+        frame_rate_slider = gr.Slider(label="프레임 레이트 배율 조절 (0.1 ~ 2.0)", minimum=0.1, maximum=2.0, step=0.1, value=1.0)
         speed_slider = gr.Slider(label="재생 속도 조절 (0.5 ~ 2.0)", minimum=0.5, maximum=2.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 생성 버튼 및 결과 출력
@@ -190,14 +226,15 @@ with gr.Blocks() as demo:
         # GIF 생성 버튼 이벤트
         generate_button.click(fn=process_video,
                               inputs=[video_input, start_time, end_time,
-                                      resolution_slider, fps_slider, speed_slider, repeat_slider],
+                                      platform_option, frame_rate_slider, speed_slider, repeat_slider],
                               outputs=[gif_preview_output, download_output, logs_output])
 
     gr.Markdown(
         "### [사용 가이드]\n"
         "1. 동영상을 업로드하세요.\n"
-        "2. 시작/끝 시간을 적절히 입력하세요.\n"
-        "3. 해상도, 프레임 레이트 배율, 재생 속��, 반복 횟수를 조절한 뒤 `GIF 생성하기` 버튼을 누르면 GIF가 생성됩니다.\n"
+        "2. 시작/끝 시간을 입력하세요.\n"
+        "3. 플랫폼별 권장 해상도/비율(원본 유지, 유튜브, 쇼츠/릴스, 정사각형, 인스타그램, 클래식)을 선택하고,\n"
+        "   프레임 레이트, 재생 속도, 반복 횟수를 조절한 후 `GIF 생성하기` 버튼을 누르면 GIF가 생성됩니다.\n"
         "   - 반복 횟수: 0을 선택하면 무한반복, 1~10은 해당 횟수만큼 반복됩니다.\n"
         "4. 결과 미리보기와 다운로드 링크를 통해 GIF를 확인할 수 있습니다."
     )