Update app.py

app.py

@@ -21,15 +21,19 @@ def chatbot(input_text, mentioned_person='Mediator John Haynes'):
     prompt = f"You are {mentioned_person}: {input_text}\n\n At the end of your answer  ask a provocative question."
     response = index.query(prompt, response_mode="compact")
 
-    # Save chat log
-    current_time = datetime.datetime.now()
-    current_time_str = current_time.strftime("%Y-%m-%d_%H-%M-%S")
-    chat_log_filename = "chathistory.txt"
-    chat_log_filepath = os.path.join('chathistory', chat_log_filename)
-    with open(chat_log_filepath, "a") as f:
-        f.write(f"Chat at {current_time_str}\n")
-        f.write(f"User: {input_text}\n")
-        f.write(f"Chatbot: {response.response}\n\n")
+# Save chat log
+current_time = datetime.datetime.now()
+current_time_str = current_time.strftime("%Y-%m-%d_%H-%M-%S")
+chat_log_filename = "chathistory.txt"
+chat_log_dir = "chathistory"
+if not os.path.exists(chat_log_dir):
+    os.makedirs(chat_log_dir)
+chat_log_filepath = os.path.join(chat_log_dir, chat_log_filename)
+with open(chat_log_filepath, "a") as f:
+    f.write(f"Chat at {current_time_str}\n")
+    f.write(f"User: {input_text}\n")
+    f.write(f"Chatbot: {response.response}\n\n")
+
     
     return response.response