Update app.py

app.py

@@ -671,7 +671,6 @@ elif input_method == "Upload Video":
         tfile_path = tfile.name
         tfile.close()
 
-        # Process button
         if st.sidebar.button("Process Video"):
             try:
                 cap = cv2.VideoCapture(tfile_path)
@@ -722,7 +721,6 @@ elif input_method == "Upload Video":
 
                     cap.release()
 
-                    # Write processed frames to a new video file
                     status_text.text("Creating processed video...")
                     output_video_path = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name
 
@@ -730,41 +728,30 @@ elif input_method == "Upload Video":
                         first_frame = processed_frames[0]["frame"]
                         h, w = first_frame.shape[:2]
 
-                        fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # fallback codec
+                        fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Use 'mp4v' for compatibility
                         out = cv2.VideoWriter(output_video_path, fourcc, fps, (w, h))
 
                         for frame_data in processed_frames:
                             out.write(frame_data["frame"])
 
                         out.release()
-                        time.sleep(0.5)  # ensure video file is flushed
+                        time.sleep(0.5)  # Ensure the file is fully flushed
 
                         try:
                             with open(output_video_path, 'rb') as video_file:
                                 video_bytes = video_file.read()
 
+                            # Generate download link
+                            b64 = base64.b64encode(video_bytes).decode()
+                            href = f'<a href="data:video/mp4;base64,{b64}" download="processed_video.mp4">📥 Click here to download the processed video</a>'
+
                             status_text.empty()
                             progress_bar.empty()
-
-                            if "huggingface.co" in os.environ.get("HOST", "") or "SPACE_ID" in os.environ:
-                                # Display video using HTML for Hugging Face Spaces
-                                video_base64 = base64.b64encode(video_bytes).decode("utf-8")
-                                st.markdown(
-                                    f"""
-                                    <video width="100%" controls autoplay>
-                                        <source src="data:video/mp4;base64,{video_base64}" type="video/mp4">
-                                        Your browser does not support the video tag.
-                                    </video>
-                                    """,
-                                    unsafe_allow_html=True
-                                )
-                            else:
-                                # Local streamlit display
-                                st.video(video_bytes)
+                            st.markdown(href, unsafe_allow_html=True)
 
                         except Exception as e:
-                            st.error(f"Error displaying video: {e}")
-                    
+                            st.error(f"Error preparing video download: {e}")
+
                     st.success(f"Video analysis complete. {len(processed_frames)} frames processed, {len(crash_frames)} crashes detected.")
 
                     if crash_frames:
@@ -825,6 +812,7 @@ elif input_method == "Upload Video":
                 except:
                     pass
 
+
 st.markdown("---")
 col1, col2, col3, col4 = st.columns(4)