app.py

# murder_mystery.py

import random
import gradio as gr
from huggingface_hub import InferenceClient

model_name = "Qwen/Qwen2.5-72B-Instruct"
client = InferenceClient(model_name)

def llm_inference(messages):
    eos_token = "<|endoftext|>"
    output = client.chat.completions.create(
        messages=messages,
        stream=False,
        temperature=0.7,
        top_p=0.1,
        max_tokens=512,
        stop=[eos_token]
    )
    response = ''
    for choice in output.choices:
        response += choice['message']['content']
    return response

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 = [
    {"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": "",
    "location": "",
    "is_game_over": False,
    "clues": [],
    "history": [],
    "accused": False
}

def initialize_game():
    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
    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.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:
        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):
    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},
        {"role": "user", "content": user_message}
    ]
    response = llm_inference(messages)
    return response.strip()

def process_input(player_input):
    if game_state["is_game_over"]:
        return "The game is over. Please restart to play again.", game_state["history"]
    if game_state["accused"]:
        return "You have already made an accusation. Please restart to play again.", game_state["history"]
    game_state["history"].append(("Detective", player_input))
    if "accuse" in player_input.lower():
        game_state["accused"] = True
        result = handle_accusation(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"]

def extract_clues(ai_response):
    clues = []
    for weapon in weapons:
        if weapon.lower() in ai_response.lower() and weapon != game_state["weapon"]:
            clues.append(f"The weapon is not the {weapon}.")
    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_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)

def handle_accusation(player_input):
    suspect_guess = None
    weapon_guess = None
    location_guess = None
    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():
            weapon_guess = weapon
            break
    for location in locations:
        if location.lower() in player_input.lower():
            location_guess = location
            break
    if (suspect_guess == game_state["murderer"] and
        weapon_guess == game_state["weapon"] and
        location_guess == game_state["location"]):
        game_state["is_game_over"] = True
        return f"Congratulations! You solved the case. It was {suspect_guess} with the {weapon_guess} in the {location_guess}."
    else:
        game_state["is_game_over"] = True
        return f"Incorrect accusation! You lose. The real murderer was {game_state['murderer']} with the {game_state['weapon']} in the {game_state['location']}."

def chat_interface(player_input):
    response, history = process_input(player_input)
    chat_history = ""
    for speaker, text in history:
        chat_history += f"**{speaker}:** {text}\n\n"
    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 "", "", ""

with gr.Blocks() as demo:
    gr.Markdown("# Murder Mystery Game")
    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)
        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, debug_info])
    restart_button.click(fn=restart_game, inputs=None, outputs=[chatbox, clues, debug_info])

initialize_game()
demo.launch()