Update learning_platform.py

learning_platform.py

@@ -135,16 +135,17 @@ class CourseBuilder:
         self.setup_graph()
 
     async def plan_course(self, state: CourseState) -> CourseState:
-        """Planner agent for course structure"""
-        st.session_state.agent_logs.append("📋 Planning course structure...")
-        
+    """Planner agent for course structure"""
+    st.session_state.agent_logs.append("📋 Planning course structure...")
+
+    try:
         # Use RAG to find similar course structures
         similar_courses = self.vector_store.similarity_search(
             f"{state['topic']} {state['difficulty']} course",
             k=2
         )
         context = "\n".join([doc.page_content for doc in similar_courses])
-        
+
         prompt = self.prompts.course_planning_prompt()
         response = await self.llm.apredict(
             prompt.format(
@@ -153,23 +154,28 @@ class CourseBuilder:
                 context=context
             )
         )
-        
+
         # Index the course plan for future reference
-        try:
-            course_plan = json.loads(response)
-            self.vector_store.add_texts(
-                [json.dumps(course_plan)],
-                metadatas=[{"type": "course_plan", "topic": state["topic"]}]
-            )
-        except Exception as e:
-            st.session_state.agent_logs.append(f"⚠️ Warning: Couldn't index course plan: {str(e)}")
-        
+        course_plan = json.loads(response)
+        self.vector_store.add_texts(
+            [json.dumps(course_plan)],
+            metadatas=[{"type": "course_plan", "topic": state["topic"]}]
+        )
+
         st.session_state.agent_logs.append("✅ Course structure planned")
         return {
             **state,
             "messages": [AIMessage(content=response)],
             "status": "planning_complete"
         }
+    except Exception as e:
+        error_message = f"Course creation error: {str(e)}"
+        st.session_state.agent_logs.append(error_message)
+        return {
+            **state,
+            "messages": [AIMessage(content=error_message)],
+            "status": "planning_failed"
+        }
 
     async def create_content(self, state: CourseState) -> CourseState:
         """Content creator agent using RAG"""