Refactor Dockerfile permissions and remove unnecessary print statements; update email parameter in collect function

Dockerfile

@@ -8,9 +8,9 @@ COPY . .
 WORKDIR /app
 
 # Create a cache directory and set permissions
-RUN chmod -R 777 /app
+# RUN chmod -R 777 /app
 RUN mkdir -p /app/cache && chmod -R 777 /app/cache
-RUN mkdir -p /app/nltk_data && chmod -R 777 /app/nltk_data
+# RUN mkdir -p /app/nltk_data && chmod -R 777 /app/nltk_data
 
 # Install requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt

app/models/mails/__init__.py

@@ -30,7 +30,6 @@ def build_gmail_service():
     if os.path.exists("token.pickle"):
         with open("token.pickle", "rb") as token:
             creds = pickle.load(token)
-            print(creds.to_json())
     if not creds or not creds.valid:
         if creds and creds.expired and creds.refresh_token:
             creds.refresh(Request())

app/retriever/__init__.py

@@ -62,7 +62,6 @@ class DocRetriever(BaseRetriever):
                 #     #     # "source": "Finfast"
                 #     # }
                 # ))
-            # print(doc_lst)
             return retrieved_docs
         except RuntimeError as e:
             logger.error("Error retrieving documents: %s", e)

app/router/auth.py

@@ -45,6 +45,27 @@ async def get_auth_url():
 
 @router.get("/auth/google/callback")
 async def google_callback(code: str, request: Request):
+    """
+    Handles the Google OAuth2 callback by exchanging the authorization code for credentials,
+    retrieving the user's Gmail profile, and saving the credentials to a file.
+
+    Args:
+        code (str): The authorization code returned by Google's OAuth2 server.
+        request (Request): The incoming HTTP request object.
+
+    Returns:
+        JSONResponse: A JSON response containing the user's Gmail profile information.
+
+    Side Effects:
+        - Saves the user's credentials to a pickle file named after their email address.
+        - Stores the credentials in the session state of the request.
+
+    Dependencies:
+        - google_auth_oauthlib.flow.InstalledAppFlow: Used to handle the OAuth2 flow.
+        - googleapiclient.discovery.build: Used to build the Gmail API service.
+        - json: Used to serialize and deserialize credentials.
+        - pickle: Used to save credentials to a file.
+    """
     flow = InstalledAppFlow.from_client_config(CLIENT_CONFIG, SCOPES)
     flow.redirect_uri = REDIRECT_URI
     flow.fetch_token(code=code)
@@ -69,7 +90,6 @@ async def google_callback(code: str, request: Request):
     # ))
     service = build("gmail", "v1", credentials=credentials)
     profile = service.users().getProfile(userId="me").execute()
-    print(({"profile": profile}))
     with open(f"{profile['emailAddress']}.pickle", "wb") as token:
         pickle.dump(credentials, token)
     return JSONResponse(profile)

app/router/mail.py

@@ -11,7 +11,7 @@ from googleapiclient.discovery import build
 router = APIRouter(prefix="/mail", tags=["mail"])
 
 @router.post("")
-def collect(emailAddress: str, request: Request):
+def collect(email: str, request: Request):
     """
     Handles the chat POST request.
 
@@ -21,9 +21,8 @@ def collect(emailAddress: str, request: Request):
     Returns:
         str: The generated response from the chat function.
     """
-    # emailAddress = req['emailAddress']
-    if os.path.exists(f"{emailAddress}.pickle"):
-        with open(f"{emailAddress}.pickle", "rb") as token:
+    if os.path.exists(f"{email}.pickle"):
+        with open(f"{email}.pickle", "rb") as token:
             credentials = pickle.load(token)
     else:
         cred_dict = request.state.session.get("credential")

app/utils.py

@@ -34,4 +34,4 @@ async def generate(req: ReqData):
                 yield "event: context\n"
                 yield f"data: {json.dumps({'context': context.metadata})}\n\n"
     yield "event: questions\n"
-    yield f"data: {json.dumps({'questions': followUpChain.invoke(req.query, contexts)})}\n\n"
+    yield f"data: {json.dumps({'questions': followUpChain.invoke(req.query, contexts)})}\n\n"