Update app.py

app.py

@@ -1,16 +1,12 @@
 import cv2
 import torch
 import requests
-import threading
 import time
 from PIL import Image
-from fastapi import FastAPI
 from supabase import create_client
 from transformers import AutoImageProcessor, AutoModelForImageClassification
 import os
 
-app = FastAPI()
-
 # ===============================
 # 🗄 Supabase Setup
 # ===============================
@@ -22,11 +18,14 @@ supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
 # ===============================
 # 🤖 Load NSFW Model (once)
 # ===============================
-model_name = "AdamCodd/vit-base-nsfw-detector"
+print("🔄 Loading NSFW model...")
 
+model_name = "AdamCodd/vit-base-nsfw-detector"
 processor = AutoImageProcessor.from_pretrained(model_name)
 model = AutoModelForImageClassification.from_pretrained(model_name)
 
+print("✅ Model loaded")
+
 # ===============================
 # 🖼 Image Check
 # ===============================
@@ -42,7 +41,7 @@ def check_image(url):
         return "explicit" if probs[0][1] > 0.5 else "safe"
 
     except Exception as e:
-        print("Image error:", e)
+        print("❌ Image error:", e)
         return "safe"
 
 # ===============================
@@ -77,19 +76,19 @@ def check_video(url, frame_sample_rate=30):
         return "safe"
 
     except Exception as e:
-        print("Video error:", e)
+        print("❌ Video error:", e)
         return "safe"
 
 # ===============================
-# 🔁 BACKGROUND WORKER (EVERY 5 MIN)
+# 🔁 MAIN WORKER LOOP
 # ===============================
 def moderation_worker():
     while True:
         try:
-            print("🚀 Running moderation job...")
+            print("\n🚀 Running moderation job...")
 
             # ===============================
-            # 🖼 Process Images (Batch)
+            # 🖼 Process Images
             # ===============================
             img_posts = supabase.table("posts") \
                 .select("*") \
@@ -113,10 +112,10 @@ def moderation_worker():
                         "result": result
                     }).execute()
 
-                    print(f"✅ IMAGE {post['id']} → {result}")
+                    print(f"🖼 IMAGE {post['id']} → {result}")
 
             # ===============================
-            # 🎥 Process Videos (Batch)
+            # 🎥 Process Videos
             # ===============================
             vid_posts = supabase.table("trendz") \
                 .select("*") \
@@ -145,23 +144,14 @@ def moderation_worker():
         except Exception as e:
             print("❌ Worker error:", e)
 
-        # ⏱ Run every 5 minutes
+        # ⏱ WAIT 5 MINUTES
+        print("⏳ Sleeping for 5 minutes...\n")
         time.sleep(300)
 
-# ===============================
-# 🚀 START WORKER ON STARTUP
-# ===============================
-@app.on_event("startup")
-def start_worker():
-    thread = threading.Thread(target=moderation_worker, daemon=True)
-    thread.start()
 
 # ===============================
-# 🌐 HEALTH CHECK ENDPOINT
+# ▶️ AUTO START
 # ===============================
-@app.get("/")
-def health():
-    return {
-        "status": "running",
-        "message": "NSFW moderation running every 5 minutes"
-    }
+if __name__ == "__main__":
+    print("🔥 Starting NSFW moderation worker...")
+    moderation_worker()