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,sign):
+def supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip,sign,cookie_value):
     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,"sign":sign}).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,"cookie":cookie_value}).execute()
 
 def supabase_insert_user(name,user_name,profile,picture,oauth_token):
     supabase = get_supabase_client()
@@ -107,6 +107,21 @@ def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token:
         ip = request.client.host
         print("Query parameters:", dict(request.query_params))
         sign = dict(request.query_params).get('__sign')
+
+        # get cookie
+        headers = request.headers.raw 
+        
+        # find 'cookie'
+        cookie_header = next((header for header in headers if header[0] == b'cookie'), None)
+        
+        if cookie_header:
+            # extract cookie
+            cookie_value = cookie_header[1].decode()
+            print(f"Cookie: {cookie_value}")
+        else:
+            cookie_value = ''
+            print("No cookie found in request headers")
+            
     
     # check login
     if profile is None:
@@ -204,7 +219,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,sign)
+    supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip,sign,cookie_value)
     
     return response_content