Update app.py

app.py

@@ -1,5 +1,3 @@
-# murder_mystery.py
-
 import random
 import gradio as gr
 from huggingface_hub import InferenceClient
@@ -54,7 +52,8 @@ game_state = {
     "turns": 0,
     "searched_locations": [],
     "eavesdropped": False,
-    "bluffed": False
+    "bluffed": False,
+    "player_questions": []
 }
 
 def initialize_game():
@@ -69,6 +68,7 @@ def initialize_game():
     game_state["searched_locations"] = []
     game_state["eavesdropped"] = False
     game_state["bluffed"] = False
+    game_state["player_questions"] = []
     random.shuffle(possible_personalities)
     alibi_locations = random.sample(locations, len(suspect_names))
     motives = ["inheritance", "jealousy", "revenge", "secret affair", "business rivalry", "blackmail"]
@@ -86,7 +86,9 @@ def initialize_game():
             "alibi_with": [],
             "knowledge": {},
             "motive": random.choice(motives),
-            "relationships": {}
+            "relationships": {},
+            "backstory": "",
+            "suspects": None
         }
         suspects[suspect_name] = suspect
     for suspect in suspects.values():
@@ -96,19 +98,20 @@ def initialize_game():
         other_suspects = [s["name"] for s in suspects.values() if s["name"] != suspect["name"]]
         for other in other_suspects:
             suspect["relationships"][other] = random.choice(relationships)
-        # Add knowledge about others' locations
+        suspect["suspects"] = random.choice([s for s in other_suspects if s not in suspect["alibi_with"]])
+        suspect["backstory"] = generate_backstory(suspect)
         suspect["knowledge"]["others_locations"] = generate_others_locations_knowledge(suspect)
     for suspect in suspects.values():
         suspect["knowledge"]["self"] = generate_knowledge(suspect)
-    print(f"Debug: Murderer is {game_state['murderer']}, Weapon is {game_state['weapon']}, Location is {game_state['location']}")
-    for suspect in suspects.values():
-        print(f"Debug: {suspect['name']} - Personality: {suspect['personality']}, Hot-headed: {suspect['hot_headed']}, Anger Threshold: {suspect['anger_threshold']}, Motive: {suspect['motive']}, Relationships: {suspect['relationships']}")
+
+def generate_backstory(suspect):
+    backstory = f"{suspect['name']} has a backstory involving {suspect['motive']}."
+    return backstory
 
 def generate_knowledge(suspect):
-    details = f"You have a motive of {suspect['motive']}."
+    details = f"You have a motive of {suspect['motive']}. {suspect['backstory']}"
     relationship_details = ". ".join([f"You are {relation} with {other}" for other, relation in suspect["relationships"].items()])
     if suspect["is_murderer"]:
-        # Murderer lies about their alibi
         suspect["alibi_location"] = random.choice([loc for loc in locations if loc != game_state["location"]])
         suspect["alibi_with"] = []
         knowledge = f"You are the murderer. Lie about your alibi and deflect suspicion. {details} {relationship_details}"
@@ -127,13 +130,10 @@ def generate_others_locations_knowledge(suspect):
             continue
         chance = random.random()
         if chance < 0.5:
-            # They know where the other was
             knowledge[other["name"]] = f"was in the {other['alibi_location']}."
         elif chance < 0.75:
-            # They are unsure
             knowledge[other["name"]] = "you are not sure where they were."
         else:
-            # They don't know
             knowledge[other["name"]] = "you have no idea where they were."
     return knowledge
 
@@ -147,6 +147,7 @@ def analyze_tone(player_input):
 def get_ai_response(suspect_name, player_input):
     suspect = suspects[suspect_name]
     game_state["turns"] += 1
+    game_state["player_questions"].append(player_input)
     tone = analyze_tone(player_input)
     if "accuse" in player_input.lower() or any(word in player_input.lower() for word in ["murderer", "kill", "guilty"]):
         suspect["trust_level"] -= 1
@@ -158,16 +159,20 @@ def get_ai_response(suspect_name, player_input):
         suspect["trust_level"] -= 1
     else:
         suspect["anger_level"] += 1
+    if suspect["anger_level"] >= suspect["anger_threshold"]:
+        suspect["personality"] = "agitated and defensive"
     personality = suspect["personality"]
     knowledge = suspect["knowledge"]["self"]
     others_knowledge = suspect["knowledge"]["others_locations"]
     anger_level = suspect["anger_level"]
     hot_headed = suspect["hot_headed"]
     trust_level = suspect["trust_level"]
