Update app.py

app.py

@@ -107,16 +107,28 @@ def process_video(uploaded_file):
     out.release()
     return output_path
 
+def get_video_download_link(file_path):
+    """Generates a download link for the processed video."""
+    with open(file_path, "rb") as file:
+        video_bytes = file.read()
+    b64 = base64.b64encode(video_bytes).decode()
+    return f'<a href="data:video/mp4;base64,{b64}" download="output_video.mp4">Download Processed Video</a>'
+
 st.title("Bottle Label Checking using YOLO & Gemini AI")
 st.sidebar.header("Upload a video")
 uploaded_file = st.sidebar.file_uploader("Choose a video file", type=["mp4", "avi", "mov"])
 
 if uploaded_file:
-    st.sidebar.write("Processing...")
+    st.sidebar.write("Processing... Please wait!")
     output_video_path = process_video(uploaded_file)
     st.sidebar.success("Processing completed!")
+
+    # Show processed video
     st.video(output_video_path)
 
+    # Provide download link for the video
+    st.markdown(get_video_download_link(output_video_path), unsafe_allow_html=True)
+
     st.subheader("AI Analysis Results")
     for track_id, response in st.session_state["responses"]:
         st.write(f"**Track ID {track_id}:** {response}")