add hg login

app.py

@@ -46,12 +46,12 @@ def is_image(file_path):
     except IOError:
         return False
 
-def supbase_insert(user_message,response_content,messages,response):
+def supbase_insert(user_message,response_content,messages,response，user_name,user_oauth_token):
     from supabase import create_client, Client
     url = os.environ.get('supabase_url')
     key = os.environ.get('supbase_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()
+    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()
 
 
 # def respond(
@@ -86,7 +86,16 @@ def supbase_insert(user_message,response_content,messages,response):
 #         response += token
 #         yield response
 
-def get_completion(message,history):
+def get_completion(message,history,profile: gr.OAuthProfile | None,oauth_token: gr.OAuthToken | None):
+    # check login
+    if profile is None:
+        raise gr.Error('Click "Sign in with Hugging Face" to continue')
+        
+    user_name = profile.username
+    user_oauth_token = oauth_token.token
+    
+    
+    # check if messages are empty
     if message["text"].strip() == "" and not message["files"]:
         raise gr.Error("Please input a query and optionally image(s).")
     
@@ -161,7 +170,7 @@ def get_completion(message,history):
         response_content = response_data['choices'][0]['message']['content']
         usage = response_data['usage']
 
-    supbase_insert(user_message,response_content,messages,response_data)
+    supbase_insert(user_message,response_content,messages,response_data，user_name,user_oauth_token)
     
     return response_content
 
@@ -174,17 +183,25 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 title = "ChatGPT-4o"
 description = "This is GPT-4o, you can use the text and image capabilities now. More capabilities like audio and video will be rolled out iteratively in the future. Stay tuned."
 
-demo = gr.ChatInterface(
-    get_completion,
-    multimodal=True,
-    title = title,
-    description = description
-    # additional_inputs=[
-    #     gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-    #     gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-    #     gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-    # ],
-)
+
+with gr.Blocks(fill_height=True) as demo:
+    gr.Markdown(
+        "# ChatGPT-4o"
+        "\n\nThis is GPT-4o, you can use the text and image capabilities now. More capabilities like audio and video will be rolled out iteratively in the future. Stay tuned."
+    )
+    gr.LoginButton()
+
+    gr.ChatInterface(
+        get_completion,
+        multimodal=True,
+        # title = title,
+        # description = description
+        # additional_inputs=[
+        #     gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+        #     gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+        #     gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+        # ],
+    )
 
 demo.queue(default_concurrency_limit=5)