Merge pull request #4 from gmook9/dev

Fixed game logic and styling

Author: gmook9

ai_bot.py

@@ -1,42 +1,50 @@
-import os
 from random import choice
 from langchain_community.llms import Ollama
-from dotenv import load_dotenv
 
-# Load environment variables from .env file
-load_dotenv()
+# TO-DO FIX THIS
+class QuestionGenerator:
+    def __init__(self, ai_bot):
+        self.ai_bot = ai_bot
+
+    def generate_random_question(self, case_synopsis):
+        # Use the provided case synopsis instead of generating a new one
+        question_prompt = (
+            f"You are playing the role of a detective in a text-based detective game. Below is a fictional case synopsis:\n\n"
+            f"'{case_synopsis}'\n\n"
+            f"Your task is to generate a question that is appropriate for this scenario. "
+            f"Remember, this is a fictional game setting, so avoid any references to real-life guidance or advice. "
+            f"The question should relate only to the details provided in the synopsis and should aim to uncover more information about the case.\n"
+            f"Please provide only the question."
+        )
+        return self.ai_bot.llm.invoke(question_prompt)
 
 class AIBot:
     def __init__(self):
         self.llm = Ollama(model="llama3")
         self.role = choice(["guilty", "innocent"])
-        self.prompt = os.getenv("PROMPT")
+        self.prompt = "Generate a random crime scenario for a detective game in 3-5 sentences. Describe a situation that has just happened. At the end of your response, specify your role as the person being interviewed at the police station. You might be a witness, bystander, or actively involved, and you could be innocent or guilty. For example: (Interviewee: Wife). Also do not put anything like (Here's a random crime scenario:) in it."
         self.conversation_history = []  # Store previous questions and answers
+        self.question_generator = QuestionGenerator(self)  # Initialize the QuestionGenerator with AIBot
 
     def generate_synopsis(self):
         return self.llm.invoke(self.prompt)
 
     def respond(self, question):
-        self.conversation_history.append(f"Question: {question}") # Add question to convo history
+        self.conversation_history.append(f"Question: {question}")  # Add question to convo history
         context = "\n".join(self.conversation_history)
         response_prompt = (
             f"Role: {self.role}. You are being questioned. "
             f"Here is the context so far:\n{context}\n\n"
-            f"Now respond to the latest question: {question}"
+            f"Now respond to the latest question: {question}. Only response in 1-2 setnences."
         )
-        response = self.llm.invoke(response_prompt) # Get AI Response
-        self.conversation_history.append(f"Response: {response}") # Add response to convo history
+        response = self.llm.invoke(response_prompt)  # Get AI Response
+        self.conversation_history.append(f"Response: {response}")  # Add response to convo history
 
         return response
 
     def is_guilty(self):
         return self.role == "guilty"
 
-    def generate_random_question(self):
-        question_prompt = (
-            "Generate a detective-style question related to a crime investigation, similar to these examples: "
-            "\"Where were you on the night of the crime?\", "
-            "\"Do you have an alibi?\", "
-            "\"Why were you near the crime scene?\""
-        )
-        return self.llm.invoke(question_prompt)
+    # Modify this method to accept synopsis as an argument
+    def generate_random_question(self, case_synopsis):
+        return self.question_generator.generate_random_question(case_synopsis)

game.py

@@ -10,13 +10,16 @@ def __init__(self):
         self.max_questions = 9
         self.questions_asked = 0
         self.exit_game = False  # Flag to track if the game is exited
+        self.synopsis = None  # Variable to store the generated synopsis
 
     def start(self):
-        self.console.print("[bold green]Welcome, Detective![/bold green]")
-        self.console.print()  # Add space
-        synopsis = self.ai_bot.generate_synopsis()
-        self.console.print(Panel(synopsis, title="Case Synopsis"))
-        self.console.print()  # Add space
+        # Welcome message
+        self.console.print("[bold green]Welcome, Detective![/bold green]\n")
+        self.console.print("[bold yellow]Loading, please wait...[/bold yellow]\n")
+
+        self.synopsis = self.ai_bot.generate_synopsis()  # Store the synopsis
+        self.console.print(Panel(self.synopsis, title="Case Synopsis"))
+        self.console.print("\n")
 
         while self.questions_asked < self.max_questions and not self.exit_game:
             action = self.select_action()
