Update main.py

main.py

@@ -1,14 +1,16 @@
-# main.py
-
-from fastapi import FastAPI, Depends, HTTPException
-from fastapi.requests import Request  # Add this import
+from fastapi import FastAPI, Depends, HTTPException, APIRouter
+from fastapi.requests import Request
 from fastapi.responses import HTMLResponse
-import auth, tts
-from auth import verify_token  # Import a function for verifying tokens
+from fastapi.templating import Jinja2Templates
+from sqlalchemy.orm import Session
+from auth import verify_token, oauth2_scheme
+from database import get_db, get_user_by_email
+import auth
+import tts
 
 app = FastAPI()
 router = APIRouter()
-
+templates = Jinja2Templates(directory="templates")
 
 # Include the authentication router with the prefix '/auth'
 app.include_router(auth.router, prefix="/auth")
@@ -21,6 +23,7 @@ def get_current_user(token: str = Depends(verify_token)):
     if not token:
         raise HTTPException(status_code=401, detail="Token not valid")
     return token
+
 @app.get("/login", response_class=HTMLResponse)
 async def login(request: Request):
     return templates.TemplateResponse("login.html", {"request": request})
@@ -34,11 +37,11 @@ async def verify_email(verification_token: str, request: Request):
     # Perform verification and return an appropriate template
     return templates.TemplateResponse("verify.html", {"request": request})
 
-    # User authentication (protected route)
+# User authentication (protected route)
 @app.get("/protected", response_model=str)
-async def protected_route(self,request: Request, token: str = Depends(oauth2_scheme), db: Session = Depends(get_db)):
+async def protected_route(request: Request, token: str = Depends(oauth2_scheme), db: Session = Depends(get_db)):
     # Verify the access token
-    user = verify_token(token, self.SECRET_KEY, self.ALGORITHM)
+    user = verify_token(token, "your-secret-key", "HS256")
     if user is None:
         raise HTTPException(status_code=401, detail="Invalid or expired token")