UPDATE: speed ups

app.py

@@ -1,6 +1,5 @@
 import io
-import os
-import tempfile
+from starlette import status
 from functions import *
 import pandas as pd
 from fastapi import FastAPI, File, UploadFile, HTTPException
@@ -11,6 +10,7 @@ from functions import client as supabase
 from urllib.parse import urlparse
 import nltk
 
+
 nltk.download('punkt_tab')
 
 app = FastAPI(title="ConversAI", root_path="/api/v1")
@@ -33,6 +33,7 @@ async def sign_up(email, username, password):
     )
     user_id = res[1].id
     r_ = createUser(user_id=user_id, username=username)
+    print(r_)
 
     response = {
         "status": "success",
@@ -49,6 +50,18 @@ async def check_session():
 
     return res
 
+@app.post("/get-user")
+async def get_user(access_token):
+    res = supabase.auth.get_user(jwt=access_token)
+    return res
+
+
+@app.post("/referesh-token")
+async def refresh_token(refresh_token):
+    res = supabase.auth.refresh_token(refresh_token)
+    return res
+
+
 
 @app.post("/login")
 async def sign_in(email, password):
@@ -60,6 +73,66 @@ async def sign_in(email, password):
         access_token = res.session.access_token
         refresh_token = res.session.refresh_token
 
+        store_session_check = supabase.table("Stores").select("*").filter("StoreID", "eq", user_id).execute()
+        store_id = None
+        if store_session_check and store_session_check.data:
+            store_id = store_session_check.data[0].get("StoreID")
+
+        userData = client.table("ConversAI_UserInfo").select("*").filter("user_id", "eq", user_id).execute().data
+        username = userData[0]["username"]
+
+        if not store_id:
+            response = (
+                supabase.table("Stores").insert(
+                    {
+                        "AccessToken": access_token,
+                        "StoreID": user_id,
+                        "RefreshToken": refresh_token,
+                    }
+                ).execute()
+            )
+
+            message = {
+                "message": "Success",
+                "code": status.HTTP_200_OK,
+                "username": username,
+                "user_id": user_id,
+                "access_token": access_token,
+                "refresh_token": refresh_token,
+            }
+            return message
+
+        elif store_id == user_id:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="You are already signed in. Please sign out first to sign in again."
+            )
+
+        else:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="Failed to sign in. Please check your credentials."
+            )
+
+    except HTTPException as http_exc:
+        raise http_exc
+
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"An unexpected error occurred during sign-in: {str(e)}"
+        )
+
+
+@app.post('login_with_token')
+async def login_with_token(token):
+    try:
+        res = supabase.auth.sign_in_with_id_token(token)
+        print(res)
+        user_id = res.user.id
+        access_token = res.session.access_token
+        refresh_token = res.session.refresh_token
+
         store_session_check = supabase.table("Stores").select("*").filter("StoreID", "eq", user_id).execute()
         store_id = None
 
@@ -108,6 +181,7 @@ async def sign_in(email, password):
         )
 
 
+
 @app.post("/set-session-data")
 async def set_session_data(access_token, refresh_token):
     res = supabase.auth.set_session(access_token, refresh_token)

requirements.txt

@@ -93,4 +93,5 @@ langsmith
 pandasai
 easyocr
 youtube-transcript-api
-pdf2image
+pdf2image
+PyPDF2