Create smart_plan.py

modules/smart_plan.py

@@ -0,0 +1,142 @@
+import datetime
+import difflib
+import re
+
+def generate_smart_plan(user_id, start_date_str, goal):
+    start_date = datetime.datetime.strptime(start_date_str, "%Y-%m-%d").date()
+
+    if not user_id:
+        return "❌ Please enter a nickname in the Welcome tab.", "", "📅 Week 1"
+
+    user_key = user_id.strip().lower()
+    goal_key = goal.lower().strip()
+
+    if goal_key not in course_suggestions:
+        close_matches = difflib.get_close_matches(goal_key, course_suggestions.keys(), n=1, cutoff=0.6)
+        if close_matches:
+            goal_key = close_matches[0]
+
+    
+    steps = []
+    steps = [smart_label_converter(s) for s in steps if isinstance(s, str) and len(s.strip()) > 1]
+    plan_markdown = ""
+    plan_source = ""
+    summary = ""
+    courses = []
+
+    # 1️⃣ Try RAG
+    recalled = recall_from_memory(user_key, goal_key)
+    if recalled and "❌" not in recalled:
+        print("✅ RAG hit")
+        steps = recalled.split("\n")
+        steps = [smart_label_converter(s) for s in steps if isinstance(s, str) and len(s.strip()) > 1]
+        plan_markdown = (f"### 🧠 Plan for {goal}\n\n_Recalled from memory_\n\n" +"\n".join(smart_label_converter(s) for s in steps))
+        plan_source = "🔁 Loaded from memory (RAG)"
+        
+    if not steps:
+        print("🧪 Trying Tavilly fallback for:", goal_key)
+
+    try:
+        result = call_tavilly_rag(user_id, goal_key)
+        print("✅ Tavilly returned something:", result)
+
+        tav_plan = result[0]
+        tav_plan = re.sub(r'(Create a weekly roadmap.*?)\1+', r'\1', tav_plan, flags=re.DOTALL)
+
+        tav_steps = result[2]
+        tav_steps = [smart_label_converter(s) for s in tav_steps if isinstance(s, str) and len(s.strip()) > 1]
+
+
+        if tav_steps and len(tav_steps) > 1:
+            print("🌐 Tavilly used")
+            steps = tav_steps
+            plan_markdown = tav_plan
+            plan_source = "🌐 Generated via Tavilly (web search)"
+
+            courses = get_courses_for_goal(goal_key)  # <-- use the fuzzy look
+
+        if courses:
+            course_section = (
+        "\n\n### 📚 Recommended Courses\n" +
+        "\n".join([f"- [{name}]({url})" for name, url in courses]))
+            
+        else:
+            course_section = ""
+
+        plan_markdown += course_section
+        pass 
+
+    except Exception as e:
+        print(f"⚠️ Tavilly failed: {e}")
+
+
+    # 3️⃣ Try GPT-4o fallback
+    if not steps:
+        try:
+            steps = run_gpt_fallback(goal_key)
+            steps = [smart_label_converter(s) for s in steps if isinstance(s, str) and len(s.strip()) > 1]
+            print("🤖 GPT-4o fallback used")
+            plan_source = "🤖 Generated with GPT-4o fallback"
+
+            diagram = render_text_roadmap(goal, steps)
+            summary = summarizer(f"Create a weekly roadmap for {goal}.", max_new_tokens=300, do_sample=False)[0]["generated_text"]
+
+            courses = course_suggestions.get(goal_key, [])
+            if courses:
+                course_section = ("\n\n### 📚 Recommended Courses\n" +"\n".join([f"- [{name}]({url})" for name, url in courses]))
+            else:
+                course_section = ""
+
+            plan_markdown = f"""
+### 🧠 Career Plan for {goal}
+
+```
+{diagram}
+```
+
+{summary}{course_section}
+
+**🗓 Do your study check-ins yourself and weekly follow up in the Memo tab.**
+
+---
+
+_Source: {plan_source}_
+"""
+            save_to_memory(user_id, goal, summary, steps, courses)
+
+        except Exception as e:
+            print(f"❌ GPT fallback failed: {e}")
+
+    # 4️⃣ Static fallback
+    if not steps:
+        print("📦 Using static backup")
+        steps = [
+            "Action: Search Coursera for a beginner course.",
+            "Resource: Watch at least 1 hour and reflect.",
+            "Milestone: Write a journal entry on what you learned."
+        ]
+        steps = [smart_label_converter(s) for s in steps if isinstance(s, str) and len(s.strip()) > 1]
+
+
+        plan_markdown = f"""
+### 🧠 Static Plan for {goal}
+
+Start with some basic research and courses. Track progress in the Courses tab.
+
+**🗓 Do your study check-ins yourself and weekly follow up in the Memo tab.**
+
+---
+
+_Source: 📦 Static backup used_
+"""
+        plan_source = "📦 Static backup used"
+
+    # 🔁 Push steps into Memo Task List
+    for step in steps:
+        if isinstance(step, str) and len(step.strip()) > 1:
+            add_task(user_id, task=step, duration=2, difficulty="Moderate", tag=None, source="career")
+
+    print("✅ Returning", len(steps), "steps for goal:", goal)
+    return plan_markdown, gr.update(choices=steps, value=[]), "📅 Week 1"
+
+