비디오 처리 기능 개선 및 코덱 호환성 향상

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import tempfile
+import os
 from lane_detection import process_video as process_video_file
 
 
@@ -11,22 +12,36 @@ def process_video(video_path, method, use_enhanced, use_segmented, progress=gr.P
     if video_path is None:
         return None
 
-    # Create temporary output file
-    temp_output = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
-    output_path = temp_output.name
-    temp_output.close()
-
-    # Progress callback function
-    def update_progress(value, desc):
-        progress(value, desc=desc)
-
-    # Process the video with selected method and progress callback
-    success = process_video_file(video_path, output_path, method, use_enhanced, use_segmented, progress_callback=update_progress)
-
-    if success:
-        return output_path
-    else:
+    # Create temporary output file with explicit .mp4 extension
+    temp_dir = tempfile.gettempdir()
+    temp_filename = f"lane_detection_{os.urandom(8).hex()}.mp4"
+    output_path = os.path.join(temp_dir, temp_filename)
+    
+    try:
+        # Progress callback function
+        def update_progress(value, desc):
+            progress(value, desc=desc)
+
+        # Process the video with selected method and progress callback
+        success = process_video_file(video_path, output_path, method, use_enhanced, use_segmented, progress_callback=update_progress)
+
+        if success and os.path.exists(output_path):
+            return output_path
+        else:
+            return None
+    except Exception as e:
+        print(f"Error processing video: {e}")
+        import traceback
+        traceback.print_exc()
         return None
+    finally:
+        # Clean up input if it was a temporary file
+        if video_path and video_path.startswith(temp_dir):
+            try:
+                if os.path.exists(video_path):
+                    os.remove(video_path)
+            except:
+                pass
 
 
 # Create Gradio interface
@@ -70,7 +85,10 @@ with gr.Blocks(title="Lane Detection Demo") as demo:
             process_btn = gr.Button("Process Video", variant="primary")
 
         with gr.Column():
-            video_output = gr.Video(label="Result (Original | Lane Detection)")
+            video_output = gr.Video(
+                label="Result (Original | Lane Detection)",
+                format="mp4"  # Explicitly set format for better browser compatibility
+            )
 
     # Update checkbox visibility based on method
     def update_checkboxes(method):

lane_detection.py

@@ -944,9 +944,29 @@ def process_video(input_path, output_path, method="advanced", use_enhanced=True,
     total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
 
     # Video writer for output (side-by-side, so width is doubled)
-    # Use mp4v codec for better compatibility
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    out = cv2.VideoWriter(output_path, fourcc, fps, (width * 2, height))
+    # Try different codecs for better browser compatibility
+    codecs_to_try = [
+        ('H264', cv2.VideoWriter_fourcc(*'H264')),  # H.264 - most compatible
+        ('h264', cv2.VideoWriter_fourcc(*'h264')),  # Alternative H.264
+        ('mp4v', cv2.VideoWriter_fourcc(*'mp4v')),  # MPEG-4 Part 2
+        ('XVID', cv2.VideoWriter_fourcc(*'XVID')),  # XVID
+    ]
+    
+    out = None
+    selected_codec = None
+    
+    for codec_name, fourcc in codecs_to_try:
+        out = cv2.VideoWriter(output_path, fourcc, fps, (width * 2, height))
+        if out.isOpened():
+            selected_codec = codec_name
+            print(f"✓ Using codec: {codec_name}")
+            break
+        else:
+            print(f"⚠ Codec {codec_name} not available, trying next...")
+    
+    if out is None or not out.isOpened():
+        print("✗ Error: No suitable video codec found!")
+        return False
 
     frame_count = 0
     print(f"Processing {total_frames} frames using {method} method...")