Update app.py

app.py

@@ -1,8 +1,10 @@
 import cv2
 import torch
 import requests
+import threading
+import time
 from PIL import Image
-from fastapi import FastAPI, Header, HTTPException
+from fastapi import FastAPI
 from supabase import create_client
 from transformers import AutoImageProcessor, AutoModelForImageClassification
 import os
@@ -10,16 +12,7 @@ import os
 app = FastAPI()
 
 # ===============================
-# सुरक्षा (API Key Protection)
-# ===============================
-API_TOKEN = os.getenv("API_TOKEN")
-
-def verify_key(x_api_key: str):
-    if API_TOKEN and x_api_key != API_TOKEN:
-        raise HTTPException(status_code=403, detail="Unauthorized")
-
-# ===============================
-# Supabase
+# 🗄 Supabase Setup
 # ===============================
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_KEY")
@@ -27,7 +20,7 @@ SUPABASE_KEY = os.getenv("SUPABASE_KEY")
 supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
 
 # ===============================
-# Load Model (only once)
+# 🤖 Load NSFW Model (once)
 # ===============================
 model_name = "AdamCodd/vit-base-nsfw-detector"
 
@@ -35,7 +28,7 @@ processor = AutoImageProcessor.from_pretrained(model_name)
 model = AutoModelForImageClassification.from_pretrained(model_name)
 
 # ===============================
-# Image check
+# 🖼 Image Check
 # ===============================
 def check_image(url):
     try:
@@ -47,11 +40,13 @@ def check_image(url):
 
         probs = torch.softmax(outputs.logits, dim=1)
         return "explicit" if probs[0][1] > 0.5 else "safe"
-    except:
+
+    except Exception as e:
+        print("Image error:", e)
         return "safe"
 
 # ===============================
-# Video check
+# 🎥 Video Check
 # ===============================
 def check_video(url, frame_sample_rate=30):
     try:
@@ -81,45 +76,92 @@ def check_video(url, frame_sample_rate=30):
         cap.release()
         return "safe"
 
-    except:
+    except Exception as e:
+        print("Video error:", e)
         return "safe"
 
 # ===============================
-# 🔥 MAIN ENDPOINT (like /recommend/all)
+# 🔁 BACKGROUND WORKER (EVERY 5 MIN)
 # ===============================
-@app.get("/moderate/all")
-async def moderate_all(x_api_key: str = Header(None)):
-
-    verify_key(x_api_key)
-
-    results = []
+def moderation_worker():
+    while True:
+        try:
+            print("🚀 Running moderation job...")
+
+            # ===============================
+            # 🖼 Process Images (Batch)
+            # ===============================
+            img_posts = supabase.table("posts") \
+                .select("*") \
+                .limit(10) \
+                .execute().data
+
+            for post in img_posts:
+                existing = supabase.table("content_moderation") \
+                    .select("id") \
+                    .eq("post_id", post["id"]) \
+                    .eq("post_type", "image") \
+                    .execute()
+
+                if len(existing.data) == 0:
+                    result = check_image(post["image_url"])
+
+                    supabase.table("content_moderation").insert({
+                        "post_id": post["id"],
+                        "post_type": "image",
+                        "file_url": post["image_url"],
+                        "result": result
+                    }).execute()
+
+                    print(f"✅ IMAGE {post['id']} → {result}")
+
+            # ===============================
+            # 🎥 Process Videos (Batch)
+            # ===============================
+            vid_posts = supabase.table("trendz") \
+                .select("*") \
+                .limit(5) \
+                .execute().data
+
+            for post in vid_posts:
+                existing = supabase.table("content_moderation") \
+                    .select("id") \
+                    .eq("post_id", post["id"]) \
+                    .eq("post_type", "video") \
+                    .execute()
+
+                if len(existing.data) == 0:
+                    result = check_video(post["video_url"])
+
+                    supabase.table("content_moderation").insert({
+                        "post_id": post["id"],
+                        "post_type": "video",
+                        "file_url": post["video_url"],
+                        "result": result
+                    }).execute()
+
+                    print(f"🎥 VIDEO {post['id']} → {result}")
+
+        except Exception as e:
+            print("❌ Worker error:", e)
+
+        # ⏱ Run every 5 minutes
+        time.sleep(300)
 
-    # Images
-    img_posts = supabase.table("posts").select("*").execute().data
-    for post in img_posts:
-        result = check_image(post["image_url"])
-
-        results.append({
-            "id": post["id"],
-            "type": "image",
-            "url": post["image_url"],
-            "result": result
-        })
-
-    # Videos
-    vid_posts = supabase.table("trendz").select("*").execute().data
-    for post in vid_posts:
-        result = check_video(post["video_url"])
-
-        results.append({
-            "id": post["id"],
-            "type": "video",
-            "url": post["video_url"],
-            "result": result
-        })
+# ===============================
+# 🚀 START WORKER ON STARTUP
+# ===============================
+@app.on_event("startup")
+def start_worker():
+    thread = threading.Thread(target=moderation_worker, daemon=True)
+    thread.start()
 
+# ===============================
+# 🌐 HEALTH CHECK ENDPOINT
+# ===============================
+@app.get("/")
+def health():
     return {
-        "status": "success",
-        "count": len(results),
-        "data": results
+        "status": "running",
+        "message": "NSFW moderation running every 5 minutes"
     }