Add error handling to collect function in mail router

app/router/mail.py

@@ -21,22 +21,25 @@ def collect(email: str, request: Request):
     Returns:
         str: The generated response from the chat function.
     """
-    if os.path.exists(f"cache/{email}.pickle"):
-        with open(f"cache/{email}.pickle", "rb") as token:
-            credentials = pickle.load(token)
-    else:
-        cred_dict = request.state.session.get("credential")
-        credentials = Credentials(
-            token=cred_dict["token"],
-            refresh_token=cred_dict["refresh_token"],
-            token_uri=cred_dict["token_uri"],
-            client_id=cred_dict["client_id"],
-            client_secret=cred_dict["client_secret"],
-            scopes=cred_dict["scopes"],
-        )
-    mailservice = build("gmail", "v1", credentials=credentials)
-    mail.collect(mailservice)
-    return JSONResponse(content={"message": "Mail collected successfully."})
+    try:
+        if os.path.exists(f"cache/{email}.pickle"):
+            with open(f"cache/{email}.pickle", "rb") as token:
+                credentials = pickle.load(token)
+        else:
+            cred_dict = request.state.session.get("credential")
+            credentials = Credentials(
+                token=cred_dict["token"],
+                refresh_token=cred_dict["refresh_token"],
+                token_uri=cred_dict["token_uri"],
+                client_id=cred_dict["client_id"],
+                client_secret=cred_dict["client_secret"],
+                scopes=cred_dict["scopes"],
+            )
+        mailservice = build("gmail", "v1", credentials=credentials)
+        mail.collect(mailservice)
+        return JSONResponse(content={"message": "Mail collected successfully."})
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=500)
 
 # @router.get("")
 # def get():