Update logging

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import datetime
 from nim_game_env import NimGameEnv
 from nim_gpt_functions import plan_move, execute_move
 
@@ -55,6 +56,9 @@ def send_chat_msg(inp, chat_history, nim_game_env, temperature, openai_api_key):
         text_obs, observation, reward, done, info = execute_move(output, nim_game_env, openai_api_key)
         ascii_art = generate_game_state_ascii_art(observation, done, reward, AI_STR)
 
+    print("\n==== date/time: " + str(datetime.datetime.now() - datetime.timedelta(hours=5)) + " ====")
+    print("inp: " + inp, ", output: ", output, ", observation: ", observation)
+
     chat_history.append((HUMAN_STR + ": " + inp, AI_STR + ": " + output))
     return chat_history, chat_history, ascii_art
 

nim_gpt_functions.py

@@ -53,7 +53,7 @@ def plan_move(text_game_state, temperature, api_key):
     llm = OpenAI(model_name='text-davinci-003', temperature=temperature, max_tokens=100,
                  openai_api_key=api_key)
     llm_chain = LLMChain(llm=llm, prompt=PLAN_MOVE_PROMPT_FROM_STRING_EXAMPLES, verbose=False)
-    planned_move = llm_chain.run({'text_game_state': text_game_state})
+    planned_move = llm_chain.run({'text_game_state': text_game_state}).strip()
     return planned_move