Update chat.py

chat.py

@@ -1,6 +1,6 @@
 from time import time, sleep
 import datetime
-# import dotenv
+import dotenv
 import os
 import openai
 import json
@@ -11,11 +11,14 @@ import re
 from langchain.memory import VectorStoreRetrieverMemory
 
 ## Read the environment variables
-# dotenv.load_dotenv('.env')
+dotenv.load_dotenv('.env')
 openai.api_key = os.getenv('OPENAI_API_KEY')
 embedding_model = os.getenv('EMBEDDING_ENGINE')
 convo_length = int(os.getenv('CONVO_LENGTH_TO_FETCH'))
 llm_model = os.getenv('LLM_MODEL')
+debug=False
+if os.getenv('DEBUG') == 'True':
+    debug=True
 
 pinecone_api_key = os.getenv('PINECONE_API_KEY')
 pinecone_env = os.getenv('PINECONE_REGION')
@@ -25,6 +28,9 @@ pinecone.init(
     environment=pinecone_env 
 )
 vector_db = pinecone.Index(pinecone_index)
+file_path = os.getenv('GAME_DOCS_FOLDER')
+file_name = os.getenv('GAME_DOCS_FILE')
+game_index = os.getenv('GAME_ID_INDEX')
 
 def timestamp_to_datetime(unix_time):
     return datetime.datetime.fromtimestamp(unix_time).strftime("%A, %B %d, %Y at %I:%M%p %Z")
@@ -116,7 +122,14 @@ def start_game(game_id, user_id, user_input):
     vector_db.upsert(payload)
        
     return(llm_output)
-  
+
+def get_game_details(game_id):
+    file_data = open_file(f"{file_path}/{game_index}")
+    tmp_json = json.loads(file_data)
+    for json_item in tmp_json["game_details"]:
+        if json_item["game_id"] == game_id:
+            return json_item
+    return "Not Found"
 
 def populate_prompt(game_id, splits):
     prompt_text = list()
@@ -135,8 +148,12 @@ def populate_prompt(game_id, splits):
 def initialize_game(game_id, user_id, user_input):
     game_details = get_game_details(game_id)
     whole_prompt = populate_prompt(game_id, game_details["splits"])
+    if debug:
+        print(whole_prompt[:1000])
     whole_prompt = whole_prompt.replace("<<USER_INPUT_MSG>>", user_input)
-       
+    if debug:
+        print(whole_prompt[:1000])
+    
     llm_prompt_op = call_gpt(whole_prompt)
     #print(llm_prompt_op.choices[0]["message"]["content"])
     fname="prompt_" + game_id + "_" + user_id + ".txt"