Update app.py

app.py

@@ -194,16 +194,29 @@ def process_video(video, num_frames):
     
     nsfw_count = 0
     total_frames = len(frames)
+    frame_results = []
 
     for frame_path in frames:
-        if check_nsfw_final(f"https://sdafd-video-nsfw-detect.hf.space/file=/tmp/gradio/{frame_path}"):
+        nsfw_detected = check_nsfw_final(f"https://sdafd-video-nsfw-detect.hf.space/file=/tmp/gradio/{frame_path}")
+        frame_results.append({
+            "frame_path": frame_path,
+            "nsfw_detected": nsfw_detected
+        })
+        if nsfw_detected:
             nsfw_count += 1
     
+    # Cleanup
     for frame_path in frames:
-        if os.path.exists(frame_path):
+        if os.path.exists('/tmp/gradio/' + frame_path):
             os.remove('/tmp/gradio/' + frame_path)
 
-    return f"{nsfw_count} out of {total_frames} frames were NSFW."
+    result = {
+        "nsfw_count": nsfw_count,
+        "total_frames": total_frames,
+        "frames": frame_results
+    }
+    
+    return result
 
 
 with gr.Blocks() as app:
@@ -213,14 +226,14 @@ with gr.Blocks() as app:
         video_input = gr.Video(label="Upload Video")
         num_frames_input = gr.Number(label="Number of Frames to Check", value=10, precision=0)
 
-    output_text = gr.Textbox(label="Result")
+    output_json = gr.JSON(label="Result")
 
     submit_button = gr.Button("Submit")
     
     submit_button.click(
         fn=process_video,
         inputs=[video_input, num_frames_input],
-        outputs=output_text
+        outputs=output_json
     )
 
 app.launch()