Update app.py

app.py

@@ -27,20 +27,10 @@ def get_video_duration(video_path):
         return duration
     except Exception as e:
         logging.error(f"비디오 길이 계산 중 오류 발생: {str(e)}")
-        raise
-
-def extract_frame(video_path, frame_number):
-    try:
-        cap = cv2.VideoCapture(video_path)
-        cap.set(cv2.CAP_PROP_POS_FRAMES, frame_number)
-        ret, frame = cap.read()
-        cap.release()
-        return frame if ret else None
-    except Exception as e:
-        logging.error(f"프레임 {frame_number} 추출 중 오류 발생: {str(e)}")
-        return None
+        return 0
 
 def extract_frames(video_path, start_time, end_time, max_frames=10000, progress=gr.Progress()):
+    logging.info(f"프레임 추출 시작: {video_path}, 시작 시간: {start_time}, 종료 시간: {end_time}")
     try:
         cap = cv2.VideoCapture(video_path)
         fps = cap.get(cv2.CAP_PROP_FPS)
@@ -50,78 +40,51 @@ def extract_frames(video_path, start_time, end_time, max_frames=10000, progress=
         end_frame = int(end_time * fps) if end_time else total_frames
         
         frame_interval = max(1, (end_frame - start_frame) // max_frames)
-        frame_numbers = range(start_frame, min(end_frame, total_frames), frame_interval)
-        
         frames = []
-        with concurrent.futures.ThreadPoolExecutor(max_workers=os.cpu_count()) as executor:
-            future_to_frame = {executor.submit(extract_frame, video_path, frame_num): frame_num for frame_num in frame_numbers}
-            for future in progress.tqdm(concurrent.futures.as_completed(future_to_frame), total=len(frame_numbers), desc="프레임 추출 중"):
-                frame = future.result()
-                if frame is not None:
-                    frames.append(frame)
-                if len(frames) >= max_frames:
-                    break
+        
+        for i in progress.tqdm(range(start_frame, min(end_frame, total_frames), frame_interval)):
+            cap.set(cv2.CAP_PROP_POS_FRAMES, i)
+            ret, frame = cap.read()
+            if not ret:
+                break
+            frames.append(frame)
+            if len(frames) >= max_frames:
+                break
         
         cap.release()
         logging.info(f"총 {len(frames)}개의 프레임이 추출되었습니다.")
         return frames
     except Exception as e:
         logging.error(f"프레임 추출 중 오류 발생: {str(e)}")
-        raise
+        return []
 
 def create_grid(frames, grid_size=(10, 10)):
-    try:
-        rows, cols = grid_size
-        n_frames = len(frames)
-        frames = frames[:rows*cols]  # Limit to grid size
-        
-        frame_height, frame_width = frames[0].shape[:2]
-        grid = np.zeros((frame_height * rows, frame_width * cols, 3), dtype=np.uint8)
-        
-        for i, frame in enumerate(frames):
-            row = i // cols
-            col = i % cols
-            grid[row*frame_height:(row+1)*frame_height, col*frame_width:(col+1)*frame_width] = frame
-        
-        return grid
-    except Exception as e:
-        logging.error(f"그리드 생성 중 오류 발생: {str(e)}")
-        raise
+    rows, cols = grid_size
+    n_frames = len(frames)
+    frames = frames[:rows*cols]  # Limit to grid size
+    
+    frame_height, frame_width = frames[0].shape[:2]
+    grid = np.zeros((frame_height * rows, frame_width * cols, 3), dtype=np.uint8)
+    
+    for i, frame in enumerate(frames):
+        row = i // cols
+        col = i % cols
+        grid[row*frame_height:(row+1)*frame_height, col*frame_width:(col+1)*frame_width] = frame
+    
+    return grid
 
 def save_frames(frames, output_dir, progress=gr.Progress()):
-    try:
-        os.makedirs(output_dir, exist_ok=True)
-        for i, frame in progress.tqdm(enumerate(frames), desc="프레임 저장 중", total=len(frames)):
-            cv2.imwrite(os.path.join(output_dir, f"frame_{i:04d}.jpg"), frame)
-        logging.info(f"{len(frames)}개의 프레임이 {output_dir}에 저장되었습니다.")
-    except Exception as e:
-        logging.error(f"프레임 저장 중 오류 발생: {str(e)}")
-        raise
+    os.makedirs(output_dir, exist_ok=True)
+    for i, frame in progress.tqdm(enumerate(frames), desc="프레임 저장 중", total=len(frames)):
+        cv2.imwrite(os.path.join(output_dir, f"frame_{i:04d}.jpg"), frame)
 
 def create_zip(output_dir, progress=gr.Progress()):
-    try:
-        zip_path = os.path.join(output_dir, "frames.zip")
-        with zipfile.ZipFile(zip_path, 'w') as zipf:
-            files = [f for f in os.listdir(output_dir) if f.endswith('.jpg')]
-            for file in progress.tqdm(files, desc="ZIP 파일 생성 중"):
-                zipf.write(os.path.join(output_dir, file), file)
-        logging.info(f"ZIP 파일이 생성되었습니다: {zip_path}")
-        return zip_path
-    except Exception as e:
-        logging.error(f"ZIP 파일 생성 중 오류 발생: {str(e)}")
-        raise
-
-def update_slider(video):
-    try:
-        if video is None:
-            return gr.Slider(minimum=0, maximum=100, value=0, step=0.1, label="CUT 시작 시간 (초)"), gr.Slider(minimum=0, maximum=100, value=100, step=0.1, label="CUT 종료 시간 (초)")
-        duration = get_video_duration(video.name)
-        logging.info(f"비디오 길이: {duration}초")
-        return gr.Slider(minimum=0, maximum=duration, value=0, step=0.1, label="CUT 시작 시간 (초)"), gr.Slider(minimum=0, maximum=duration, value=duration, step=0.1, label="CUT 종료 시간 (초)")
-    except Exception as e:
-        logging.error(f"슬라이더 업데이트 중 오류 발생: {str(e)}")
-        raise
-
+    zip_path = os.path.join(output_dir, "frames.zip")
+    with zipfile.ZipFile(zip_path, 'w') as zipf:
+        files = [f for f in os.listdir(output_dir) if f.endswith('.jpg')]
+        for file in progress.tqdm(files, desc="ZIP 파일 생성 중"):
+            zipf.write(os.path.join(output_dir, file), file)
+    return zip_path
 
 def process_video(video, start_time, end_time, progress=gr.Progress()):
     logging.info("process_video 함수 시작")
@@ -130,16 +93,16 @@ def process_video(video, start_time, end_time, progress=gr.Progress()):
             logging.warning("비디오 파일이 없습니다.")
             return None, None, "비디오 파일을 업로드해주세요."
         
-        logging.info(f"비디오 파일: {video.name}, 시작 시간: {start_time}, 종료 시간: {end_time}")
+        logging.info(f"비디오 파일: {video}, 시작 시간: {start_time}, 종료 시간: {end_time}")
         
-        file_size = os.path.getsize(video.name)
+        file_size = os.path.getsize(video)
         if file_size > 2 * 1024 * 1024 * 1024:  # 2GB
             logging.warning("파일 크기가 2GB를 초과합니다.")
             return None, None, "파일 크기가 2GB를 초과합니다. 더 작은 파일을 업로드해주세요."
         
         with tempfile.TemporaryDirectory() as temp_dir:
             logging.info("프레임 추출 시작")
-            frames = extract_frames(video.name, start_time, end_time, max_frames=10000, progress=progress)
+            frames = extract_frames(video, start_time, end_time, max_frames=10000, progress=progress)
             if not frames:
                 logging.error("프레임 추출 실패")
                 return None, None, "프레임 추출에 실패했습니다. 비디오 파일을 확인해주세요."
@@ -158,7 +121,18 @@ def process_video(video, start_time, end_time, progress=gr.Progress()):
         logging.error(f"비디오 처리 중 오류 발생: {str(e)}", exc_info=True)
         return None, None, f"오류가 발생했습니다: {str(e)}"
 
-# Gradio 인터페이스 설정
+def on_video_change(video):
+    logging.info("비디오 파일이 변경되었습니다.")
+    if video is None:
+        return gr.Slider(minimum=0, maximum=100, value=0, step=0.1), gr.Slider(minimum=0, maximum=100, value=100, step=0.1)
+    try:
+        duration = get_video_duration(video)
+        logging.info(f"비디오 길이: {duration}초")
+        return gr.Slider(minimum=0, maximum=duration, value=0, step=0.1), gr.Slider(minimum=0, maximum=duration, value=duration, step=0.1)
+    except Exception as e:
+        logging.error(f"비디오 정보 획득 중 오류 발생: {str(e)}")
+        return gr.Slider(minimum=0, maximum=100, value=0, step=0.1), gr.Slider(minimum=0, maximum=100, value=100, step=0.1)
+
 with gr.Blocks(theme="Nymbo/Nymbo_Theme") as iface:
     gr.Markdown("# 비디오 to 이미지(프레임)")
     gr.Markdown("원하는 구간의 프레임을 추출하여 그리드로 표시하고, 개별 이미지를 ZIP 파일로 다운로드할 수 있습니다.")
@@ -168,8 +142,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme") as iface:
         video_input = gr.Video(label="Upload and Play Video")
     
     with gr.Row():
-        start_slider = gr.Slider(minimum=0, maximum=10000, value=0, step=1, label="CUT 시작 시간 (초)")
-        end_slider = gr.Slider(minimum=0, maximum=10000, value=10000, step=1, label="CUT 종료 시간 (초)")
+        start_slider = gr.Slider(minimum=0, maximum=100, value=0, step=0.1, label="CUT 시작 시간 (초)")
+        end_slider = gr.Slider(minimum=0, maximum=100, value=100, step=0.1, label="CUT 종료 시간 (초)")
     
     with gr.Row():
         process_button = gr.Button("프레임 추출")
@@ -180,21 +154,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme") as iface:
     
     message_output = gr.Textbox(label="메시지")
     
-    def on_video_change(video):
-        logging.info("비디오 파일이 변경되었습니다.")
-        if video is None:
-            return gr.Slider(minimum=0, maximum=100, value=0, step=0.1), gr.Slider(minimum=0, maximum=100, value=100, step=0.1)
-        duration = get_video_duration(video.name)
-        logging.info(f"비디오 길이: {duration}초")
-        return gr.Slider(minimum=0, maximum=duration, value=0, step=0.1), gr.Slider(minimum=0, maximum=duration, value=duration, step=0.1)
-
     video_input.change(on_video_change, inputs=[video_input], outputs=[start_slider, end_slider])
-    
-    def on_process_click(video, start_time, end_time):
-        logging.info("프레임 추출 버튼이 클릭되었습니다.")
-        return process_video(video, start_time, end_time)
-
-    process_button.click(on_process_click, inputs=[video_input, start_slider, end_slider], outputs=[image_output, file_output, message_output])
+    process_button.click(process_video, inputs=[video_input, start_slider, end_slider], outputs=[image_output, file_output, message_output])
 
 if __name__ == "__main__":
     logging.info("애플리케이션 시작")