fix: update temp file logic

app.py

@@ -29,16 +29,15 @@ def load_model():
     return YOLO(new_cached_model_path)
 
 
-def process_video(video_path):
+def process_video(video_path, output_path):
     cap = cv2.VideoCapture(video_path)
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     fps = int(cap.get(cv2.CAP_PROP_FPS))
     total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+
     fourcc = cv2.VideoWriter_fourcc(*'avc1')
-    out = cv2.VideoWriter(temp_file.name, fourcc, fps, (width, height))
+    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
     
     progress_text = "Processing video... Please wait."
     progress_bar = st.progress(0)
@@ -72,7 +71,6 @@ def process_video(video_path):
     progress_bar.empty()
     status_text.text(f"Processed {total_frames} frames")
     time_text.text(f"Total time: {time.time() - start_time:.2f}s")
-    return temp_file.name
 
 
 model = load_model()
@@ -102,15 +100,17 @@ if 'video_path' in locals():
     st.video(video_path)
     
     if st.button("Detect"):
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+
         with st.spinner("Processing video..."):
-            processed_video_path = process_video(video_path)
+            process_video(video_path, temp_file.name)
         
         st.success("Video processing complete!")
         st.header("Processed Video")
-        st.video(processed_video_path)
+        st.video(temp_file.name)
         
         # Add download link
-        with open(processed_video_path, "rb") as file:
+        with open(temp_file.name, "rb") as file:
             btn = st.download_button(
                 label="Download Video",
                 data=file,
@@ -118,8 +118,8 @@ if 'video_path' in locals():
                 mime="video/mp4"
             )
         
-        # Clean up temporary files
-        os.unlink(processed_video_path)
+        # # Clean up temporary files
+        # os.unlink(temp_file.name)
         if video_option == "Upload video":
             os.unlink(video_path)