Create get_credentials function to handle oauth google

.gitignore

@@ -5,6 +5,7 @@ app/media/
 credentials.json
 app/credentials.json
 app/token.pickle
+app/token.json
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

app/ai/tools/google_calendar.py

@@ -1,19 +0,0 @@
-from typing import Union, Dict, Tuple
-from langchain_core.tools import BaseTool
-from ai.integration.google_calendar import Calendar
-
-
-class GoogleCalendarListEventTool(BaseTool):
-    name = "google_calendar_list_event_tool"
-    description = "Useful for when you need to answer questions about events from Google calendar"
-
-    def _to_args_and_kwargs(self, tool_input: Union[str, Dict]) -> Tuple[Tuple, Dict]:
-        return (), {}
-
-    def _run(self) -> str:
-        calendar = Calendar()
-        return calendar.get_all_events()
-
-    async def _arun(self) -> str:
-        """Use the tool asynchronously."""
-        raise NotImplementedError("custom_search does not support async")

app/integration/google_auth.py

@@ -0,0 +1,17 @@
+import os.path
+from google_auth_oauthlib.flow import InstalledAppFlow
+from google.oauth2.credentials import Credentials
+
+
+SCOPES = ["https://www.googleapis.com/auth/calendar.readonly"]
+
+
+def get_credentials() -> None | Credentials:
+    if os.path.exists("credentials.json"):
+        flow = InstalledAppFlow.from_client_secrets_file("credentials.json", SCOPES)
+        creds = flow.run_local_server(port=0)
+        with open("token.json", "w") as token:
+            token.write(creds.to_json())
+        return Credentials.from_authorized_user_file("token.json", SCOPES)
+    else:
+        print("Credentials were not provided")

app/main.py

@@ -5,8 +5,7 @@
 
 from config.settings import settings
 from routers import health, chat, media
-import google_auth
-
+from integration.google_auth import get_credentials
 # from config.database import engine, Base
 # Base.metadata.create_all(bind=engine)
 
@@ -16,6 +15,7 @@
     title=settings.TITLE,
 )
 
+get_credentials()
 app.include_router(health.router)
 app.include_router(chat.router)
 app.include_router(media.router)
@@ -34,15 +34,8 @@
 def startup_event() -> None:
     """
     On app start check if 'media' directory exists and if not create it
-    Create credentials.json file
     """
     if not os.path.exists("media"):
         print("Create media directory")
         os.mkdir("media")
         print("Media directory created")
-
-
-@app.get("/")
-def root():
-    google_auth.main()
-    return {"message": "Hello World"}

app/routers/media.py

@@ -4,12 +4,11 @@
 from fastapi import File, UploadFile
 from fastapi import Response
 
-from ai.integration.notion import notion, get_map_category
+from integration.notion import notion, get_map_category
 from ai.sql_chat_history import get_all_conversations
 from ai.vector import save_to_pinecone
 from ai.vector import split_files
 from config.settings import settings
-from schemas.media import FileCategory
 
 import os