Synced repo using 'sync_with_huggingface' Github Action

main.py

@@ -2,15 +2,17 @@ from fastapi import FastAPI
 from src.config.appConfig import ENV_VAR
 from starlette.middleware.gzip import GZipMiddleware
 from starlette.middleware.cors import CORSMiddleware
+from starlette.middleware.sessions import SessionMiddleware
 
 from src.main import main
 
 # import os
-origins = ["http://localhost:3000","https://chatbothub.vercel.app"]
+origins = ["http://localhost:3000", "https://chatbothub.vercel.app"]
 
 # origins = os.getenv("ALLOWED_ORIGINS", "").split(",")
 app = FastAPI(debug=ENV_VAR.DEBUG)
 
+app.add_middleware(SessionMiddleware, secret_key=ENV_VAR.SESSION_SECRET)
 app.add_middleware(GZipMiddleware)
 app.add_middleware(
     CORSMiddleware,

requirements.txt

@@ -2,6 +2,7 @@ langchain==0.1.8
 python-dotenv==1.0.0
 huggingface_hub
 typing
+itsdangerous
 
 # pdf
 PyPDF2==3.0.1

src/config/appConfig.py

@@ -16,6 +16,7 @@ class ENV_VAR():
     HUGGINGFACEHUB_API_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
     MONGO_DB_NAME_CACHE = os.environ.get("MONGO_DB_NAME_CACHE")
     JWT_SECRET = os.environ.get("JWT_SECRET")
+    SESSION_SECRET = os.environ.get("SESSION_SECRET")
     AUTH_API_END = os.environ.get("AUTH_API_END")
     DEBUG = (
         os.environ.get("DEBUG").lower() == "true" if os.environ.get("DEBUG") else False

src/controllers/auth/auth_controller.py

@@ -1,4 +1,5 @@
-from fastapi import APIRouter, Response, Request, UploadFile
+from fastapi import APIRouter, Request, UploadFile
+from fastapi.responses import RedirectResponse
 from src.config.appConfig import ENV_VAR
 from src.helpers.response import ResponseHandler
 from src.helpers.send_request import sendRequest
@@ -39,6 +40,16 @@ async def get_user_names():
         return ResponseHandler.error(9999, e)
 
 
+@router.get("/login/google/")
+async def google_login_control(req: Request, baseurl: str):
+    try:
+        # Construct the URL with the query parameters
+        auth_url = f"{AUTH_API_END}/api/v1/auth/google?baseurl={baseurl}"
+        return RedirectResponse(url=auth_url)
+    except Exception as e:
+        return ResponseHandler.error(9999, e)
+
+
 @router.post("/login/")
 async def login_control(req: dict):
     try:
@@ -61,3 +72,25 @@ async def change_pass_control(req: dict, authorization: str = None):
         return ResponseHandler.success_mediator(response)
     except Exception as e:
         return ResponseHandler.error(9999, e)
+
+
+@router.get("/get/session/")
+async def get_session(req: Request):
+    try:
+        response = sendRequest(
+            f"{AUTH_API_END}/api/v1/auth/get/session/", "get", cookies=req.cookies
+        )
+        return ResponseHandler.success_mediator(response)
+    except Exception as e:
+        return ResponseHandler.error(9999, e)
+
+
+@router.get("/logout/")
+async def logout_control(req: Request):
+    try:
+        response = sendRequest(
+            f"{AUTH_API_END}/api/v1/auth/logout/", "get", cookies=req.cookies
+        )
+        return ResponseHandler.success_mediator(response)
+    except Exception as e:
+        return ResponseHandler.error(9999, e)

src/helpers/send_request.py

@@ -1,10 +1,12 @@
 from requests import request
 
 
-def sendRequest(url, method="get", data=None, files=None, headers=None):
+def sendRequest(url, method="get", data=None, files=None, headers=None, cookies=None):
     try:
         if files:
-            return request(method, url, data=data, files=files, headers=headers)
+            return request(
+                method, url, data=data, files=files, headers=headers, cookies=cookies
+            )
         return request(method, url, json=data, headers=headers)
     except Exception as e:
         # Handle exceptions and return None