Update app.py

app.py

@@ -8,6 +8,7 @@ import random
 import string
 import json
 import ast
+import uuid
 
     
 def check_nsfw(img_url):
@@ -175,34 +176,32 @@ def extract_frames(video_path, num_frames):
     frames = []
     
     for i in range(num_frames):
-        frame_number = int(i * total_frames / num_frames)  # Sample evenly
+        frame_number = int(i * total_frames / num_frames) 
         vidcap.set(cv2.CAP_PROP_POS_FRAMES, frame_number)
         success, image = vidcap.read()
         if success:
-            # Save frame to temporary file and append it to list
-            temp_img = tempfile.NamedTemporaryFile(suffix=".jpg", delete=False)
-            cv2.imwrite(temp_img.name, image)
-            frames.append(temp_img.name)
+            random_filename = f"{uuid.uuid4().hex}.jpg"
+            cv2.imwrite(random_filename, image)
+            frames.append(random_filename)
         else:
             break
 
     vidcap.release()
     return frames
 
-# Gradio app logic
 def process_video(video, num_frames):
-    
-    # Extract frames from the video
     frames = extract_frames(video, num_frames)
     
     nsfw_count = 0
     total_frames = len(frames)
 
-    # Check each frame for NSFW content
     for frame_path in frames:
         if check_nsfw_final(f"https://sdafd-video-nsfw-detect.hf.space/file={frame_path}"):
             nsfw_count += 1
-        os.remove(frame_path)  # Clean up temp frame files
+    
+    for frame_path in frames:
+        if os.path.exists(frame_path):
+            os.remove(frame_path)
 
     return f"{nsfw_count} out of {total_frames} frames were NSFW."
 
@@ -224,4 +223,4 @@ with gr.Blocks() as app:
         outputs=output_text
     )
 
-app.launch()
+app.launch()