Update app.py

app.py

@@ -70,7 +70,15 @@ agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=False)
 
 client = InferenceClient(token=os.environ["HUGGINGFACEHUB_API_TOKEN"])
 
+LOG_PATH = "/mnt/data/chat_log.txt"
+
+def log_message(role: str, text: str):
+    with open(LOG_PATH, "a", encoding="utf-8") as f:
+        f.write(f"{role.upper()}: {text}\n\n")
+
 def respond(message, history: list[dict]):
+
+    log_message("user", message)
     
     agent_output = agent_executor.invoke({
         "query": message,
@@ -93,6 +101,7 @@ def respond(message, history: list[dict]):
         assistant_text = raw
 
     response = {"role": "assistant", "content": assistant_text}
+    log_message("assistant", assistant_text)
 
     yield response