Update app.py

app.py

@@ -166,23 +166,25 @@ def update_ratings(state, winner_index, collection, category):
     update_elo_rating(collection, elo_ratings, winner_adapter, loser_adapter, category)
     return [('Winner: ', winner), ('Loser: ', loser)]
     
-def vote_up_model(state, chatbot, chatbot2, user_input):
+def vote_up_model(state, chatbot, chatbot2, character_name, character_description, user_name):
+    user_input = format_prompt(state, 0, character_name, character_description, user_name)
     collection = init_database()
     top_category = classify_vote(user_input)
     if top_category:
         update_message = update_ratings(state, 0, collection, top_category)
         chatbot.append(update_message[0])
         chatbot2.append(update_message[1])
-    return chatbot, chatbot2, gr.Button.update(interactive=False), gr.Button.update(interactive=False), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
+    return chatbot, chatbot2, gr.Button.update(interactive=True), gr.Button.update(interactive=True), gr.Textbox.update(value='', interactive=True), gr.Button.update(interactive=True)
 
-def vote_down_model(state, chatbot, chatbot2, user_input):
+def vote_down_model(state, chatbot, chatbot2, character_name, character_description, user_name):
+    user_input = format_prompt(state, 1, character_name, character_description, user_name)
     collection = init_database()
     top_category = classify_vote(user_input)
     if top_category:
         update_message = update_ratings(state, 1, collection, top_category)
         chatbot2.append(update_message[0])
         chatbot.append(update_message[1])
-    return chatbot, chatbot2, gr.Button.update(interactive=False), gr.Button.update(interactive=False), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
+    return chatbot, chatbot2, gr.Button.update(interactive=True), gr.Button.update(interactive=True), gr.Textbox.update(value='', interactive=True), gr.Button.update(interactive=True)
     
 async def user_ask(state, chatbot1, chatbot2, textbox, character_name, character_description, user_name):
     if character_name and len(character_name) > 20:
@@ -347,8 +349,10 @@ with gr.Blocks() as demo:
         textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox, character_name, character_description, user_name], outputs=[state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b], queue=True)
         collection = init_database()
 
-        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1, chatbot2, textbox], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
-        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1, chatbot2, textbox], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
+        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1, chatbot2, character_name, character_description, user_name], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
+        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1, chatbot2, character_name, character_description, user_name], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
+
+    # ...
     with gr.Tab("💬 Direct Chat"):
         gr.Markdown("## 🗣️ Chat directly with a model!")