Update app.py

app.py

@@ -29,9 +29,8 @@ def init_database():
     uri = f"mongodb+srv://new-user:{password}@cluster0.xb2urf6.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0"
     client = MongoClient(uri)
     db = client["elo_ratings"]
-    elo_collection = db["elo_ratings"]
-    user_collection = db["user_data"]
-    return elo_collection, user_collection
+    collection = db["elo_ratings"]
+    return collection
 
 def get_user_elo_ratings(collection):
     rows = list(collection.find())
@@ -176,7 +175,7 @@ async def user_ask(state, chatbot1, chatbot2, textbox):
     if len(user_input) > 200:
         user_input = user_input[:200]  # Limit user input to 200 characters
 
-    elo_collection, user_collection = init_database() 
+    collection = init_database()  # Initialize the collection object
 
     # Updating state with the current ELO ratings
     state["elo_ratings"] = get_user_elo_ratings(collection)
@@ -202,12 +201,6 @@ async def user_ask(state, chatbot1, chatbot2, textbox):
     state["history"] = state["history"][-10:]
 
     # Format the conversation in ChatML format
-    if user_id is not None:
-        user_collection.update_one(
-            {"user_id": user_id},
-            {"$push": {"chat_logs": {"user_input": user_input, "bot1_response": bot1_response, "bot2_response": bot2_response}}},
-            upsert=True
-        )
 
     return state, chatbot1, chatbot2, gr.update(value=''),enable_btn,enable_btn
 import pandas as pd
@@ -229,8 +222,8 @@ def submit_model(model_name):
     else:
         return "Discord webhook URL not configured."
         
-def generate_leaderboard(elo_collection):
-    rows = list(elo_collection.find())
+def generate_leaderboard(collection):
+    rows = list(collection.find())
     leaderboard_data = pd.DataFrame(rows, columns=['bot_name', 'elo_rating', 'games_played'])
     leaderboard_data['original_model'] = leaderboard_data['bot_name'].apply(lambda x: next(entry['original_model'] for entry in chatbots_data if entry['adapter'] == x))
     leaderboard_data = leaderboard_data[['original_model', 'elo_rating', 'games_played']]
@@ -240,8 +233,8 @@ def generate_leaderboard(elo_collection):
     return leaderboard_data
     
 def refresh_leaderboard():
-    elo_collection, user_collection = init_database()
-    return generate_leaderboard(elo_collection)
+    collection = init_database()
+    return generate_leaderboard(collection)
 async def direct_chat(model, user_input, chatbot):
     adapter = next(entry['adapter'] for entry in chatbots_data if entry['original_model'] == model)
     temp_state = {
@@ -255,91 +248,12 @@ async def direct_chat(model, user_input, chatbot):
     return "", chatbot
 def reset_direct_chat():
     return "", [], gr.Dropdown.update(value=model_dropdown.value)
-def save_character(user_id, character_name, character_data):
-    if user_id is None:
-        return "Please log in to save characters."
-    elo_collection, user_collection = init_database()
-    user_collection.update_one(
-        {"user_id": user_id},
-        {"$set": {f"characters.{character_name}": character_data}},
-        upsert=True
-    )
-    return f"Character '{character_name}' saved successfully!"
-
-def load_character(user_id, character_name):
-    if user_id is None:
-        return None
-    elo_collection, user_collection = init_database()
-    user_data = user_collection.find_one({"user_id": user_id})
-    if user_data and character_name in user_data.get("characters", {}):
-        return user_data["characters"][character_name]
-    else:
-        return None
-
-def delete_character(user_id, character_name):
-    if user_id is None:
-        return "Please log in to delete characters."
-    elo_collection, user_collection = init_database()
-    user_collection.update_one(
-        {"user_id": user_id},
-        {"$unset": {f"characters.{character_name}": ""}}
-    )
-    return f"Character '{character_name}' deleted successfully!"
-
-def load_chat_logs(user_id):
-    if user_id is None:
-        return []
-    elo_collection, user_collection = init_database()
-    user_data = user_collection.find_one({"user_id": user_id})
-    if user_data:
-        return user_data.get("chat_logs", [])
-    else:
-        return []
-
-def delete_chat_log(user_id, chat_log_index):
-    if user_id is None:
-        return "Please log in to delete chat logs."
-    elo_collection, user_collection = init_database()
-    user_collection.update_one(
-        {"user_id": user_id},
-        {"$unset": {f"chat_logs.{chat_log_index}": ""}}
-    )
-    user_collection.update_one(
-        {"user_id": user_id},
-        {"$pull": {"chat_logs": None}}
-    )
-    return "Chat log deleted successfully!"
-
-def delete_message(user_id, chat_log_index, message_index):
-    if user_id is None:
-        return "Please log in to delete messages."
-    elo_collection, user_collection = init_database()
-    user_collection.update_one(
-        {"user_id": user_id},
-        {"$unset": {f"chat_logs.{chat_log_index}.{message_index}": ""}}
-    )
-    user_collection.update_one(
-        {"user_id": user_id},
-        {"$pull": {f"chat_logs.{chat_log_index}": None}}
-    )
-    return "Message deleted successfully!"
-
-def authorize_user(hf_token):
-    headers = {"Authorization": f"Bearer {hf_token}"}
-    response = requests.get("https://huggingface.co/api/whoami", headers=headers)
-    if response.status_code == 200:
-        user_data = response.json()
-        user_id = user_data["id"]
-        return user_id
-    else:
-        return None
-
 refresh_leaderboard()
 # Gradio interface setup
 # Gradio interface setup
 with gr.Blocks() as demo:
     state = gr.State({})
-    user_id = gr.State(None)
+
     with gr.Tab("🤖 Chatbot Arena"):
         gr.Markdown("## 🥊 Let's see which chatbot wins!")
         
@@ -388,13 +302,14 @@ with gr.Blocks() as demo:
             direct_regenerate_btn = gr.Button(value="🔄 Regenerate")
             direct_reset_btn = gr.Button(value="🗑️ Reset Chat")
 
+        # ...
+
         direct_regenerate_btn.click(direct_regenerate, inputs=[model_dropdown, direct_textbox, direct_chatbot], outputs=[direct_textbox, direct_chatbot])
-        direct_textbox.submit(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot, user_id], outputs=[direct_textbox, direct_chatbot])
-        direct_submit_btn.click(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot, user_id], outputs=[direct_textbox, direct_chatbot])
+        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])
         direct_reset_btn.click(reset_direct_chat, None, [direct_textbox, direct_chatbot, model_dropdown])
 
 
