Update app.py

app.py

@@ -1,118 +1,121 @@
-from fastapi import FastAPI, Depends, HTTPException
-from fastapi.middleware.cors import CORSMiddleware
-from sqlalchemy.orm import Session
-from typing import List
-import database, schemas, moderator
-
-# Init DB
-database.init_db()
-
-app = FastAPI(title="SafeStream API")
-
-# Allow all origins for dev
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-def get_db():
-    db = database.SessionLocal()
-    try:
-        yield db
-    finally:
-        db.close()
-
-@app.on_event("startup")
-def startup_event():
-    # Seed data if empty
-    db = database.SessionLocal()
-    if db.query(database.Video).count() == 0:
-        # Using real YouTube IDs for the iframe
-        seed_videos = [
-            database.Video(title="Daily Vlog #102", url="dQw4w9WgXcQ", description="Just another day in the life. Comment below!"),
-            database.Video(title="Gaming Highlights", url="jNQXAC9IVRw", description="Insane plays from last night."),
-            database.Video(title="Relaxing Rain Sounds", url="q76bMs-NwRk", description="Sleep aid.")
-        ]
-        db.add_all(seed_videos)
-        db.commit()
-    db.close()
-
-@app.get("/videos", response_model=List[schemas.Video])
-def get_videos(db: Session = Depends(get_db)):
-    return db.query(database.Video).all()
-
-@app.get("/videos/{video_id}", response_model=schemas.Video)
-def get_video(video_id: int, db: Session = Depends(get_db)):
-    video = db.query(database.Video).filter(database.Video.id == video_id).first()
-    if not video:
-        raise HTTPException(status_code=404, detail="Video not found")
-    return video
-
-
-@app.get("/videos/{video_id}/comments", response_model=List[schemas.Comment])
-def get_comments(video_id: int, db: Session = Depends(get_db)):
-    return db.query(database.Comment).filter(database.Comment.video_id == video_id).all()
-
-@app.post("/comments", response_model=schemas.Comment)
-def create_comment(comment: schemas.CommentCreate, db: Session = Depends(get_db)):
-    # 1. Analyze for toxicity
-    analysis = moderator.moderator.analyze(comment.text)
-    
-    # 2. Save to DB
-    db_comment = database.Comment(
-        video_id=comment.video_id,
-        user=comment.user,
-        text=comment.text,
-        timestamp=comment.timestamp,
-        is_toxic=analysis["is_toxic"],
-        toxicity_score=analysis["score"],
-        flagged_reason=analysis["reason"]
-    )
-    db.add(db_comment)
-    db.commit()
-    db.refresh(db_comment)
-    return db_comment
-
-@app.delete("/comments")
-def clear_comments(db: Session = Depends(get_db)):
-    db.query(database.Comment).delete()
-    db.commit()
-    return {"message": "All comments cleared"}
-
-@app.get("/stats")
-def get_stats(db: Session = Depends(get_db)):
-    comments = db.query(database.Comment).all()
-    total = len(comments)
-    toxic_comments = [c for c in comments if c.is_toxic]
-    toxic_count = len(toxic_comments)
-    
-    # Custom Classification Logic for Demo
-    types = {
-        "Insult": 0, 
-        "Identity Hate": 0, 
-        "Threat": 0, 
-        "Online Harassment": 0
-    }
-    
-    for c in toxic_comments:
-        text = c.text.lower()
-        if any(w in text for w in ["kill", "die", "hurt", "attack", "gun", "shoot", "murder"]):
-            types["Threat"] += 1
-        elif any(w in text for w in ["hate", "racist", "gay", "lesbian", "black", "white", "jew", "muslim", "women", "men", "trans"]):
-            types["Identity Hate"] += 1
-        elif any(w in text for w in ["stupid", "idiot", "dumb", "ugly", "fat", "loser", "moron", "coward"]):
-            types["Insult"] += 1
-        else:
-            # Catch-all for other toxic comments
-            types["Online Harassment"] += 1
-            
-    return {
-        "total_comments": total, 
-        "toxic_comments": toxic_count, 
-        "toxic_ratio": (toxic_count/total) if total > 0 else 0,
-        "types_breakdown": [{"name": k, "value": v} for k, v in types.items()]
-    }
-
+import uvicorn
+from fastapi import FastAPI, Depends, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from sqlalchemy.orm import Session
+from typing import List
+
+# Import your local files
+import database
+import schemas
+import moderator
+
+# Initialize Database
+database.init_db()
+
+app = FastAPI(title="SafeStream API")
+
+# Allow all origins for development
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+def get_db():
+    db = database.SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+@app.on_event("startup")
+def startup_event():
+    # Seed data if empty
+    db = database.SessionLocal()
+    if db.query(database.Video).count() == 0:
+        seed_videos = [
+            database.Video(title="Daily Vlog #102", url="dQw4w9WgXcQ", description="Just another day in the life. Comment below!"),
+            database.Video(title="Gaming Highlights", url="jNQXAC9IVRw", description="Insane plays from last night."),
+            database.Video(title="Relaxing Rain Sounds", url="q76bMs-NwRk", description="Sleep aid.")
+        ]
+        db.add_all(seed_videos)
+        db.commit()
+    db.close()
+
+@app.get("/")
+def read_root():
+    return {"message": "SafeStream API is running. Visit /docs for documentation."}
+
+@app.get("/videos", response_model=List[schemas.Video])
+def get_videos(db: Session = Depends(get_db)):
+    return db.query(database.Video).all()
+
+@app.get("/videos/{video_id}", response_model=schemas.Video)
+def get_video(video_id: int, db: Session = Depends(get_db)):
+    video = db.query(database.Video).filter(database.Video.id == video_id).first()
+    if not video:
+        raise HTTPException(status_code=404, detail="Video not found")
+    return video
+
+@app.get("/videos/{video_id}/comments", response_model=List[schemas.Comment])
+def get_comments(video_id: int, db: Session = Depends(get_db)):
+    return db.query(database.Comment).filter(database.Comment.video_id == video_id).all()
+
+@app.post("/comments", response_model=schemas.Comment)
+def create_comment(comment: schemas.CommentCreate, db: Session = Depends(get_db)):
+    # 1. Analyze for toxicity using your moderator tool
+    analysis = moderator.moderator.analyze(comment.text)
+        
+    # 2. Save to DB
+    db_comment = database.Comment(
+        video_id=comment.video_id,
+        user=comment.user,
+        text=comment.text,
+        timestamp=comment.timestamp,
+        is_toxic=analysis["is_toxic"],
+        toxicity_score=analysis["score"],
+        flagged_reason=analysis["reason"]
+    )
+    db.add(db_comment)
+    db.commit()
+    db.refresh(db_comment)
+    return db_comment
+
+@app.delete("/comments")
+def clear_comments(db: Session = Depends(get_db)):
+    db.query(database.Comment).delete()
+    db.commit()
+    return {"message": "All comments cleared"}
+
+@app.get("/stats")
+def get_stats(db: Session = Depends(get_db)):
+    comments = db.query(database.Comment).all()
+    total = len(comments)
+    toxic_comments = [c for c in comments if c.is_toxic]
+    toxic_count = len(toxic_comments)
+    
+    types = {"Insult": 0, "Identity Hate": 0, "Threat": 0, "Online Harassment": 0}
+    
+    for c in toxic_comments:
+        text = c.text.lower()
+        if any(w in text for w in ["kill", "die", "hurt", "attack", "gun", "shoot", "murder"]):
+            types["Threat"] += 1
+        elif any(w in text for w in ["hate", "racist", "gay", "lesbian", "black", "white", "jew", "muslim", "women", "men", "trans"]):
+            types["Identity Hate"] += 1
+        elif any(w in text for w in ["stupid", "idiot", "dumb", "ugly", "fat", "loser", "moron", "coward"]):
+            types["Insult"] += 1
+        else:
+            types["Online Harassment"] += 1
+                
+    return {
+        "total_comments": total, 
+        "toxic_comments": toxic_count, 
+        "toxic_ratio": (toxic_count/total) if total > 0 else 0,
+        "types_breakdown": [{"name": k, "value": v} for k, v in types.items()]
+    }
+
+# CRITICAL: This part runs the server on Hugging Face's port
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)