Update app.py

app.py

@@ -734,22 +734,38 @@ elif input_method == "Upload Video":
                     
                     status_text.text("Creating processed video...")
                     
-                    # Create memory buffer for video data
-                    video_buffer = io.BytesIO()
+                    # Create a temporary file for the processed video
+                    # Using a unique filename in the temporary directory
+                    temp_output_path = os.path.join(tempfile.gettempdir(), f"processed_video_{int(time.time())}.mp4")
                     
-                    # Write the processed frames directly to memory
-                    with imageio.get_writer(video_buffer, format='FFMPEG', fps=fps, codec='libx264', 
-                                          pixelformat='yuv420p', quality=8) as writer:
-                        for frame_data in processed_frames:
-                            # Convert frame from BGR (OpenCV) to RGB (imageio expects RGB)
-                            rgb_frame = cv2.cvtColor(frame_data["frame"], cv2.COLOR_BGR2RGB)
-                            writer.append_data(rgb_frame)
+                    # Create video writer
+                    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Use mp4v codec which is widely compatible
+                    out = cv2.VideoWriter(
+                        temp_output_path, 
+                        fourcc, 
+                        fps, 
+                        (frame_width, frame_height)
+                    )
                     
-                    # Reset buffer to beginning for reading
-                    video_buffer.seek(0)
+                    # Write frames to video
+                    for frame_data in processed_frames:
+                        out.write(frame_data["frame"])
                     
-                    # Display video directly from memory without saving to disk
-                    st.video(video_buffer)
+                    # Release the writer
+                    out.release()
+                    
+                    # Read the file into memory
+                    with open(temp_output_path, 'rb') as f:
+                        video_bytes = f.read()
+                    
+                    # Display video from bytes
+                    st.video(video_bytes)
+                    
+                    # Clean up the temporary output file
+                    try:
+                        os.remove(temp_output_path)
+                    except:
+                        pass
                     
                     # Analysis complete, display results
                     st.success(f"Video analysis complete. {len(processed_frames)} frames processed, {len(crash_frames)} crashes detected.")