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,cookie_value):
+def supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip,sign,cookie_value,content_type):
     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,"cookie":cookie_value}).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,"content_type":content_type}).execute()
 
 def supabase_insert_user(name,user_name,profile,picture,oauth_token):
     supabase = get_supabase_client()
@@ -155,6 +155,7 @@ def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token:
     user_message = [
         {"type": "text", "text": text},
     ]
+    content_type = 'text'
     if message['files']:
         file = message['files'][0]
         public_url = upload_file_to_gcs_blob(file)
@@ -165,6 +166,7 @@ def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token:
                     "url": public_url,
                 },}
             user_message.append(content_image)
+            content_type = 'image'
         else:
             raise gr.Error("Only support image files now.")
 
@@ -243,7 +245,7 @@ def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token:
     print('-----------------------------------\n')
     response_data = {}
     
-    supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip,sign,cookie_value)
+    supabase_insert_message(user_message,response_content,messages,response_data,user_name,user_oauth_token,ip,sign,cookie_value,content_type)