Update app.py

app.py

@@ -695,9 +695,8 @@ def process_video_file(video_file, analysis_types):
         temp_file.write(video_file.read())
         temp_video_path = temp_file.name
     
-    # Create a temp file for the output video with more reliable path handling
-    # Use Path from pathlib to ensure proper path handling across platforms
-    output_path = str(Path(tempfile.gettempdir()) / f"{Path(temp_video_path).stem}_processed.mp4")
+    # Create a temp file for the output video
+    output_path = f"{temp_video_path}_processed.mp4"
     
     # Open the video file
     cap = cv2.VideoCapture(temp_video_path)
@@ -723,19 +722,8 @@ def process_video_file(video_file, analysis_types):
     output_fps = fps * 0.6
     st.info(f"Output video will be slowed down to {output_fps:.1f} FPS (60% of original speed) for better visualization.")
     
-    # Create video writer with H.264 codec for better compatibility
-    # Try different codec options based on platform compatibility
-    try:
-        # First try H.264 codec (most widely compatible)
-        if cv2.__version__.startswith('4'):
-            fourcc = cv2.VideoWriter_fourcc(*'avc1')  # H.264 codec
-        else:
-            # Fallback to older codec naming
-            fourcc = cv2.VideoWriter_fourcc(*'H264')
-    except Exception:
-        # If H.264 isn't available, fall back to MP4V
-        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    
+    # Create video writer with higher quality settings
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # MP4 codec
     out = cv2.VideoWriter(output_path, fourcc, output_fps, (width, height), isColor=True)
     
     # Process every Nth frame to reduce API calls but increase from 10 to 5 for more detail
@@ -926,20 +914,13 @@ def process_video_file(video_file, analysis_types):
         progress_bar.empty()
         status_text.empty()
         
-        # Verify the file exists before attempting to read it
-        if not os.path.exists(output_path):
-            raise FileNotFoundError(f"Processed video file not found at {output_path}")
-            
         # Read the processed video as bytes for download
         with open(output_path, 'rb') as file:
             processed_video_bytes = file.read()
         
         # Clean up temporary files
-        try:
-            os.unlink(temp_video_path)
-            os.unlink(output_path)
-        except Exception as e:
-            st.warning(f"Warning: Could not delete temporary files: {str(e)}")
+        os.unlink(temp_video_path)
+        os.unlink(output_path)
         
         # Return both the video and the detection statistics
         results = {"detection_stats": detection_stats}
@@ -957,15 +938,10 @@ def process_video_file(video_file, analysis_types):
         cap.release()
         if 'out' in locals():
             out.release()
-        
-        # Only try to delete files if they exist
-        if os.path.exists(temp_video_path):
-            os.unlink(temp_video_path)
+        os.unlink(temp_video_path)
         if os.path.exists(output_path):
             os.unlink(output_path)
-            
-        # Raise the error with more context
-        raise Exception(f"Error processing video: {str(e)} (input: {temp_video_path}, output: {output_path})")
+        raise e
 
 def load_bigquery_table(dataset_id, table_id, limit=1000):
     """Load data directly from an existing BigQuery table"""
@@ -1703,14 +1679,8 @@ def main():
                                     mime="video/mp4"
                                 )
                                 
-                                # Display processed video with fallback options
-                                st.markdown("### Processed Video")
-                                st.markdown("If the video below doesn't play correctly, please use the download button above to view it locally.")
-                                try:
-                                    st.video(processed_video)
-                                except Exception as e:
-                                    st.error(f"Error displaying video in browser: {str(e)}")
-                                    st.info("Please download the video using the button above and play it in your local media player.")
+                                # Display processed video
+                                st.video(processed_video)
                                 
                                 # Show detailed analysis results
                                 st.markdown("### Detailed Analysis Results")