Upload 4 files

db/blog_schema.py

@@ -0,0 +1,20 @@
+from pydantic import BaseModel
+from datetime import datetime
+
+class PostResponse(BaseModel):
+    id: int
+    title: str
+    slug: str
+    excerpt: str
+    content: str
+    thumbnail_url: str | None = None
+    product_url: str | None = None
+    image_urls: str | None = None
+    published_at: datetime
+    tags: str | None = None               # ✅ Added
+    rating: float | None = None           # ✅ New field
+    repeat_purchases: str | None = None   # ✅ New field
+
+    model_config = {
+        "from_attributes": True
+    }

db/database.py

@@ -0,0 +1,25 @@
+# db/database.py
+
+import os
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+from fastapi.security import OAuth2PasswordBearer
+from dotenv import load_dotenv
+
+load_dotenv()
+
+# ✅ PostgreSQL URL from Neon
+SQLALCHEMY_DATABASE_URL = os.getenv("NEON_DATABASE_URL")  # put this in .env
+
+# ✅ Create PostgreSQL engine
+engine = create_engine(SQLALCHEMY_DATABASE_URL)
+
+# ✅ PostgreSQL session
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+# ✅ Base class for models
+Base = declarative_base()
+
+# ✅ Token config
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/login")

db/models.py

@@ -0,0 +1,27 @@
+# db/models.py
+
+from sqlalchemy import Column, Integer, String, ForeignKey, DateTime
+from sqlalchemy.orm import relationship
+from datetime import datetime
+from .database import Base
+
+class User(Base):
+    __tablename__ = "users"
+
+    id = Column(Integer, primary_key=True, index=True)
+    email = Column(String, unique=True, index=True, nullable=False)
+    name = Column(String, nullable=False)
+    hashed_password = Column(String, nullable=False)
+    created_at = Column(DateTime, default=datetime.utcnow)
+
+    threads = relationship("Thread", back_populates="owner")
+
+class Thread(Base):
+    __tablename__ = "threads"
+
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String, default="New Chat")
+    created_at = Column(DateTime, default=datetime.utcnow)
+    user_id = Column(Integer, ForeignKey("users.id"))
+
+    owner = relationship("User", back_populates="threads")

db/schemas.py

@@ -0,0 +1,66 @@
+from pydantic import BaseModel, EmailStr
+from typing import List, Optional
+from datetime import datetime
+
+# --------------------------
+# User Schemas
+# --------------------------
+
+class UserCreate(BaseModel):
+    name: str
+    email: EmailStr
+    password: str
+
+
+class UserLogin(BaseModel):
+    email: EmailStr
+    password: str
+
+
+class Token(BaseModel):
+    access_token: str
+    token_type: str
+
+
+class TokenData(BaseModel):
+    email: Optional[str] = None
+
+
+# --------------------------
+# Thread Schemas
+# --------------------------
+
+class ThreadBase(BaseModel):
+    title: Optional[str] = "New Chat"
+
+
+class ThreadCreate(ThreadBase):
+    pass
+
+
+class ThreadOut(ThreadBase):
+    id: int
+    created_at: datetime
+
+    class Config:
+        from_attributes = True  # ✅ for Pydantic v2
+
+
+# --------------------------
+# User Output with Threads
+# --------------------------
+
+class UserResponse(BaseModel):
+    id: int
+    name: str
+    email: EmailStr
+    created_at: datetime
+    threads: List[ThreadOut] = []
+
+    class Config:
+        from_attributes = True
+
+
+# If needed separately (e.g., for /auth/me route):
+class UserWithThreads(UserResponse):
+    pass