Update app.py

app.py

@@ -1,4 +1,4 @@
-# murder_mystery.py
+
 
 import random
 import gradio as gr
@@ -118,6 +118,38 @@ def get_ai_response(suspect_name, player_input):
     response = llm_inference(messages)
     return response.strip()
 
+def get_group_response(suspect_names_list, player_input):
+    responses = []
+    for suspect_name in suspect_names_list:
+        suspect = suspects[suspect_name]
+        suspect["anger_level"] += 1
+        personality = suspect["personality"]
+        knowledge = suspect["knowledge"]
+        anger_level = suspect["anger_level"]
+        hot_headed = suspect["hot_headed"]
+        relationships = get_relationships(suspect_name, suspect_names_list)
+        system_prompt = f"You are {suspect_name}, who is {personality}. {knowledge} {relationships}"
+        if hot_headed:
+            system_prompt += f" You are getting angrier each time you are questioned. Your anger level is {anger_level}."
+        if suspect["is_murderer"]:
+            system_prompt += " You are the murderer and will lie to protect yourself."
+        user_message = f"The detective asks: \"{player_input}\" As {suspect_name}, respond in first person, staying in character, and interact with the other suspects present."
+        messages = [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_message}
+        ]
+        response = llm_inference(messages)
+        responses.append(f"**{suspect_name}:** {response.strip()}")
+    return "\n\n".join(responses)
+
+def get_relationships(suspect_name, suspect_names_list):
+    other_suspects = [name for name in suspect_names_list if name != suspect_name]
+    if other_suspects:
+        relationships = f"You are currently with {', '.join(other_suspects)}. You might have personal opinions about them, and you may attempt to shift blame onto them."
+    else:
+        relationships = ""
+    return relationships
+
 def process_input(player_input):
     if game_state["is_game_over"]:
         return "The game is over. Please restart to play again.", game_state["history"]
@@ -130,13 +162,20 @@ def process_input(player_input):
         game_state["history"].append(("System", result))
         return result, game_state["history"]
     else:
-        for suspect_name in suspect_names:
-            if suspect_name.lower() in player_input.lower():
-                ai_response = get_ai_response(suspect_name, player_input)
-                game_state["history"].append((suspect_name, ai_response))
-                extract_clues(ai_response)
-                return ai_response, game_state["history"]
-        return "Please direct your question to a suspect.", game_state["history"]
+        mentioned_suspects = [suspect_name for suspect_name in suspect_names if suspect_name.lower() in player_input.lower()]
+        if len(mentioned_suspects) == 0:
+            return "Please direct your question to a suspect.", game_state["history"]
+        elif len(mentioned_suspects) == 1:
+            suspect_name = mentioned_suspects[0]
+            ai_response = get_ai_response(suspect_name, player_input)
+            game_state["history"].append((suspect_name, ai_response))
+            extract_clues(ai_response)
+            return ai_response, game_state["history"]
+        else:
+            ai_response = get_group_response(mentioned_suspects, player_input)
+            game_state["history"].append(("Group", ai_response))
+            extract_clues(ai_response)
+            return ai_response, game_state["history"]
 
 def extract_clues(ai_response):
     clues = []
@@ -215,6 +254,7 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     gr.Markdown("## Commands")
                     gr.Markdown("- **Ask a question to a suspect**: Include the suspect's name in your question.\n  Example: *\"Miss Scarlett, where were you during the evening?\"*")
+                    gr.Markdown("- **Interrogate multiple suspects**: Include multiple suspects' names in your question.\n  Example: *\"Colonel Mustard and Mrs. Peacock, what can you tell me about the night?\"*")
                     gr.Markdown("- **Make an accusation**: Include the word *\"accuse\"*.\n  Example: *\"I accuse Colonel Mustard with the Rope in the Study.\"*")
                     player_input = gr.Textbox(lines=1, label="Your Input")
                     send_button = gr.Button("Send")