@@ -29,71 +32,64 @@ def start(self):
                 break
 
         if not self.exit_game:
-            self.console.print()  # Add space
-            self.console.print("[bold red]Game over! You've used all your questions.[/bold red]")
+            self.console.print("[bold red]Game over! You've used all your questions.[/bold red]\n")
 
     def select_action(self):
         self.console.print("[bold blue]Select an action:[/bold blue]")
         self.console.print("1. Ask a question")
         self.console.print("2. Make a final decision")
-        self.console.print("3. Exit")
-        self.console.print()  # Add space
-        choice = input("Enter your choice (1/2/3): ").strip()
+        self.console.print("3. Exit\n")
+        choice = self.console.input("[bold yellow]Enter your choice (1/2/3): [/bold yellow]").strip()
         if choice == "1":
             return "Ask a question"
         elif choice == "2":
             return "Make a final decision"
         elif choice == "3":
             return "Exit"
         else:
-            self.console.print("[bold red]Invalid choice. Please try again.[/bold red]")
-            self.console.print()  # Add space
+            self.console.print("[bold red]Invalid choice. Please try again.[/bold red]\n")
             return self.select_action()
 
     def ask_question(self):
-        self.console.print("[bold blue]Question Type:[/bold blue]")
-        self.console.print("1. Type my own")
-        self.console.print("2. Random")
-        self.console.print()  # Add space
-        choice = input("Enter your choice (1/2): ").strip()
+        self.console.print("\n[bold magenta]Question Type:[/bold magenta]")
+        self.console.print("1. Random")
+        self.console.print("2. Type my own\n")
+        choice = self.console.input("[bold yellow]Enter your choice (1/2): [/bold yellow]").strip()
 
         if choice == "1":
-            user_question = input("Type your question below: ").strip()
+            user_question = self.ai_bot.generate_random_question(self.synopsis)  # Pass the stored synopsis
         elif choice == "2":
-            user_question = self.ai_bot.generate_random_question()
+            user_question = self.console.input("Type your question below: ").strip()
         else:
-            self.console.print("[bold red]Invalid choice. Please try again.[/bold red]")
-            self.console.print()  # Add space
+            self.console.print("[bold red]Invalid choice. Please try again.[/bold red]\n")
             return self.ask_question()
 
         self.questions_asked += 1
-        self.console.print()  # Add space
+        self.console.print("\n")
         self.console.print(Text(f"Question {self.questions_asked}: {user_question}", style="bold red"))
         response = self.ai_bot.respond(user_question)
         self.console.print(Panel(response, title=f"Response [{self.questions_asked}/{self.max_questions}]"))
-        self.console.print()  # Add space
+        self.console.print("\n")
 
     def make_final_decision(self):
-        self.console.print("[bold blue]Final Decision:[/bold blue]")
+        self.console.print("\n[bold bright_magenta]Final Decision:[/bold bright_magenta]")
         self.console.print("1. Release")
-        self.console.print("2. Throw behind bars")
-        self.console.print()  # Add space
-        choice = input("Enter your choice (1/2): ").strip()
+        self.console.print("2. Throw behind bars\n")
+        choice = self.console.input("[bold yellow]Enter your choice (1/2): [/bold yellow]").strip()
 
         if choice == "1":
             decision = "Release"
         elif choice == "2":
             decision = "Throw behind bars"
         else:
-            self.console.print("[bold red]Invalid choice. Please try again.[/bold red]")
-            self.console.print()  # Add space
+            self.console.print("[bold red]Invalid choice. Please try again.[/bold red]\n")
             return self.make_final_decision()
 
-        self.console.print()  # Add space
+        self.console.print("\n")
         if self.ai_bot.is_guilty() and decision == "Throw behind bars":
-            self.console.print("[bold green]Correct! The suspect was guilty.[/bold green]")
+            self.console.print("[bold green]Correct! The suspect was guilty.[/bold green]\n")
         elif not self.ai_bot.is_guilty() and decision == "Release":
-            self.console.print("[bold green]Correct! The suspect was innocent.[/bold green]")
+            self.console.print("[bold green]Correct! The suspect was innocent.[/bold green]\n")
         else:
-            self.console.print("[bold red]Wrong decision! You lost.[/bold red]")
+            self.console.print("[bold red]Wrong decision! You lost.[/bold red]\n")
         exit()