Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 import boto3
 import uuid
 import os
+from AdminDashboard import render_admin_dashboard
 
 # Load configuration from environment variables (Hugging Face Spaces)
 API_ENDPOINT = os.getenv("API_ENDPOINT", "https://oau6sljd4l.execute-api.ap-south-1.amazonaws.com/production/process")
@@ -20,7 +21,6 @@ AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
 AWS_DEFAULT_REGION = os.getenv("AWS_DEFAULT_REGION", "ap-south-1")
 
 # Default values
-DEFAULT_COURSE_ID = int(os.getenv("DEFAULT_COURSE_ID", "47"))
 DEFAULT_USER_ID = int(os.getenv("DEFAULT_USER_ID", "30")) 
 DEFAULT_PERSONALIZATION_ID = int(os.getenv("DEFAULT_PERSONALIZATION_ID", "100"))
 API_TIMEOUT = int(os.getenv("API_TIMEOUT", "30"))
@@ -29,6 +29,16 @@ API_TIMEOUT = int(os.getenv("API_TIMEOUT", "30"))
 VALID_USERNAME = os.getenv("APP_USERNAME")
 VALID_PASSWORD = os.getenv("APP_PASSWORD")
 
+# Course configurations
+COURSE_CONFIGS = {
+    "AI Ready Course": {"course_id": 47, "chapter_id": 647},
+    "MERN": {"course_id": 48, "chapter_id": 648},
+    "Machine Learning": {"course_id": 49, "chapter_id": 649},
+    "Data Science": {"course_id": 50, "chapter_id": 650},
+    "Flask": {"course_id": 51, "chapter_id": 651},
+    "Django": {"course_id": 52, "chapter_id": 652}
+}
+
 # Set page configuration
 st.set_page_config(
     page_title="Base Course Personalization",
@@ -272,105 +282,11 @@ if 'session_ids' not in st.session_state:
 if 'authenticated' not in st.session_state:
     st.session_state.authenticated = False
 if 'current_page' not in st.session_state:
-    st.session_state.current_page = "Course Personalization"
-
-# Admin Dashboard Functions
-def fetch_sessions():
-    """Fetch all sessions from DynamoDB"""
-    try:
-        dynamodb = boto3.client(
-            "dynamodb",
-            region_name=DYNAMODB_REGION,
-            aws_access_key_id=AWS_ACCESS_KEY_ID,
-            aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
-        )
-        
-        response = dynamodb.scan(TableName=SESSION_TABLE)
-        items = response.get("Items", [])
-        sessions = []
-        for item in items:
-            sessions.append({
-                "session_id": item.get("session_id", {}).get("S"),
-                "status": item.get("status", {}).get("S", "unknown"),
-                "node": item.get("node", {}).get("S", "-"),
-                "updated_at": item.get("updated_at", {}).get("S", "-"),
-                "video_url": item.get("video_url", {}).get("S", None),
-            })
-        return sessions
-    except Exception as e:
-        st.error(f"⚠️ Error fetching sessions: {e}")
-        return []
-
-def render_admin_dashboard():
-    """Render the admin dashboard page"""
-    st.markdown("<h1 style='color:#4F97FF;text-align:center;'>📊 Base Video Generation Dashboard</h1>", unsafe_allow_html=True)
-    
-    # Auto-refresh controls
-    col1, col2, col3 = st.columns([1, 1, 2])
-    with col1:
-        if st.button("🔄 Refresh Now", key="manual_refresh"):
-            st.rerun()
-    
-    with col2:
-        auto_refresh = st.toggle("Auto Refresh (15s)", value=False)
-    
-    # Fetch sessions
-    sessions = fetch_sessions()
-    
-    if not sessions:
-        st.warning("No sessions found yet.")
-        st.info("Sessions will appear here once you start generating courses from the Course Personalization page.")
-    else:
-        # Display session count
-        st.markdown(f"### 📈 Found {len(sessions)} session(s)")
-        
-        # Sort sessions by updated_at (newest first)
-        try:
-            sessions.sort(key=lambda x: x['updated_at'] if x['updated_at'] != '-' else '1970-01-01', reverse=True)
-        except:
-            pass
-        
-        # Display sessions
-        for i, session in enumerate(sessions):
-            status_color = {
-                'completed': 'lime',
-                'processing': 'orange',
-                'in_progress': 'orange', 
-                'running': 'orange',
-                'failed': 'red',
-                'error': 'red'
-            }.get(session['status'].lower(), 'yellow')
-            
-            with st.container():
-                st.markdown(f"""
-                <div style="background:#2D2D2D;padding:1.5rem;border-radius:12px;margin-bottom:1rem;border:1px solid #444;">
-                    <div style="display:flex;justify-content:space-between;align-items:center;margin-bottom:1rem;">
-                        <h4 style="color:#4F97FF;margin:0;">🆔 Session {i+1}</h4>
-                        <span style="background:{status_color};color:black;padding:0.3rem 0.8rem;border-radius:15px;font-size:0.8rem;font-weight:bold;">{session['status'].upper()}</span>
-                    </div>
-                    <div style="display:grid;grid-template-columns:1fr 1fr;gap:1rem;">
-                        <div>
-                            <b>Session ID:</b><br><code style="background:#1A1A1A;padding:0.3rem;border-radius:4px;">{session['session_id']}</code>
-                        </div>
-                        <div>
-                            <b>Current Node:</b><br><span style="color:#E0E0E0;">{session['node']}</span>
-                        </div>
-                    </div>
-                    <div style="margin-top:1rem;">
-                        <b>⏱️ Last Update:</b> <span style="color:#888;">{session['updated_at']}</span>
-                    </div>
-                    {f"<div style='margin-top:1rem;'><b>🎥 Final Video:</b> <a href='{session['video_url']}' target='_blank' style='color:#4F97FF;'>📺 Watch Video</a></div>" if session['video_url'] else ""}
-                </div>
-                """, unsafe_allow_html=True)
-    
-    # Auto-refresh mechanism
-    if auto_refresh:
-        time.sleep(15)
-        st.rerun()
+    st.session_state.current_page = "Course Generation"
 
 # Authentication check
 if not st.session_state.authenticated:
-    st.markdown('<h1 class="main-header">🔐 Login to Course Personalization</h1>', unsafe_allow_html=True)
+    st.markdown('<h1 class="main-header">🔐 Login to Course Management</h1>', unsafe_allow_html=True)
     
     # Check if credentials are configured
     if not VALID_USERNAME or not VALID_PASSWORD:
@@ -412,13 +328,13 @@ if not st.session_state.authenticated:
 # Navigation and Header (after login)
 col1, col2, col3, col4 = st.columns([3, 2, 2, 1])
 with col1:
-    st.markdown('<h1 class="main-header">📚 Base Course Management</h1>', unsafe_allow_html=True)
+    st.markdown('<h1 class="main-header">📚 Course Management System</h1>', unsafe_allow_html=True)
 
 with col2:
     page = st.selectbox(
         "📋 Navigate to:",
-        ["Course Personalization", "Admin Dashboard"],
-        index=0 if st.session_state.current_page == "Course Personalization" else 1,
+        ["Course Generation", "Admin Dashboard"],
+        index=0 if st.session_state.current_page == "Course Generation" else 1,
         key="page_selector"
     )
     st.session_state.current_page = page
@@ -428,15 +344,30 @@ with col4:
         st.session_state.authenticated = False
         st.session_state.topics_list = [{"topic_title": "What is Flask", "chapter_title": "Introduction to Flask"}]
         st.session_state.session_ids = []
-        st.session_state.current_page = "Course Personalization"
+        st.session_state.current_page = "Course Generation"
         st.rerun()
 
 # Page routing
 if st.session_state.current_page == "Admin Dashboard":
     render_admin_dashboard()
 else:
-    # Course Personalization Page (Original Content)
+    # Course Generation Page (Original Content)
+    
+    # Course Selection Section
+    st.markdown('<div class="section-header">📚 Course Selection</div>', unsafe_allow_html=True)
     
+    col1, col2 = st.columns([2, 3])
+    with col1:
+        selected_course = st.selectbox(
+            "Select Course Type",
+            options=list(COURSE_CONFIGS.keys()),
+            index=0,
+            help="Choose the course you want to generate content for"
+        )
+    
+    with col2:
+        st.info(f"**Selected:** {selected_course} (Course ID: {COURSE_CONFIGS[selected_course]['course_id']})")
+
     # Topics Section (Outside form for dynamic interaction)
     st.markdown('<div class="section-header">📋 Topics Configuration</div>', unsafe_allow_html=True)
 
@@ -598,13 +529,18 @@ else:
         st.markdown("<br>", unsafe_allow_html=True)
         col1, col2, col3 = st.columns([1, 2, 1])
         with col2:
-            submitted = st.form_submit_button("🚀 Generate Base Course", use_container_width=True)
+            submitted = st.form_submit_button("🚀 Generate Course", use_container_width=True)
 
     # Handle submission
     if submitted:
         # Use the topics from session state directly
         topics_to_process = st.session_state.topics_list
         
+        # Get course configuration
+        course_config = COURSE_CONFIGS[selected_course]
+        course_id = course_config["course_id"]
+        chapter_id = course_config["chapter_id"]
+        
         # Validate inputs
         valid_topics = []
         for topic in topics_to_process:
@@ -615,14 +551,13 @@ else:
             st.error("❌ Please enter at least one topic with both topic title and chapter title")
         else:
             # Show loading state with better animation
-            with st.spinner("🎬 Generating your personalized course... This may take a few moments."):
+            with st.spinner(f"🎬 Generating your {selected_course} course... This may take a few moments."):
                 progress_bar = st.progress(0)
                 for i in range(100):
                     time.sleep(0.02)  # Simulating progress
                     progress_bar.progress(i + 1)
                 
                 # Hardcoded values
-                course_id = DEFAULT_COURSE_ID
                 user_id = DEFAULT_USER_ID 
                 personalization_id = DEFAULT_PERSONALIZATION_ID
                 
@@ -659,16 +594,16 @@ else:
                     "video_type": "base_video"
                 }
                 
