Upload 2 files

database.py

@@ -0,0 +1,34 @@
+from sqlalchemy import create_engine, Column, Integer, String, Boolean, Float, Text, ForeignKey
+from sqlalchemy.orm import sessionmaker, declarative_base
+
+SQLALCHEMY_DATABASE_URL = "sqlite:///./safestream.db"
+
+engine = create_engine(
+    SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
+)
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+Base = declarative_base()
+
+class Video(Base):
+    __tablename__ = "videos"
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String, index=True)
+    url = Column(String) # YouTube Embed ID or URL
+    description = Column(String)
+
+class Comment(Base):
+    __tablename__ = "comments"
+    id = Column(Integer, primary_key=True, index=True)
+    video_id = Column(Integer, ForeignKey("videos.id"))
+    user = Column(String)
+    text = Column(Text)
+    timestamp = Column(String) # ISO format
+    
+    # Analysis fields
+    is_toxic = Column(Boolean, default=False)
+    toxicity_score = Column(Float, default=0.0)
+    flagged_reason = Column(String, nullable=True) 
+
+def init_db():
+    Base.metadata.create_all(bind=engine)

schemas.py

@@ -0,0 +1,35 @@
+from pydantic import BaseModel
+from typing import Optional, List
+
+# Video Schemas
+class VideoBase(BaseModel):
+    title: str
+    url: str
+    description: Optional[str] = None
+
+class VideoCreate(VideoBase):
+    pass
+
+class Video(VideoBase):
+    id: int
+    class Config:
+        orm_mode = True
+
+# Comment Schemas
+class CommentBase(BaseModel):
+    video_id: int
+    user: str
+    text: str
+    timestamp: Optional[str] = None
+
+class CommentCreate(CommentBase):
+    pass
+
+class Comment(CommentBase):
+    id: int
+    is_toxic: bool
+    toxicity_score: float
+    flagged_reason: Optional[str] = None
+
+    class Config:
+        orm_mode = True