Create memory.py

modules/memory.py

@@ -0,0 +1,59 @@
+# modules/memory.py
+
+from datetime import datetime
+from modules.data_helpers import smart_label_converter
+from modules.task_management import roadmap_unlock
+from modules.analysis import render_text_roadmap
+from utils.constants import course_suggestions
+from utils.api_clients import pine_index
+from utils.embedding_model import initialize_embedding_model
+
+embed_model = initialize_embedding_model()
+
+def save_to_memory(user_id, goal, summary, steps, courses):
+    try:
+        text_blob = f"Goal: {goal}\nSummary: {summary}\nSteps: {' | '.join(steps)}\nCourses: {' | '.join([c[0] for c in courses])}"
+        embedding = embed_model.embed_query(text_blob)
+        metadata = {
+            "user_id": user_id,
+            "goal": goal,
+            "summary": summary,
+            "steps": steps,
+            "courses": [f"{c[0]} | {c[1]}" for c in courses],
+            "timestamp": datetime.utcnow().isoformat()
+        }
+        pine_index.upsert([(user_id + ":" + goal.replace(" ", "_"), embedding, metadata)])
+        return True
+    except Exception as e:
+        print(f"❌ Failed to save memory: {e}")
+        return False
+
+
+def recall_from_memory(user_id, goal):
+    try:
+        query = user_id + ":" + goal.replace(" ", "_")
+        result = pine_index.fetch([query])
+        if query not in result.vectors:
+            return "❌ No saved plan found for this goal."
+
+        metadata = result.vectors[query].get("metadata", {})
+        steps = metadata.get("steps", [])
+        steps = [smart_label_converter(s) for s in steps if isinstance(s, str) and len(s.strip()) > 1]
+        summary = metadata.get("summary", "")
+        courses = metadata.get("courses", [])
+        diagram = render_text_roadmap(goal, steps)
+
+        course_section = "\n\n### 📚 Recommended Courses\n" + "\n".join(
+            [f"- [{c['name']}]({c['url']})" for c in courses if isinstance(c, dict) and 'name' in c and 'url' in c]
+        ) if courses else ""
+
+        return f"""### 🔁 Recalled Plan for {goal}
+
+{diagram}
+
+{summary}{course_section}
+
+**🗓 Book your weekly study check-in:** [Click here](https://calendly.com)
+"""
+    except Exception as e:
+        return f"❌ Error recalling memory: {e}"