+    previous_questions = " ".join(game_state["player_questions"][-3:])
+    clues_found = " ".join(game_state["clues"])
     system_prompt = f"You are {suspect_name}, who is {personality}. {knowledge}"
-    # Include knowledge about others
     for other_name, info in others_knowledge.items():
         system_prompt += f" You know that {other_name} {info}"
+    system_prompt += f" You suspect that {suspect['suspects']} might be involved due to {suspect['motive']}."
     if hot_headed and anger_level >= suspect["anger_threshold"]:
         system_prompt += f" You are extremely angry and may respond aggressively."
     if suspect["is_murderer"]:
@@ -176,7 +181,8 @@ def get_ai_response(suspect_name, player_input):
         system_prompt += " You trust the detective and are willing to share information."
     elif trust_level <= 3:
         system_prompt += " You do not trust the detective and may withhold information."
-    user_message = f"The detective asks: \"{player_input}\" As {suspect_name}, respond in first person, staying in character."
+    system_prompt += f" Previously, the detective asked: {previous_questions}"
+    user_message = f"The detective asks: \"{player_input}\" As {suspect_name}, respond in first person, staying in character. Provide a detailed response, and consider any previous interactions. If you lie, include subtle hints like hesitations or contradictions."
     messages = [
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": user_message}
@@ -274,7 +280,7 @@ def eavesdrop():
     game_state["eavesdropped"] = True
     game_state["turns"] += 1
     suspect1, suspect2 = random.sample(suspect_names, 2)
-    conversation = f"{suspect1} whispers to {suspect2}: 'I hope the detective doesn't find out about our {suspects[suspect1]['relationships'][suspect2]}.'"
+    conversation = f"{suspect1} whispers to {suspect2}: 'I think {suspects[suspect1]['suspects']} might be involved. They had a motive because of {suspects[suspects[suspect1]['suspects']]['motive']}.'"
     extract_clues(conversation)
     return f"You overhear a conversation: {conversation}"
 
@@ -295,7 +301,6 @@ def bluff(player_input):
 
 def analyze_response():
     game_state["turns"] += 1
-    # For simplicity, we'll simulate the outcome
     success = random.choice([True, False])
     if success:
         return "You successfully detect that the suspect is lying!"
@@ -366,7 +371,9 @@ def get_debug_info():
         info += f"Alibi Location: {suspect['alibi_location']}\n"
         info += f"Alibi With: {', '.join(suspect['alibi_with'])}\n"
         info += f"Motive: {suspect['motive']}\n"
+        info += f"Suspects: {suspect['suspects']}\n"
         info += f"Relationships: {suspect['relationships']}\n"
+        info += f"Backstory: {suspect['backstory']}\n"
         info += f"Knowledge: {suspect['knowledge']}\n\n"
         debug_info += info
     return debug_info
@@ -382,14 +389,14 @@ with gr.Blocks() as demo:
             with gr.Row():
                 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("- **Search a location**: Type *\"Search [Location]\"*.\n  Example: *\"Search Kitchen\"*")
+                    gr.Markdown("- **Ask a question to a suspect**: Include the suspect's name in your question.")
+                    gr.Markdown("- **Interrogate multiple suspects**: Include multiple suspects' names in your question.")
+                    gr.Markdown("- **Search a location**: Type *\"Search [Location]\"*.")
                     gr.Markdown("- **Eavesdrop on suspects**: Type *\"Eavesdrop\"*")
                     gr.Markdown("- **Bluff a suspect**: Type *\"Bluff [Suspect]\"*")
                     gr.Markdown("- **Analyze a response**: Type *\"Analyze\"*")
                     gr.Markdown("- **Reveal your deductions**: Type *\"Reveal: [Your deductions]\"*")
-                    gr.Markdown("- **Make an accusation**: Include the word *\"accuse\"*.\n  Example: *\"I accuse Colonel Mustard with the Rope in the Study.\"*")
+                    gr.Markdown("- **Make an accusation**: Include the word *\"accuse\"*.")
                     player_input = gr.Textbox(lines=1, label="Your Input")
                     send_button = gr.Button("Send")
                     restart_button = gr.Button("Restart Game")