Update app.py

app.py

@@ -136,6 +136,7 @@ def detect_image(image):
     results = model(image)
     return results.render()[0]
 
+
 def detect_video(video_path):
     video = cv2.VideoCapture(video_path)
     frame_rate = video.get(cv2.CAP_PROP_FPS)
@@ -143,6 +144,12 @@ def detect_video(video_path):
     # Create a directory to store the frames
     frames_dir = 'frames'
     os.makedirs(frames_dir, exist_ok=True)
+    
+    # Remove existing contents of the frames directory
+    for file_name in os.listdir(frames_dir):
+        file_path = os.path.join(frames_dir, file_name)
+        if os.path.isfile(file_path):
+            os.remove(file_path)
 
     frame_count = 0
     process_frame_count = 0
@@ -162,6 +169,12 @@ def detect_video(video_path):
     # Process the frames with object detection and save the results
     results_dir = 'results'
     os.makedirs(results_dir, exist_ok=True)
+    
+    # Remove existing contents of the results directory
+    for file_name in os.listdir(results_dir):
+        file_path = os.path.join(results_dir, file_name)
+        if os.path.isfile(file_path):
+            os.remove(file_path)
 
     for i in range(process_frame_count):
         frame_path = os.path.join(frames_dir, f'frame_{i:04d}.jpg')
@@ -188,12 +201,11 @@ def detect_video(video_path):
     video_writer.release()
 
     # Clean up the temporary directories
-    os.rmdir(frames_dir)
-    os.rmdir(results_dir)
+    shutil.rmtree(frames_dir)
+    shutil.rmtree(results_dir)
 
     return video_output_path
 
-
 # Create Gradio interfaces for different modes
 img_interface = gr.Interface(
     fn=detect_image,