Update app.py

app.py

@@ -13,6 +13,10 @@ import logging
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
+# GPU 사용 가능 여부 확인
+USE_GPU = cv2.cuda.getCudaEnabledDeviceCount() > 0
+logging.info(f"GPU 사용 가능: {USE_GPU}")
+
 def get_video_duration(video_path):
     try:
         cap = cv2.VideoCapture(video_path)
@@ -118,73 +122,51 @@ def update_slider(video):
         logging.error(f"슬라이더 업데이트 중 오류 발생: {str(e)}")
         raise
 
+
 def process_video(video, start_time, end_time, progress=gr.Progress()):
+    logging.info("process_video 함수 시작")
     try:
-        logging.info("비디오 처리 시작")
         if video is None:
+            logging.warning("비디오 파일이 없습니다.")
             return None, None, "비디오 파일을 업로드해주세요."
         
+        logging.info(f"비디오 파일: {video.name}, 시작 시간: {start_time}, 종료 시간: {end_time}")
+        
         file_size = os.path.getsize(video.name)
-        if file_size > 2 * 1024 * 1024 * 1024:  # 2GB in bytes
+        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)
             if not frames:
+                logging.error("프레임 추출 실패")
                 return None, None, "프레임 추출에 실패했습니다. 비디오 파일을 확인해주세요."
             
+            logging.info(f"추출된 프레임 수: {len(frames)}")
             save_frames(frames, temp_dir, progress=progress)
             grid = create_grid(frames)
             zip_path = create_zip(temp_dir, progress=progress)
             
-            # 결과 파일을 임시 디렉토리 밖으로 복사
             final_zip_path = "frames.zip"
             shutil.copy(zip_path, final_zip_path)
         
         logging.info("비디오 처리 완료")
-        return Image.fromarray(cv2.cvtColor(grid, cv2.COLOR_BGR2RGB)), final_zip_path, f"프레임 추출이 완료되었습니다. 총 {len(frames)}개의 프레임이 추출되었습니다."
+        return Image.fromarray(cv2.cvtColor(grid, cv2.COLOR_BGR2RGB)), final_zip_path, f"프레임 추출이 완료되었습니다. 총 {len(frames)}개의 프레임이 추출되었습니다. ({'GPU' if USE_GPU else 'CPU'} 사용)"
     except Exception as e:
-        logging.error(f"비디오 처리 중 오류 발생: {str(e)}")
+        logging.error(f"비디오 처리 중 오류 발생: {str(e)}", exc_info=True)
         return None, None, f"오류가 발생했습니다: {str(e)}"
 
-css = """
-footer {
-    visibility: hidden;
-}
-#video-time {
-    font-size: 1.2em;
-    font-weight: bold;
-    margin-top: 10px;
-}
-"""
-
-js = """
-function updateVideoTime(videoElement) {
-    const timeDisplay = document.getElementById('video-time');
-    videoElement.addEventListener('timeupdate', () => {
-        const currentTime = videoElement.currentTime.toFixed(2);
-        timeDisplay.textContent = `현재 재생 시간: ${currentTime}초`;
-    });
-}
-document.addEventListener('DOMContentLoaded', () => {
-    const videoElement = document.querySelector('video');
-    if (videoElement) {
-        updateVideoTime(videoElement);
-    }
-});
-"""
-
-with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, js=js) as iface:
+# Gradio 인터페이스 설정
+with gr.Blocks(theme="Nymbo/Nymbo_Theme") as iface:
     gr.Markdown("# 비디오 to 이미지(프레임)")
     gr.Markdown("원하는 구간의 프레임을 추출하여 그리드로 표시하고, 개별 이미지를 ZIP 파일로 다운로드할 수 있습니다.")
-    gr.Markdown("최대 2GB 크기의 비디오 파일을 업로드할 수 있으며, 최대 10000개의 프레임을 추출합니다.")
+    gr.Markdown(f"최대 2GB 크기의 비디오 파일을 업로드할 수 있으며, 최대 10000개의 프레임을 추출합니다. ({'GPU' if USE_GPU else 'CPU'} 사용)")
     
     with gr.Row():
         video_input = gr.Video(label="Upload and Play Video")
     
-    with gr.Row():
-        gr.HTML("<div id='video-time'>현재 재생 시간: 0.00초</div>")
-    
     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 종료 시간 (초)")
@@ -198,8 +180,23 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, js=js) as iface:
     
     message_output = gr.Textbox(label="메시지")
     
-    video_input.change(update_slider, inputs=[video_input], outputs=[start_slider, end_slider])
-    process_button.click(process_video, inputs=[video_input, start_slider, end_slider], outputs=[image_output, file_output, message_output])
+    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])
 
 if __name__ == "__main__":
-    iface.launch(share=True)
+    logging.info("애플리케이션 시작")
+    iface.launch(share=True)
+    logging.info("애플리케이션 종료")