Update app.py

app.py

@@ -562,7 +562,7 @@ def process_video_bg(vid, bg_type="색상", bg_image=None, bg_video=None, color=
         print(f"Error: {e}")
         elapsed_time = time.time() - start_time
         yield gr.update(visible=False), gr.update(visible=True), f"비디오 처리 오류: {e}. 경과 시간: {elapsed_time:.2f}초"
-        yield None, f"비디오 처리 오류: {e}", f"비디오 처리 오류: {e}. 경과 시간: {elapsed_time:.2f}초"
+        yield None, None, f"비디오 처리 오류: {e}. 경과 시간: {elapsed_time:.2f}초"
 
 @spaces.GPU
 def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
@@ -570,7 +570,7 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
     if not video_files:
         return None, "비디오 파일을 업로드해주세요."
     
-    if len(video_files) > 10:
+    if isinstance(video_files, list) and len(video_files) > 10:
         return None, "최대 10개의 비디오만 업로드 가능합니다."
     
     try:
@@ -677,6 +677,18 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
         logging.error(f"Video merge error: {str(e)}")
         return None, f"❌ 오류 발생: {str(e)}"
 
+# GPU 초기화 함수 추가
+def dummy_gpu_init():
+    """GPU 초기화를 위한 더미 함수"""
+    if torch.cuda.is_available():
+        try:
+            # 간단한 텐서 연산으로 GPU 초기화
+            dummy_tensor = torch.zeros(1).to(device)
+            del dummy_tensor
+            logging.info("GPU initialized successfully")
+        except Exception as e:
+            logging.warning(f"GPU initialization warning: {e}")
+
 # CSS
 css = """
 :root {
@@ -1143,4 +1155,5 @@ try:
 except:
     pass
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()