Update app.py

app.py

@@ -128,15 +128,6 @@ def process_video_and_save(uploaded_file):
         out.write(frame_bgr)
 
     out.release()
-
-    # Create a download button for the processed video
-    with open(output_path, "rb") as video_file:
-        btn = st.download_button(
-            label="Download Processed Video",
-            data=video_file,
-            file_name="processed_video.mp4",
-            mime="video/mp4"
-        )
     
     return output_path
 
@@ -146,20 +137,24 @@ def process_video_and_save(uploaded_file):
 def live_video_feed():
     stframe = st.empty()
     video = cv2.VideoCapture(0)
-    start_time = time.time()
+
+    if not video.isOpened():
+        st.error("Unable to access the webcam.")
+        return
 
     while True:
         ret, frame = video.read()
         if not ret:
+            st.error("Failed to capture frame.")
             break
 
+        # Run YOLO on the captured frame
         results = run_yolo(frame)
-        processed_frame = process_results(results, frame)
-        processed_frame_rgb = cv2.cvtColor(processed_frame, cv2.COLOR_BGR2RGB)
-        stframe.image(processed_frame_rgb, channels="RGB", use_column_width=True)
+        annotated_frame = process_results(results, frame)
+        annotated_frame_rgb = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
 
-        elapsed_time = time.time() - start_time
-        st.write(f"Elapsed Time: {elapsed_time:.2f} seconds")
+        # Display the frame with detections
+        stframe.image(annotated_frame_rgb, channels="RGB", use_column_width=True)
 
         if st.button("Stop"):
             break
@@ -231,7 +226,15 @@ def main():
         uploaded_file = st.file_uploader("Choose a video...", type=["mp4", "mov"])
         if uploaded_file is not None:
             output_path = process_video_and_save(uploaded_file)
-            st.video(output_path)
+
+            # Now, move the download button here, outside the cached function
+            with open(output_path, "rb") as video_file:
+                btn = st.download_button(
+                    label="Download Processed Video",
+                    data=video_file,
+                    file_name="processed_video.mp4",
+                    mime="video/mp4"
+                )
 
     elif input_type == "Live Feed":
         live_video_feed()