-                # Generate topics data with user input for titles and hardcoded values for others
+                # Generate topics data with user input for titles and course-specific values
                 topics_data = []
                 for i, topic in enumerate(valid_topics):
                     topics_data.append({
                         "topic_id": 10834 + i,  # Hardcoded with increment
                         "topic_title": topic["topic_title"].strip(),
-                        "chapter_id": 647,  # Hardcoded
+                        "chapter_id": chapter_id,  # From selected course
                         "chapter_title": topic["chapter_title"].strip(),  # User input
-                        "course_id": course_id,
-                        "video_url": f"https://techlearn-dev.s3.ap-south-1.amazonaws.com/course_videos/47/647/172906{4365+i*50}.mp4",  # Hardcoded with variation
+                        "course_id": course_id,  # From selected course
+                        "video_url": f"https://techlearn-dev.s3.ap-south-1.amazonaws.com/course_videos/{course_id}/{chapter_id}/172906{4365+i*50}.mp4",  # Course-specific
                         "video_duration": 462 + i*20,  # Hardcoded with variation
                         "sequence_number": i + 1,
                     })
@@ -697,7 +632,7 @@ else:
                         response_data = response.json()
                         session_ids = response_data.get("session_ids", [])
                         
-                        st.success("🎉 Base Course started successfully!")
+                        st.success(f"🎉 {selected_course} generation started successfully!")
                         
                         # Store session IDs in session state
                         st.session_state.session_ids.extend(session_ids)
@@ -708,6 +643,7 @@ else:
                         col1, col2 = st.columns(2)
                         with col1:
                             st.markdown(f"""
+                            **Course**: {selected_course}  
                             **Course ID**: {course_id}  
                             **Programming Language**: {programming_language.capitalize()}  
                             **Target Language**: {target_language.capitalize()}  
@@ -716,7 +652,8 @@ else:
                             st.markdown(f"""
                             **Voice**: {tts_voice.capitalize()} ({tts_gender.capitalize()})  
                             **Topics Count**: {len(valid_topics)}  
-                            **Hinglish**: {"Enabled" if toggle_hinglish else "Disabled"}
+                            **Hinglish**: {"Enabled" if toggle_hinglish else "Disabled"}  
+                            **Chapter ID**: {chapter_id}
                             """)
                         
                         # Display session IDs
@@ -734,7 +671,7 @@ else:
                         # Information about tracking
                         st.info(f"""
                         💡 **Tracking Information**  
-                        You can track the progress of your video generation using the session IDs above.  
+                        You can track the progress of your {selected_course} course generation using the session IDs above.  
                         The processing status will be updated in DynamoDB table: {SESSION_TABLE}  
                         Region: {DYNAMODB_REGION}