Update app.py

app.py

@@ -84,7 +84,7 @@ def vote_down_model(state, chatbot):
     update_message = update_ratings(state, 1)
     chatbot.append(update_message)
     return chatbot
-def user_ask(state, chatbot1, chatbot2, textbox):
+def user_ask(state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b):
     user_input = textbox.value
     bot1_response, bot2_response = chat_with_bots(user_input)
     
@@ -95,7 +95,12 @@ def user_ask(state, chatbot1, chatbot2, textbox):
     chatbot2.append("Bot 2: " + bot2_response['output'])
 
     state['last_bots'] = [bot1_response['model_name'], bot2_response['model_name']]
-    return state, chatbot1, chatbot2, ""
+    
+    # Enable voting buttons
+    upvote_btn_a.interactive = True
+    upvote_btn_b.interactive = True
+
+    return state, chatbot1, chatbot2, "", upvote_btn_a, upvote_btn_b
 
 # ... [Rest of your existing functions] ...
 state = gr.State({})