Update app.py

app.py

@@ -12,7 +12,7 @@ with open('chatbot_urls.json', 'r') as file:
 
 # Initialize or get user-specific ELO ratings
 def get_user_elo_ratings(state):
-    return state.get('elo_ratings', {model: 1200 for model in chatbots.keys()})
+    return state['elo_ratings']
 
 # Read and write ELO ratings to file (thread-safe)
 def read_elo_ratings():
@@ -29,7 +29,7 @@ def write_elo_ratings(elo_ratings):
 # Function to get bot response
 def format_chatml_prompt(state):
     chatml_prompt = ""
-    for message in state.get("history", []):
+    for message in state["history"]
         chatml_prompt += f"<|im_start|> "+message['role']+"- "+ message['content']+"<|im_end|>\n"
     return chatml_prompt
 def get_bot_response(url, prompt):
@@ -84,6 +84,7 @@ def vote_down_model(state, chatbot):
     chatbot.append(update_message)
     return chatbot
 def user_ask(state, chatbot1, chatbot2, textbox):
+    global enable_btn
     user_input = textbox
     if len(user_input) > 200:
         user_input = user_input[:200]  # Limit user input to 200 characters
@@ -110,7 +111,7 @@ def user_ask(state, chatbot1, chatbot2, textbox):
 
     # Format the conversation in ChatML format
 
-    return state, chatbot1, chatbot2, textbox
+    return state, chatbot1, chatbot2, textbox,enable_btn,enable_btn
 
 # Gradio interface setup
 with gr.Blocks() as demo:
@@ -119,17 +120,17 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             chatbot1 = gr.Chatbot(label='Model A').style(height=600)
-            upvote_btn_a = gr.Button(value="👍 Upvote A")
+            upvote_btn_a = gr.Button(value="👍 Upvote A",interactive=False)
         
         with gr.Column():
             chatbot2 = gr.Chatbot(label='Model B').style(height=600)
-            upvote_btn_b = gr.Button(value="👍 Upvote B")        
+            upvote_btn_b = gr.Button(value="👍 Upvote B",interactive=False)        
 
     textbox = gr.Textbox(placeholder="Enter your prompt (up to 200 characters)", max_chars=200)
     submit_btn = gr.Button(value="Send")
 
-    textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox])
-    submit_btn.click(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox])
+    textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox,upvote_btn_a,upvote_btn_b])
+    submit_btn.click(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox,upvote_btn_a,upvote_btn_b])
     upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1], outputs=[chatbot1])
     upvote_btn_b.click(vote_down_model, inputs=[state, chatbot2], outputs=[chatbot2])