Update app.py

app.py

@@ -84,7 +84,7 @@ def initialize_game():
             "trust_level": 5,
             "alibi_location": alibi_locations[i],
             "alibi_with": [],
-            "knowledge": "",
+            "knowledge": {},
             "motive": random.choice(motives),
             "relationships": {}
         }
@@ -96,8 +96,10 @@ 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["knowledge"]["others_locations"] = generate_others_locations_knowledge(suspect)
     for suspect in suspects.values():
-        suspect["knowledge"] = generate_knowledge(suspect)
+        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']}")
@@ -106,6 +108,7 @@ def generate_knowledge(suspect):
     details = f"You have a motive of {suspect['motive']}."
     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}"
@@ -117,6 +120,23 @@ def generate_knowledge(suspect):
             knowledge += f" You know that the location of the murder was the {game_state['location']}."
     return knowledge
 
+def generate_others_locations_knowledge(suspect):
+    knowledge = {}
+    for other in suspects.values():
+        if other["name"] == suspect["name"]:
+            continue
+        chance = random.random()
+        if chance < 0.5:
+            # They know where the other was
+            knowledge[other["name"]] = f"They were 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
+
 def get_ai_response(suspect_name, player_input):
     suspect = suspects[suspect_name]
     game_state["turns"] += 1
@@ -132,11 +152,15 @@ def get_ai_response(suspect_name, player_input):
     else:
         suspect["anger_level"] += 1
     personality = suspect["personality"]
-    knowledge = suspect["knowledge"]
+    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"]
     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}"
     if hot_headed and anger_level >= suspect["anger_threshold"]:
         system_prompt += f" You are extremely angry and may respond aggressively."
     if suspect["is_murderer"]:
@@ -153,13 +177,6 @@ def get_ai_response(suspect_name, player_input):
     response = llm_inference(messages)
     return response.strip()
 
-def analyze_tone(player_input):
-    accusatory_words = ["did you", "murderer", "kill", "guilty", "crime"]
-    if any(word in player_input.lower() for word in accusatory_words):
-        return "accusatory"
-    else:
-        return "neutral"
-
 def get_group_response(suspect_names_list, player_input):
     responses = []
     for suspect_name in suspect_names_list:
@@ -273,11 +290,8 @@ def analyze_response():
     game_state["turns"] += 1
     puzzle = "Solve this riddle to analyze the suspect's response: What has keys but can't open locks?"
     correct_answer = "piano"
-    player_answer = gr.Textbox(placeholder="Your Answer").launch()
-    if player_answer.strip().lower() == correct_answer:
-        return "You successfully detect that the suspect is lying!"
-    else:
-        return "You fail to detect any lies."
+    # For the purpose of this code, we'll assume the player always fails
+    return "You fail to detect any lies."
 
 def handle_accusation(player_input):
     suspect_guess = None