-
     with gr.Tab("🏆 Leaderboard"):
         gr.Markdown("## 📊 Check out the top-performing models!")
         
@@ -415,40 +330,9 @@ with gr.Blocks() as demo:
             model_name_input = gr.Textbox(placeholder="Enter the model name")
             submit_model_btn = gr.Button(value="Submit Model")
             submit_model_btn.click(submit_model, inputs=[model_name_input], outputs=[model_name_input])
-    with gr.Tab("👤 User Account"):
-        gr.Markdown("## 👤 Manage your account")
-
-        with gr.Row():
-            hf_token = gr.Textbox(placeholder="Enter your Hugging Face token")
-            authorize_btn = gr.Button(value="Authorize")
-
-        with gr.Row():
-            character_name = gr.Textbox(placeholder="Enter character name")
-            character_data = gr.Textbox(placeholder="Enter character data (JSON)")
-            save_character_btn = gr.Button(value="Save Character")
-
-        with gr.Row():
-            load_character_dropdown = gr.Dropdown(label="Select character to load")
-            load_character_btn = gr.Button(value="Load Character")
-            delete_character_btn = gr.Button(value="Delete Character")
-
-        with gr.Row():
-            chat_logs_dropdown = gr.Dropdown(label="Select chat log")
-            delete_chat_log_btn = gr.Button(value="Delete Chat Log")
-        with gr.Row():
-            chat_log_index = gr.Number(label="Chat Log Index")
-            message_index = gr.Number(label="Message Index")
-            delete_message_btn = gr.Button(value="Delete Message")
-
-        authorize_btn.click(authorize_user, inputs=[hf_token], outputs=[user_id])
-        save_character_btn.click(save_character, inputs=[user_id, character_name, character_data], outputs=[])
-        load_character_btn.click(load_character, inputs=[user_id, load_character_dropdown], outputs=[character_data])
-        delete_character_btn.click(delete_character, inputs=[user_id, load_character_dropdown], outputs=[])
-        chat_logs_dropdown.change(load_chat_logs, inputs=[user_id], outputs=[chat_logs_dropdown])
-        delete_chat_log_btn.click(delete_chat_log, inputs=[user_id, chat_logs_dropdown], outputs=[])
-        delete_message_btn.click(delete_message, inputs=[user_id, chat_log_index, message_index], outputs=[])
 
 
 # Launch the Gradio interface
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch(share=False)
+