Update app.py

app.py

@@ -46,13 +46,27 @@ def is_image(file_path):
     except IOError:
         return False
 
-def supbase_insert(user_message,response_content,messages,response,user_name,user_oauth_token):
+def get_supabase_client() -> Client:
     from supabase import create_client, Client
     url = os.environ.get('supabase_url')
     key = os.environ.get('supbase_key')
-    supabase = create_client(url, key)
+    return create_client(url, key)
+
+def supbase_insert(user_message,response_content,messages,response,user_name,user_oauth_token):
+    supabase = get_supabase_client()
     data, count = supabase.table('messages').insert({"user_message": user_message, "response_content": response_content,"messages":messages,"response":response,"user_name":user_name,"user_oauth_token":user_oauth_token}).execute()
 
+def supabase_insert_user(name,user_name,profile,picture,oauth_token):
+    supabase = get_supabase_client()
+    data.count = supabase.table('users').insert({"name":name,"user_name":user_name,"profile":profile,"picture":picture,"oauth_token":oauth_token}).execute()
+
+
+def supabase_fetch_user(user_name):
+    supabase = get_supabase_client()
+    data,count = supabase.table('users').select("*").eq('user_name',user_name).execute()
+    return data
+        
+
 
 # def respond(
 #     message,
@@ -94,6 +108,11 @@ def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token:
     user_name = profile.username
     user_oauth_token = oauth_token.token
     
+    # check if user exists
+    user_data = supabase_fetch_user(user_name)
+    if not user_data[1]:
+        supabase_insert_user(profile.name,user_name,profile.profile,profile.picture,user_oauth_token)
+
     
     # check if messages are empty
     if message["text"].strip() == "" and not message["files"]: