Update app.py

app.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, HTTPException
+from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 import cv2
 import uuid
@@ -181,9 +182,14 @@ def check_nsfw_final(img_url):
 
     return result
 
+# Directory for serving frame images
+FRAMES_DIR = "/tmp/frames"
+
+# Mount static file route
+app.mount("/frames", StaticFiles(directory=FRAMES_DIR), name="frames")
+
 # Frame extraction
 def extract_frames(video_path, num_frames, temp_dir):
-    print(video_path)
     vidcap = cv2.VideoCapture(video_path)
     total_frames = int(vidcap.get(cv2.CAP_PROP_FRAME_COUNT))
     frames = []
@@ -204,29 +210,28 @@ def extract_frames(video_path, num_frames, temp_dir):
 
 # Video processing
 def process_video(video_path, num_frames):
-    temp_dir = os.path.join("/tmp", f"frames_{uuid.uuid4().hex}")
-    os.makedirs(temp_dir, exist_ok=True)
+    os.makedirs(FRAMES_DIR, exist_ok=True)  # Ensure frames directory exists
 
-    frames = extract_frames(video_path, num_frames, temp_dir)
+    # Extract frames to the static files directory
+    frames = extract_frames(video_path, num_frames, FRAMES_DIR)
     
     nsfw_count = 0
     total_frames = len(frames)
     frame_results = []
 
     for frame_path in frames:
-        img_url = f"file://{frame_path}"
-        print(img_url)
+        # Construct HTTP URL for the frame
+        frame_filename = os.path.basename(frame_path)
+        img_url = f"https://sdafd-nsfw-video-detect-api.hf.space/frames/{frame_filename}"
+
         nsfw_detected = check_nsfw_final(img_url)
         frame_results.append({
-            "frame_path": frame_path,
+            "frame_path": img_url,  # Use HTTP URL
             "nsfw_detected": nsfw_detected
         })
         if nsfw_detected:
             nsfw_count += 1
     
-    # Cleanup
-    shutil.rmtree(temp_dir, ignore_errors=True)
-
     result = {
         "nsfw_count": nsfw_count,
         "total_frames": total_frames,