Update app.py

app.py

@@ -77,6 +77,16 @@ def process_image(uploaded_file):
     processed_image_rgb = cv2.cvtColor(processed_image, cv2.COLOR_BGR2RGB)
     st.image(processed_image_rgb, caption='Detected Image', use_column_width=True)
 
+    # Create a download button for the processed image
+    im_pil = Image.fromarray(processed_image_rgb)
+    im_pil.save("processed_image.png")
+    with open("processed_image.png", "rb") as file:
+        btn = st.download_button(
+            label="Download Processed Image",
+            data=file,
+            file_name="processed_image.png",
+            mime="image/png"
+        )
 
 # Process and save uploaded videos
 @st.cache_data
@@ -118,9 +128,20 @@ 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
 
 
+
 # Live video feed processing
 def live_video_feed():
     stframe = st.empty()