Update app.py

app.py

@@ -46,6 +46,14 @@ def is_image(file_path):
     except IOError:
         return False
 
+def supbase_insert(user_message,response_content,messages,response):
+    from supabase import create_client, Client
+    url = os.environ.get('supabase_url')
+    key = os.environ.get('supabase_key')
+    supabase = create_client(url, key)
+    data, count = supabase.table('messages').insert({"user_message": user_message, "response_content": response_content,"messages":messages,"response":response}).execute()
+
+
 # def respond(
 #     message,
 #     history: list[tuple[str, str]],
@@ -86,7 +94,7 @@ def get_completion(message,history):
         raise gr.Error("Please input a text query along the image(s).")
     
     text = message['text']
-    content = [
+    user_message = [
         {"type": "text", "text": text},
     ]
     if message['files']:
@@ -98,7 +106,7 @@ def get_completion(message,history):
                 "image_url": {
                     "url": public_url,
                 },}
-            content.append(content_image)
+            user_message.append(content_image)
         else:
             raise gr.Error("Only support image files now.")
 
@@ -109,7 +117,7 @@ def get_completion(message,history):
             continue
         history_openai_format.append({"role": "user", "content": human })
         history_openai_format.append({"role": "assistant", "content":assistant})
-    history_openai_format.append({"role": "user", "content": content})
+    history_openai_format.append({"role": "user", "content": user_message})
     # print(history_openai_format)
     
     system_message = '''You are GPT-4o("o" for omni), OpenAI's new flagship model that can reason across audio, vision, and text in real time. 
@@ -150,6 +158,8 @@ def get_completion(message,history):
     else:
         response_content = response_data['choices'][0]['message']['content']
         usage = response_data['usage']
+
+    supbase_insert(user_message,response_content,messages,response_data)
     
     return response_content