Update app.py

app.py

@@ -204,6 +204,10 @@ def generate_leaderboard(collection):
 def refresh_leaderboard():
     collection = init_database()
     return generate_leaderboard(collection)
+async def direct_chat(model, user_input, chatbot):
+    response = await get_bot_response(model, user_input, None, 0)
+    chatbot.append((user_input, response))
+    return "", chatbot
 
 # ...
 refresh_leaderboard()
@@ -232,6 +236,13 @@ with gr.Blocks() as demo:
         collection = init_database()
         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])
+    with gr.Tab("Direct Chat"):
+        model_dropdown = gr.Dropdown(choices=chatbots, label="Select a model")
+        direct_chatbot = gr.Chatbot(label="Direct Chat").style(height=600)
+        direct_textbox = gr.Textbox(placeholder="Enter your message")
+        direct_submit_btn = gr.Button(value="Send")
+        direct_textbox.submit(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot], outputs=[direct_textbox, direct_chatbot])
+        direct_submit_btn.click(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot], outputs=[direct_textbox, direct_chatbot])
     with gr.Tab("Leaderboard"):
         try:
             leaderboard = gr.Dataframe(refresh_leaderboard())