Update app.py

app.py

@@ -128,12 +128,13 @@ def vote_up_model(state, chatbot, chatbot2, collection):
     update_message = update_ratings(state, 0, collection)
     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)  # Disable voting buttons
+    return chatbot, chatbot2, gr.Button.update(interactive=False), gr.Button.update(interactive=False), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
+
 def vote_down_model(state, chatbot, chatbot2, collection):
     update_message = update_ratings(state, 1, collection)
     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)  # Disable voting buttons
+    return chatbot, chatbot2, gr.Button.update(interactive=False), gr.Button.update(interactive=False), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
 
 async def user_ask(state, chatbot1, chatbot2, textbox):
     global enable_btn
@@ -180,43 +181,39 @@ def generate_leaderboard(collection):
 def refresh_leaderboard():
     return generate_leaderboard()
 # Gradio interface setup
+# Gradio interface setup
 with gr.Blocks() as demo:
     state = gr.State({})
     with gr.Tab("Chatbot Arena"):
         with gr.Row():
             with gr.Column():
                 chatbot1 = gr.Chatbot(label='Model A').style(height=600)
-                upvote_btn_a = gr.Button(value="👍 Upvote A",interactive=False)
+                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",interactive=False)        
+                upvote_btn_b = gr.Button(value="👍 Upvote B", interactive=False)        
     
-        textbox = gr.Textbox(placeholder="Enter your prompt (up to 200 characters)", max_chars=200)
+        textbox = gr.Textbox(placeholder="Enter your prompt (up to 200 characters)")
         with gr.Row():
             submit_btn = gr.Button(value="Send")
             reset_btn = gr.Button(value="Reset")
-        reset_btn.click(clear_chat, inputs=[state], outputs=[state, chatbot1, chatbot2, upvote_btn_a, upvote_btn_b,textbox,submit_btn])
-        textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox,upvote_btn_a,upvote_btn_b], queue=True)
-        submit_btn.click(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox,upvote_btn_a,upvote_btn_b], queue=True)
-        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1,chatbot2], outputs=[chatbot1,chatbot2,upvote_btn_a,upvote_btn_b,textbox,submit_btn])
-        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1,chatbot2], outputs=[chatbot1,chatbot2,upvote_btn_a,upvote_btn_b,textbox,submit_btn])
+        reset_btn.click(clear_chat, inputs=[state], outputs=[state, chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
+        textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b], queue=True)
+        submit_btn.click(user_ask, inputs=[state, chatbot1, chatbot2, textbox], 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, collection], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
+        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1, chatbot2, collection], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
+        
     with gr.Tab("Leaderboard"):
         try:
-            leaderboard = gr.Dataframe(refresh_leaderboard())
+            leaderboard = gr.Dataframe(refresh_leaderboard(collection))
         except:
-            leaderboard=gr.Dataframe(columns=['Chatbot', 'ELO Score', 'Games Played'])
+            leaderboard = gr.Dataframe()
         refresh_btn = gr.Button("Refresh Leaderboard")
+        refresh_btn.click(refresh_leaderboard, inputs=[collection], outputs=[leaderboard])
 
-    # Function to refresh leaderboard
-
-
-    # Event handler for the refresh button
-    refresh_btn.click(refresh_leaderboard, inputs=[], outputs=[leaderboard])
-
-    # Launch the Gradio interface
+# Launch the Gradio interface
 if __name__ == "__main__":
-    collection = init_database()
-    upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1, chatbot2, collection], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
-    upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1, chatbot2, collection], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
     demo.launch(share=False)