Update utils/visualizer.py

utils/visualizer.py

@@ -1,20 +1,41 @@
 import matplotlib.pyplot as plt
 
-def generate_timeline(text):
-    stages = ["Skill Development", "Certifications", "Higher Education", "Job Applications", "Visa Process"]
-    durations = [2, 2, 3, 1, 1]  # Example durations in months or quarters
+def generate_timeline(cv_type, education_level, score):
+    stages = []
+    durations = []
+
+    if score < 60:
+        stages.append("Skill Development")
+        durations.append(3)
+    elif score < 80:
+        stages.append("Skill Polishing")
+        durations.append(2)
+
+    if education_level.lower() in ["high school", "bachelor"]:
+        stages.append("Certifications")
+        durations.append(2)
+
+    if education_level.lower() in ["high school", "bachelor"]:
+        stages.append("Higher Education")
+        durations.append(3)
+
+    stages.append("Job Applications")
+    durations.append(1)
+
+    if cv_type.lower() in ["technical", "engineering", "it"]:
+        stages.append("Interview Prep")
+        durations.append(1)
+
+    stages.append("Visa Process")
+    durations.append(1)
 
     fig, ax = plt.subplots(figsize=(10, 2))
     ax.barh(["Roadmap"], [sum(durations)], color="lightgray")
-
     left = 0
     for stage, duration in zip(stages, durations):
         ax.barh(["Roadmap"], [duration], left=left, label=stage)
         left += duration
-
-    ax.set_xlim(0, sum(durations))
-    ax.set_xlabel("Timeline (months)")
-    ax.set_yticks([])
-    ax.legend(loc='upper center', bbox_to_anchor=(0.5, 1.25), ncol=3)
-    plt.tight_layout()
+    ax.legend(bbox_to_anchor=(1.05, 1), loc='upper left')
+    ax.set_title("Career Roadmap Timeline")
+    ax.axis('off')
     return fig