Update app.py

app.py

@@ -60,29 +60,39 @@ def create_dynamic_roadmap(skills, certs, scholarships, edu_opps):
     now = datetime.now()
     roadmap = []
 
-    for i, cert in enumerate(certs['Certification'].tolist()[:2]):
-        roadmap.append({
-            "Task": f"Complete {cert}",
-            "Start": (now + timedelta(days=i*30)).strftime("%Y-%m-%d"),
-            "Finish": (now + timedelta(days=(i+1)*30)).strftime("%Y-%m-%d")
-        })
-
-    for i, scholarship in enumerate(scholarships['Scholarship'].tolist()[:2]):
-        roadmap.append({
-            "Task": f"Apply for {scholarship}",
-            "Start": (now + timedelta(days=90 + i*30)).strftime("%Y-%m-%d"),
-            "Finish": (now + timedelta(days=120 + i*30)).strftime("%Y-%m-%d")
-        })
-
-    for i, degree in enumerate(edu_opps['Program'].tolist()[:1]):
-        roadmap.append({
-            "Task": f"Pursue {degree}",
-            "Start": (now + timedelta(days=180)).strftime("%Y-%m-%d"),
-            "Finish": (now + timedelta(days=720)).strftime("%Y-%m-%d")
-        })
+    # Detect a valid certification column
+    cert_col = next((col for col in certs.columns if col.lower() in ["certification", "name", "title"]), None)
+    if cert_col:
+        for i, cert in enumerate(certs[cert_col].dropna().tolist()[:2]):
+            roadmap.append({
+                "Task": f"Complete {cert}",
+                "Start": (now + timedelta(days=i * 30)).strftime("%Y-%m-%d"),
+                "Finish": (now + timedelta(days=(i + 1) * 30)).strftime("%Y-%m-%d")
+            })
+
+    # Detect a valid scholarship column
+    scholarship_col = next((col for col in scholarships.columns if col.lower() in ["scholarship", "name", "title"]), None)
+    if scholarship_col:
+        for i, scholarship in enumerate(scholarships[scholarship_col].dropna().tolist()[:2]):
+            roadmap.append({
+                "Task": f"Apply for {scholarship}",
+                "Start": (now + timedelta(days=90 + i * 30)).strftime("%Y-%m-%d"),
+                "Finish": (now + timedelta(days=120 + i * 30)).strftime("%Y-%m-%d")
+            })
+
+    # Detect a valid education column
+    edu_col = next((col for col in edu_opps.columns if col.lower() in ["program", "course", "degree", "title"]), None)
+    if edu_col:
+        for i, degree in enumerate(edu_opps[edu_col].dropna().tolist()[:1]):
+            roadmap.append({
+                "Task": f"Pursue {degree}",
+                "Start": (now + timedelta(days=180)).strftime("%Y-%m-%d"),
+                "Finish": (now + timedelta(days=720)).strftime("%Y-%m-%d")
+            })
 
     return pd.DataFrame(roadmap)
 
+
 # UI
 st.title("📊 Personalized Skill Scoring & Career Roadmap App")
 st.markdown("Upload your CV and get a detailed roadmap with live job listings.")