Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import uvicorn
 from fastapi import FastAPI, Depends, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
@@ -10,39 +11,33 @@ import database
 import schemas
 import moderator
 
-# Initialize Database
+# 1. Initialize Database
 database.init_db()
 
+# 2. Setup FastAPI (for your API endpoints)
 app = FastAPI(title="SafeStream API")
+app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"])
 
-# --- FastAPI Logic ---
 def get_db():
     db = database.SessionLocal()
-    try:
-        yield db
-    finally:
-        db.close()
+    try: yield db
+    finally: db.close()
 
 @app.post("/comments", response_model=schemas.Comment)
 def create_comment(comment: schemas.CommentCreate, db: Session = Depends(get_db)):
     analysis = moderator.moderator.analyze(comment.text)
     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"]
+        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
 
-# --- GRADIO UI LOGIC ---
+# 3. Setup Gradio UI
 def moderate_chat(user, text):
-    # This calls your moderation logic for the UI
     analysis = moderator.moderator.analyze(text)
     if analysis["is_toxic"]:
         return f"🚫 BLOCKED: {analysis['reason']} (Score: {analysis['score']:.2f})", "🔴 High Toxicity"
@@ -50,23 +45,20 @@ def moderate_chat(user, text):
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🛡️ SafeStream AI Moderator")
-    
     with gr.Row():
         with gr.Column(scale=2):
             gr.HTML('<iframe width="100%" height="400" src="https://www.youtube.com/embed/dQw4w9WgXcQ" frameborder="0" allowfullscreen></iframe>')
-        
         with gr.Column(scale=1):
             gr.Markdown("### Live Chat Moderation")
             user_input = gr.Textbox(label="Username", value="Guest")
             msg_input = gr.Textbox(label="Type a comment...", placeholder="Try typing something mean or nice!")
             status_box = gr.Label(label="Moderation Status")
             output_text = gr.Textbox(label="Chat Feed", interactive=False)
-            
             submit_btn = gr.Button("Post Comment")
             submit_btn.click(moderate_chat, inputs=[user_input, msg_input], outputs=[output_text, status_box])
 
-# Mount Gradio into FastAPI
-app = gr.mount_gradio_app(app, demo, path="/")
-
+# 4. LAUNCH (The Fix)
+# Instead of mounting, we run Gradio directly on port 7860.
+# Hugging Face will automatically find it.
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)