Update app.py

app.py

@@ -54,9 +54,9 @@ def get_supabase_client():
     supabase = create_client(url, key)
     return supabase
 
-def supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip):
+def supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip,sign):
     supabase = get_supabase_client()
-    data, count = supabase.table('messages').insert({"user_message": user_message, "response_content": response_content,"messages":messages,"response":response_data,"user_name":user_name,"user_oauth_token":user_oauth_token,"ip":ip}).execute()
+    data, count = supabase.table('messages').insert({"user_message": user_message, "response_content": response_content,"messages":messages,"response":response_data,"user_name":user_name,"user_oauth_token":user_oauth_token,"ip":ip,"sign":sign}).execute()
 
 def supabase_insert_user(name,user_name,profile,picture,oauth_token):
     supabase = get_supabase_client()
@@ -105,18 +105,27 @@ def supabase_fetch_user(user_name):
 def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token: gr.OAuthToken | None,request: gr.Request):
     if request:
         ip = request.client.host
+        sign = request.query_params['__sign']
     
     # check login
     if profile is None:
-        raise gr.Error('Click "Sign in with Hugging Face" to continue')
+        # raise gr.Error('Click "Sign in with Hugging Face" to continue')
+        user_name = 'unknown'
+        user_oauth_token = ''
+        name = 'unknown'
+        pf = ''
+        pic = ''
+    else:
+        user_name = profile.username
+        user_oauth_token = oauth_token.token
+        name = profile.name
+        pf = profile.profile
+        pic = profile.picture
         
-    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)
+        supabase_insert_user(name,user_name,pf,pic,user_oauth_token)
 
     
     # check if messages are empty
@@ -194,7 +203,7 @@ def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token:
         response_content = response_data['choices'][0]['message']['content']
         usage = response_data['usage']
 
-    supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip)
+    supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip,sign)
     
     return response_content