Update app.py

app.py

@@ -8,69 +8,75 @@ from elo import update_elo_ratings  # Custom function for ELO ratings
 with open('chatbot_urls.json', 'r') as file:
     chatbots = json.load(file)
 
-# Function to read ELO ratings from a file
+# Load existing ELO ratings or initialize them
 def read_elo_ratings():
     try:
         with open('elo_ratings.json', 'r') as file:
             return json.load(file)
     except FileNotFoundError:
-        # Initialize ELO ratings for existing models
         return {model: 1200 for model in chatbots.keys()}
 
-# Function to update ELO ratings in a file
+# Update ELO ratings in a file
 def write_elo_ratings(elo_ratings):
     with open('elo_ratings.json', 'w') as file:
         json.dump(elo_ratings, file, indent=4)
 
-# Load existing ELO ratings or initialize them
 elo_ratings = read_elo_ratings()
 
-def chat_with_bots(user_input, bot1_url, bot2_url):
-    # Function to interact with both chatbots
-    def get_bot_response(url, prompt):
-        payload = {
-            "input": {
-                "prompt": prompt,
-                "sampling_params": {
-                    "max_new_tokens": 16,
-                    "temperature": 0.7,
-                }
+def get_bot_response(url, prompt):
+    payload = {
+        "input": {
+            "prompt": prompt,
+            "sampling_params": {
+                "max_new_tokens": 16,
+                "temperature": 0.7,
             }
         }
-        headers = {
-            "accept": "application/json",
-            "content-type": "application/json",
-            "authorization": os.environ.get("RUNPOD_TOKEN")
-        }
-        response = requests.post(url, json=payload, headers=headers)
-        return response.json()
+    }
+    headers = {
+        "accept": "application/json",
+        "content-type": "application/json",
+        "authorization": os.environ.get("RUNPOD_TOKEN")
+    }
+    response = requests.post(url, json=payload, headers=headers)
+    return response.json()
+
+def chat_with_bots(user_input):
+    bot1_url = chatbots[list(chatbots.keys())[0]]
+    bot2_url = chatbots[list(chatbots.keys())[1]]
 
     bot1_response = get_bot_response(bot1_url, user_input)
     bot2_response = get_bot_response(bot2_url, user_input)
 
     return bot1_response, bot2_response
 
-def update_ratings(winner_model):
-    # Update ELO ratings based on the winner
+def update_ratings(bot_index):
     global elo_ratings
-    loser_model = set(elo_ratings.keys()) - {winner_model}
-    elo_ratings = update_elo_ratings(elo_ratings, winner_model, loser_model)
+    bot_names = list(chatbots.keys())
+    winner = bot_names[bot_index]
+    loser = bot_names[1 - bot_index]
+    
+    elo_ratings = update_elo_ratings(elo_ratings, winner, loser)
     write_elo_ratings(elo_ratings)
+    
+    return f"Bot 1 is {bot_names[0]} with ELO {elo_ratings[bot_names[0]]}\nBot 2 is {bot_names[1]} with ELO {elo_ratings[bot_names[1]]}"
 
 # Gradio Interface
 iface = gr.Interface(
     fn=chat_with_bots,
     inputs=[
-        gr.inputs.Textbox(label="Your message"),
-        gr.inputs.Radio(list(chatbots.values()), label="Bot 1"),
-        gr.inputs.Radio(list(chatbots.values()), label="Bot 2")
+        gr.inputs.Textbox(label="Your message")
     ],
     outputs=[
         gr.outputs.Textbox(label="Bot 1 Response"),
-        gr.outputs.Textbox(label="Bot 2 Response"),
-        gr.components.Button("Vote for the Best Response", elem_id="vote_button")
+        gr.outputs.Textbox(label="Bot 2 Response")
     ],
     live=True
 )
 
+iface.add_component(gr.inputs.Radio(["Bot 1", "Bot 2"], label="Vote for the Best Response"), "vote")
+iface.add_component(gr.outputs.Textbox(label="Voting Result"), "vote_result")
+
+iface.update(value=update_ratings, component_name="vote", outputs=["vote_result"])
+
 iface.launch()