Update app.py

app.py

@@ -1,12 +1,20 @@
 import streamlit as st
 import os
-from learning_platform import LearningPlatform
+from learning_platform import EnhancedCourseBuilder
 import asyncio
 import nest_asyncio
 from datetime import datetime
+from sqlalchemy.orm import sessionmaker
+from models import create_engine
 
 nest_asyncio.apply()
 
+# Database setup
+db_url = os.getenv('DATABASE_URL', 'sqlite:///learning_platform.db')
+engine = create_engine(db_url)
+Session = sessionmaker(bind=engine)
+db_session = Session()
+
 # Page config
 st.set_page_config(page_title="MicroGuru", page_icon="🎓", layout="wide")
 
@@ -86,7 +94,7 @@ api_key = os.environ.get('OPENAI_API_KEY')
 if not api_key:
     st.error("⚠️ OpenAI API key not found.")
     st.stop()
-platform = LearningPlatform(api_key)
+platform = EnhancedCourseBuilder(api_key, db_session)
 
 def display_agent_logs():
     """Display agent activity logs in sidebar"""
@@ -200,7 +208,7 @@ def create_new_course():
             with st.spinner('Our AI agents are crafting your personalized course...'):
                 try:
                     async def create_course():
-                        return await platform.create_course(topic, difficulty)
+                        return await platform.create_course(topic, difficulty, user_id=1)
                     
                     path = asyncio.run(create_course())
                     st.session_state.learning_path = path
@@ -266,7 +274,7 @@ def display_current_course():
             else:
                 with st.spinner("Generating next module..."):
                     asyncio.run(
-                        platform.generate_next_module(path, next_idx)
+                        platform.generate_module_content(module_id=path.modules[next_idx].id)
                     )
                 st.rerun()