Update app.py

app.py

@@ -47,7 +47,7 @@ def explain_word(message, history: list[tuple[str, str]],max_tokens=128, tempera
 
 
 # Function for generating debate responses
-def generate_response(llm, position, topic, message):
+def generate_response(llm, position, who, topic, message):
     system_message = {
         "role": "system",
         "content": f"You are a debate participant tasked with defending the position '{position}' on the topic '{topic}'. Your goal is to articulate your arguments with clarity, logic, and professionalism while addressing counterpoints made by the opposing side. "
@@ -59,7 +59,7 @@ def generate_response(llm, position, topic, message):
     messages = [system_message]
     messages.append({"role": "user", "content": message})
 
-    response = ""
+    response = f"{who}:\n"
     for message_chunk in llm.chat_completion(
             messages, max_tokens=128, stream=True, temperature=0.4, top_p=0.95):
         response += message_chunk.choices[0].delta.content
@@ -80,7 +80,7 @@ def start_debate(topic, position_1, position_2):
     turn = "Master-1"
     history = []  # Reset history
     initial_message = "Opening Statement"
-    response = generate_response(Master1, position_1, topic, initial_message)
+    response = generate_response(Master1, position_1, 'Master-1', topic, initial_message)
     history.append((initial_message, response))
     return f"The debate has started! {turn} begins.", history
 
@@ -94,13 +94,13 @@ def next_turn(topic, position_1, position_2, current_history):
     # Alternate turn logic
     if turn == "Master-1":
         turn = "Master-2"
-        llm, position = Master2, position_2
+        llm, position, who = Master2, position_2, 'Master-2'
     else:
         turn = "Master-1"
-        llm, position = Master1, position_1
+        llm, position, who = Master1, position_1, "Master-1"
 
     last_response = current_history[-1][1]  # Get the last message
-    response = generate_response(llm, position, topic, last_response)
+    response = generate_response(llm, position, who, topic, last_response)
     history.append(("", response))  # Add the response to history
     return f"It's now {turn}'s turn.", history
 
@@ -133,7 +133,7 @@ def debate_respond(message, history: list[tuple[str, str]],
 
     # Generating the response
     response = ""
-    for message in client.chat_completion(
+    for message in Master1.chat_completion(
             messages,
             max_tokens=max_tokens,
             stream=True,