Update app.py

app.py

@@ -14,7 +14,7 @@ def llm_inference(messages):
         stream=False,
         temperature=0.7,
         top_p=0.1,
-        max_tokens=412,
+        max_tokens=512,
         stop=[eos_token]
     )
     response = ''
@@ -47,11 +47,16 @@ suspects = {
         "personality": "",
         "knowledge": "",
         "is_murderer": False
+    },
+    "Dr. Orchid": {
+        "personality": "",
+        "knowledge": "",
+        "is_murderer": False
     }
 }
 
-weapons = ["Candlestick", "Dagger", "Lead Pipe", "Revolver", "Rope", "Wrench"]
-locations = ["Kitchen", "Ballroom", "Conservatory", "Dining Room", "Library", "Study"]
+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",
@@ -63,7 +68,9 @@ possible_personalities = [
     "calm and collected",
     "mysterious and aloof",
     "jovial but cunning",
-    "nervous and jittery"
+    "nervous and jittery",
+    "sarcastic and witty",
+    "arrogant and dismissive"
 ]
 
 game_state = {
@@ -72,7 +79,8 @@ game_state = {
     "location": "",
     "is_game_over": False,
     "clues": [],
-    "history": []
+    "history": [],
+    "accused": False
 }
 
 def initialize_game():
@@ -82,6 +90,7 @@ def initialize_game():
     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):
@@ -98,22 +107,23 @@ def generate_knowledge(suspect_name):
         knowledge.append("You are the murderer. Deflect suspicion subtly.")
     else:
         known_elements = []
-        if random.choice([True, False]):
-            known_elements.append(f"The weapon is {game_state['weapon']}.")
-        if random.choice([True, False]):
-            known_elements.append(f"The location is {game_state['location']}.")
-        if random.choice([True, False]):
+        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"The murderer is not {not_murderer}.")
-        if not known_elements:
-            known_elements.append("You don't have any specific knowledge about the murder.")
+            known_elements.append(f"You know that {not_murderer} is not the murderer.")
         knowledge.extend(known_elements)
     return ' '.join(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}"
+    system_prompt = f"You are {suspect_name}, who is {personality}. {knowledge} When answering questions, you may reveal hints subtly based on what you know."
     user_message = f"The detective asks: \"{player_input}\" As {suspect_name}, respond in first person, staying in character."
     messages = [
         {"role": "system", "content": system_prompt},
@@ -125,57 +135,37 @@ def get_ai_response(suspect_name, player_input):
 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"]
-    elif "was it" in player_input.lower() or "suggest" in player_input.lower():
-        result = handle_suggestion(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))
+                extract_clues(ai_response)
                 return ai_response, game_state["history"]
         return "Please direct your question to a suspect.", game_state["history"]
 
-def handle_suggestion(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
-            break
+def extract_clues(ai_response):
+    clues = []
     for weapon in weapons:
-        if weapon.lower() in player_input.lower():
-            weapon_guess = weapon
-            break
+        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 player_input.lower():
-            location_guess = location
-            break
-    feedback = []
-    if suspect_guess == game_state["murderer"]:
-        feedback.append("The suspect is correct.")
-    elif suspect_guess:
-        feedback.append("The suspect is incorrect.")
-    if weapon_guess == game_state["weapon"]:
-        feedback.append("The weapon is correct.")
-    elif weapon_guess:
-        feedback.append("The weapon is incorrect.")
-    if location_guess == game_state["location"]:
-        feedback.append("The location is correct.")
-    elif location_guess:
-        feedback.append("The location is incorrect.")
-    if not feedback:
-        return "Your suggestion could not be understood. Please specify a suspect, weapon, and location."
-    clue = ' '.join(feedback)
-    game_state["clues"].append(clue)
-    return clue
+        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 clue in clues:
+        if clue not in game_state["clues"]:
+            game_state["clues"].append(clue)
 
 def handle_accusation(player_input):
     suspect_guess = None
@@ -200,7 +190,7 @@ def handle_accusation(player_input):
         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! The real murderer was {game_state['murderer']} with the {game_state['weapon']} in the {game_state['location']}."
+        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)
@@ -219,8 +209,7 @@ 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 murder?\"*")
-            gr.Markdown("- **Make a suggestion**: Use phrases like *\"Was it [Suspect] with the [Weapon] in the [Location]?\"*\n  Example: *\"Was it Professor Plum with the Dagger in the Library?\"*")
+            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")