Update app.py

app.py

@@ -22,57 +22,27 @@ def llm_inference(messages):
         response += choice['message']['content']
     return response
 
-suspects = {
-    "Colonel Mustard": {
-        "personality": "",
-        "knowledge": "",
-        "is_murderer": False
-    },
-    "Miss Scarlett": {
-        "personality": "",
-        "knowledge": "",
-        "is_murderer": False
-    },
-    "Professor Plum": {
-        "personality": "",
-        "knowledge": "",
-        "is_murderer": False
-    },
-    "Mrs. Peacock": {
-        "personality": "",
-        "knowledge": "",
-        "is_murderer": False
-    },
-    "Mr. Green": {
-        "personality": "",
-        "knowledge": "",
-        "is_murderer": False
-    },
-    "Dr. Orchid": {
-        "personality": "",
-        "knowledge": "",
-        "is_murderer": False
-    }
-}
-
+suspect_names = ["Colonel Mustard", "Miss Scarlett", "Professor Plum", "Mrs. Peacock", "Mr. Green", "Dr. Orchid"]
 weapons = ["Candlestick", "Dagger", "Lead Pipe", "Revolver", "Rope", "Wrench", "Poison"]
 locations = ["Kitchen", "Ballroom", "Conservatory", "Dining Room", "Library", "Study", "Billiard Room", "Lounge"]
 
 possible_personalities = [
-    "stern and suspicious",
-    "charming but evasive",
-    "intellectual and nervous",
-    "gracious but secretive",
-    "amiable yet deceptive",
-    "hot-headed and impulsive",
-    "calm and collected",
-    "mysterious and aloof",
-    "jovial but cunning",
-    "nervous and jittery",
-    "sarcastic and witty",
-    "arrogant and dismissive"
+    {"description": "stern and suspicious", "hot_headed": False},
+    {"description": "charming but evasive", "hot_headed": False},
+    {"description": "intellectual and nervous", "hot_headed": False},
+    {"description": "gracious but secretive", "hot_headed": False},
+    {"description": "amiable yet deceptive", "hot_headed": False},
+    {"description": "hot-headed and impulsive", "hot_headed": True},
+    {"description": "calm and collected", "hot_headed": False},
+    {"description": "mysterious and aloof", "hot_headed": False},
+    {"description": "jovial but cunning", "hot_headed": False},
+    {"description": "nervous and jittery", "hot_headed": False},
+    {"description": "sarcastic and witty", "hot_headed": False},
+    {"description": "arrogant and dismissive", "hot_headed": True}
 ]
 
