Added count star messages

app/auth/auth.py

@@ -0,0 +1,17 @@
+from fastapi import APIRouter, HTTPException
+from datetime import datetime, timedelta
+from app.config import settings
+import jwt
+
+router = APIRouter(prefix="/auth", tags=["Auth"])
+jwt_secret_key = settings.jwt_secret
+
+@router.post("/login")
+async def login(email: str, password: str):
+    user = await get_user_by_email(email)
+    if not user or not user.verify_password(password):
+        raise HTTPException(status_code=401, detail="Invalid Credentials")
+     
+    token_expires = datetime.utcnow() + timedelta(hours=1)
+    token=jwt.encode({"sub": email, "exp": token_expires}, jwt_secret_key, algorithm="HS256")
+

app/models/user.py

@@ -0,0 +1,11 @@
+from pydantic import BaseModel, EmailStr
+from passlib.contetx import CryptContext
+
+pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
+
+class User(BaseModel):
+    email: EmailStr
+    hashed_password: str
+
+    def verify_password(self, password: str) -> bool:
+        return pwd_context.verify(password, self.hashed_password)

app/routers/chats.py

@@ -37,6 +37,8 @@ async def get_chat_stats():
     total_requests = await chat_logs.count_documents({})
     unique_sessions = await chat_logs.distinct("session_id")
     total_sessions = len(unique_sessions)
+    
+
     pipeline = [
         {
             "$group": {
@@ -51,10 +53,14 @@ async def get_chat_stats():
     else:
         avg_latency = 0.0
 
+    
+    num_starred = await chat_logs.count_documents({"starred": True})
+
     return {
         "total_sessions": total_sessions,
         "total_requests": total_requests,
-        "avg_latency": avg_latency
+        "avg_latency": avg_latency,
+        "num_starred": num_starred
     }
 
 #Chat History Toggle Star