Update app.py

app.py

@@ -29,19 +29,27 @@ def write_elo_ratings(elo_ratings):
 # Function to get bot response
 def format_alpaca_prompt(state):
     alpaca_prompt = "Below is an instruction that describes a task. Write a response that appropriately completes the request."
-    for message in state["history"]:
+    alpaca_prompt2 = "Below is an instruction that describes a task. Write a response that appropriately completes the request."
+    for message in state["history"][0]:
         j=""
         if message['role']=='user':
             j="### Instruction:\n"
         else:
             j="### Response:\n"
         alpaca_prompt += j+ message['content']+"\n\n"
-    return alpaca_prompt+"### Response:\n"
-def get_bot_response(url, prompt,state):
+    for message in state["history"][1]:
+        j=""
+        if message['role']=='user':
+            j="### Instruction:\n"
+        else:
+            j="### Response:\n"
+        alpaca_prompt2 += j+ message['content']+"\n\n"
+    return [alpaca_prompt+"### Response:\n",alpaca_prompt2+"### Response:\n"]
+def get_bot_response(url, prompt,state,bot_index):
     alpaca_prompt = format_alpaca_prompt(state)
     payload = {
         "input": {
-            "prompt": alpaca_prompt,
+            "prompt": alpaca_prompt[bot_index],
             "sampling_params": {
                 "max_new_tokens": 50,
                 "temperature": 0.7,
@@ -65,8 +73,8 @@ def chat_with_bots(user_input, state):
     # Update the state with the names of the last bots
     state.update({'last_bots': [bot_names[0], bot_names[1]]})
 
-    bot1_response = get_bot_response(bot1_url, user_input,state)
-    bot2_response = get_bot_response(bot2_url, user_input,state)
+    bot1_response = get_bot_response(bot1_url, user_input,state,0)
+    bot2_response = get_bot_response(bot2_url, user_input,state,1)
 
     return bot1_response, bot2_response