+suspects = {}
+
 game_state = {
     "murderer": "",
     "weapon": "",
@@ -84,46 +54,62 @@ game_state = {
 }
 
 def initialize_game():
-    game_state["murderer"] = random.choice(list(suspects.keys()))
+    game_state["murderer"] = random.choice(suspect_names)
     game_state["weapon"] = random.choice(weapons)
     game_state["location"] = random.choice(locations)
     game_state["is_game_over"] = False
     game_state["clues"] = []
     game_state["history"] = []
     game_state["accused"] = False
-    personalities_copy = possible_personalities.copy()
-    random.shuffle(personalities_copy)
-    for i, suspect in enumerate(suspects):
-        suspects[suspect]["is_murderer"] = (suspect == game_state["murderer"])
-        suspects[suspect]["personality"] = personalities_copy[i % len(personalities_copy)]
-        suspects[suspect]["knowledge"] = generate_knowledge(suspect)
+    random.shuffle(possible_personalities)
+    alibi_locations = random.sample(locations, len(suspect_names))
+    alibis = {}
+    for i, suspect_name in enumerate(suspect_names):
+        suspect = {
+            "name": suspect_name,
+            "is_murderer": suspect_name == game_state["murderer"],
+            "personality": possible_personalities[i % len(possible_personalities)]["description"],
+            "hot_headed": possible_personalities[i % len(possible_personalities)]["hot_headed"],
+            "anger_level": 0,
+            "alibi_location": alibi_locations[i],
+            "alibi_with": [],
+            "knowledge": ""
+        }
+        suspects[suspect_name] = suspect
+    for suspect in suspects.values():
+        others_in_same_location = [s["name"] for s in suspects.values() if s["alibi_location"] == suspect["alibi_location"] and s["name"] != suspect["name"]]
+        suspect["alibi_with"] = others_in_same_location
+    for suspect in suspects.values():
+        suspect["knowledge"] = 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:
-        print(f"Debug: {suspect}'s personality is {suspects[suspect]['personality']}")
-
-def generate_knowledge(suspect_name):
-    knowledge = []
-    if suspects[suspect_name]["is_murderer"]:
-        knowledge.append("You are the murderer. Deflect suspicion subtly.")
+    for suspect in suspects.values():
+        print(f"Debug: {suspect['name']} - Personality: {suspect['personality']}, Hot-headed: {suspect['hot_headed']}, Alibi: {suspect['alibi_location']} with {', '.join(suspect['alibi_with'])}")
+
+def generate_knowledge(suspect):
+    if suspect["is_murderer"]:
+        suspect["alibi_location"] = random.choice([loc for loc in locations if loc != game_state["location"]])
+        suspect["alibi_with"] = []
+        knowledge = "You are the murderer. Lie about your alibi and deflect suspicion. You have no knowledge of the real weapon or location."
     else:
-        known_elements = []
-        elements = ["weapon", "location", "murderer"]
-        random.shuffle(elements)
-        known_item = elements[0]
-        if known_item == "weapon":
-            known_elements.append(f"You know that the weapon is {game_state['weapon']}.")
-        elif known_item == "location":
-            known_elements.append(f"You know that the location is {game_state['location']}.")
-        elif known_item == "murderer":
-            not_murderer = random.choice([s for s in suspects if s != suspect_name and s != game_state["murderer"]])
-            known_elements.append(f"You know that {not_murderer} is not the murderer.")
-        knowledge.extend(known_elements)
-    return ' '.join(knowledge)
+        knowledge = f"You were in the {suspect['alibi_location']} with {', '.join(suspect['alibi_with'])} during the murder."
+        if random.choice([True, False]):
+            knowledge += f" You know that the weapon is the {game_state['weapon']}."
+        if random.choice([True, False]):
+            knowledge += f" You know that the location of the murder was the {game_state['location']}."
+    return knowledge
 
 def get_ai_response(suspect_name, player_input):
-    personality = suspects[suspect_name]["personality"]
-    knowledge = suspects[suspect_name]["knowledge"]
-    system_prompt = f"You are {suspect_name}, who is {personality}. {knowledge} When answering questions, you may reveal hints subtly based on what you know."
+    suspect = suspects[suspect_name]
+    suspect["anger_level"] += 1
+    personality = suspect["personality"]
+    knowledge = suspect["knowledge"]
+    anger_level = suspect["anger_level"]
+    hot_headed = suspect["hot_headed"]
+    system_prompt = f"You are {suspect_name}, who is {personality}. {knowledge}"
+    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."
     messages = [
         {"role": "system", "content": system_prompt},
@@ -144,10 +130,10 @@ def process_input(player_input):
         game_state["history"].append(("System", result))
         return result, game_state["history"]
     else:
-        for suspect in suspects:
-            if suspect.lower() in player_input.lower():
-                ai_response = get_ai_response(suspect, player_input)
-                game_state["history"].append((suspect, ai_response))
+        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"]
@@ -160,9 +146,9 @@ def extract_clues(ai_response):
     for location in locations:
         if location.lower() in ai_response.lower() and location != game_state["location"]:
             clues.append(f"The location is not the {location}.")
-    for suspect in suspects:
-        if suspect.lower() in ai_response.lower() and suspect != game_state["murderer"]:
-            clues.append(f"{suspect} is not the murderer.")
+    for suspect_name in suspect_names:
+        if suspect_name.lower() in ai_response.lower() and suspect_name != game_state["murderer"]:
+            clues.append(f"{suspect_name} is not the murderer.")
     for clue in clues:
         if clue not in game_state["clues"]:
             game_state["clues"].append(clue)
@@ -171,9 +157,9 @@ def handle_accusation(player_input):
     suspect_guess = None
     weapon_guess = None
     location_guess = None
-    for suspect in suspects:
-        if suspect.lower() in player_input.lower():
-            suspect_guess = suspect
+    for suspect_name in suspect_names:
+        if suspect_name.lower() in player_input.lower():
+            suspect_guess = suspect_name
             break
     for weapon in weapons:
         if weapon.lower() in player_input.lower():
@@ -200,36 +186,60 @@ def chat_interface(player_input):
     clues_display = "\n".join(game_state["clues"])
     return chat_history, clues_display
 
+def get_debug_info():
+    debug_info = ""
+    debug_info += f"Murderer: {game_state['murderer']}\n"
+    debug_info += f"Weapon: {game_state['weapon']}\n"
+    debug_info += f"Location: {game_state['location']}\n\n"
+    for suspect in suspects.values():
+        info = f"Name: {suspect['name']}\n"
+        info += f"Personality: {suspect['personality']}\n"
+        info += f"Hot-headed: {suspect['hot_headed']}\n"
+        info += f"Anger Level: {suspect['anger_level']}\n"
+        info += f"Is Murderer: {suspect['is_murderer']}\n"
+        info += f"Alibi Location: {suspect['alibi_location']}\n"
+        info += f"Alibi With: {', '.join(suspect['alibi_with'])}\n"
+        info += f"Knowledge: {suspect['knowledge']}\n\n"
+        debug_info += info
+    return debug_info
+
 def restart_game():
     initialize_game()
-    return "", ""
+    return "", "", ""
 
 with gr.Blocks() as demo:
     gr.Markdown("# Murder Mystery Game")
-    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("- **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")
-            restart_button = gr.Button("Restart Game")
-            chatbox = gr.Textbox(lines=15, label="Chat History", interactive=False)
-        with gr.Column():
-            gr.Markdown("## Suspects")
-            gr.Markdown(', '.join(suspects.keys()))
-            gr.Markdown("## Weapons")
-            gr.Markdown(', '.join(weapons))
-            gr.Markdown("## Locations")
-            gr.Markdown(', '.join(locations))
-            clues = gr.Textbox(lines=15, label="Discovered Clues", interactive=False)
+    with gr.Tabs():
+        with gr.TabItem("Game"):
+            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("- **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")
+                    restart_button = gr.Button("Restart Game")
+                    chatbox = gr.Textbox(lines=15, label="Chat History", interactive=False)
+                with gr.Column():
+                    gr.Markdown("## Suspects")
+                    gr.Markdown(', '.join(suspect_names))
+                    gr.Markdown("## Weapons")
+                    gr.Markdown(', '.join(weapons))
+                    gr.Markdown("## Locations")
+                    gr.Markdown(', '.join(locations))
+                    clues = gr.Textbox(lines=15, label="Discovered Clues", interactive=False)
+        with gr.TabItem("Debug"):
+            debug_info = gr.Textbox(lines=30, label="Debug Information", interactive=False)
+            update_debug_button = gr.Button("Update Debug Info")
+            update_debug_button.click(fn=lambda: get_debug_info(), inputs=None, outputs=debug_info)
 
     def send_message(player_input):
         chat_history, clues_display = chat_interface(player_input)
-        return chat_history, "", clues_display
+        debug_information = get_debug_info()
+        return chat_history, "", clues_display, debug_information
 
-    send_button.click(send_message, inputs=player_input, outputs=[chatbox, player_input, clues])
-    restart_button.click(fn=restart_game, inputs=None, outputs=[chatbox, clues])
+    send_button.click(send_message, inputs=player_input, outputs=[chatbox, player_input, clues, debug_info])
+    restart_button.click(fn=restart_game, inputs=None, outputs=[chatbox, clues, debug_info])
 
 initialize_game()
 demo.launch()