Update app.py

app.py

@@ -4,7 +4,7 @@ from src.utils.hf_packager import HFSpacePackager
 from agents.planner import plan_task
 from agents.executor import execute_step
 from agents.critic import review_result
-from memory import init_memory, add_to_memory, search_memory
+from memory import init_memory, add_to_memory, search_memory, get_memory
 import uuid
 from datetime import datetime
 import os
@@ -46,7 +46,13 @@ def status():
 
 @app.get("/generate")
 def generate(prompt: str):
-    return {"response": llm_model(prompt)}
+    response = llm_model(prompt)
+    add_to_memory(prompt, response)
+    return {"response": response}
+
+@app.get("/memory")
+def view_memory():
+    return {"history": get_memory()}
 
 # Test the model at startup (optional)
 if __name__ == "__main__":