Update app.py

app.py

@@ -54,9 +54,9 @@ def get_supabase_client():
     supabase = create_client(url, key)
     return supabase
 
-def supbase_insert(user_message,response_content,messages,response,user_name,user_oauth_token):
+def supabase_insert_message(user_message,response_content,messages,response,user_name,user_oauth_token,ip):
     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()
+    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,"ip":ip}).execute()
 
 def supabase_insert_user(name,user_name,profile,picture,oauth_token):
     supabase = get_supabase_client()
@@ -102,7 +102,10 @@ def supabase_fetch_user(user_name):
 #         response += token
 #         yield response
 
-def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token: gr.OAuthToken | None):
+def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token: gr.OAuthToken | None,request: gr.Request):
+    if request:
+        ip = request.client.host
+    
     # check login
     if profile is None:
         raise gr.Error('Click "Sign in with Hugging Face" to continue')
@@ -191,7 +194,7 @@ def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token:
         response_content = response_data['choices'][0]['message']['content']
         usage = response_data['usage']
 
-    supbase_insert(user_message,response_content,messages,response_data,user_name,user_oauth_token)
+    supabase_insert_message(user_message,response_content,messages,response,user_name,user_oauth_token,ip)
     
     return response_content