Update app.py

app.py

@@ -81,20 +81,21 @@ def vote_down_model(state, chatbot):
     return chatbot
 def user_ask(state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b):
     user_input = textbox
-    bot1_response, bot2_response = chat_with_bots(user_input,state)
-    
+    bot1_response, bot2_response = chat_with_bots(user_input, state)
+
     chatbot1.append("User: " + user_input)
     chatbot1.append("Bot 1: " + bot1_response)
 
     chatbot2.append("User: " + user_input)
-    chatbot2.append("Bot 2: " + bot2_response)    
+    chatbot2.append("Bot 2: " + bot2_response)
+
     # Enable voting buttons
     upvote_btn_a.interactive = True
     upvote_btn_b.interactive = True
-    print(state, chatbot1, chatbot2, "", upvote_btn_a, upvote_btn_b)
+
     updated_elo_ratings = get_user_elo_ratings(state)
     state.update({'elo_ratings': updated_elo_ratings})
-    return state, chatbot1, chatbot2, "", upvote_btn_a, upvote_btn_b
+    return state, chatbot1, chatbot2
 
 # ... [Rest of your existing functions] ...
 
@@ -104,25 +105,32 @@ with gr.Blocks() as demo:
         # First column for Model A
         with gr.Column():
             chatbot1 = gr.Chatbot(label='Model A')
-            with gr.Row():
-                upvote_btn_a = gr.Button(value="👍 Upvote A", interactive=False)
+            upvote_btn_a = gr.Button(value="👍
+
+# ... [Rest of your existing functions] ...
+
+with gr.Blocks() as demo:
+    state = gr.State({})
+    with gr.Row():
+        # First column for Model A
+        with gr.Column():
+            chatbot1 = gr.Chatbot(label='Model A')
+            upvote_btn_a = gr.Button(value="👍 Upvote A", interactive=False)
         
         # Second column for Model B
         with gr.Column():
             chatbot2 = gr.Chatbot(label='Model B')
-            with gr.Row():
-                upvote_btn_b = gr.Button(value="👍 Upvote B", interactive=False)        
+            upvote_btn_b = gr.Button(value="👍 Upvote B", interactive=False)        
+
     # Textbox and submit button at the bottom
-    with gr.Row():
-        textbox = gr.Textbox(placeholder="Enter your prompt and press ENTER")
-        submit_btn = gr.Button(value="Send")
+    textbox = gr.Textbox(placeholder="Enter your prompt and press ENTER")
+    submit_btn = gr.Button(value="Send")
 
     # Interaction logic
-    textbox.submit(user_ask, [state, chatbot1, chatbot2, textbox,upvote_btn_a, upvote_btn_b], [state, chatbot1, chatbot2, textbox,upvote_btn_a, upvote_btn_b])
-    submit_btn.click(user_ask, [state, chatbot1, chatbot2, textbox,upvote_btn_a, upvote_btn_b], [state, chatbot1, chatbot2, textbox,upvote_btn_a, upvote_btn_b])
-    upvote_btn_a.click(vote_up_model, [state, chatbot1], [chatbot1])
-    upvote_btn_b.click(vote_up_model, [state, chatbot2], [chatbot2])
+    textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b], outputs=[state, chatbot1, chatbot2])
+    submit_btn.click(user_ask, inputs=[state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b], outputs=[state, chatbot1, chatbot2])
+    upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1], outputs=[chatbot1])
+    upvote_btn_b.click(vote_up_model, inputs=[state, chatbot2], outputs=[chatbot2])
 
     # Start the interface
-demo.queue()
-demo.launch(server_name='0.0.0.0', server_port=7860)
+    demo.launch()