Update app.py

app.py

@@ -3,9 +3,15 @@ import json
 from datetime import datetime
 import time
 import requests
+import boto3
+import uuid
 
-# API endpoint (replace with your actual endpoint)
-API_ENDPOINT = "https://api.example.com/personalize-course"
+# API endpoint - Replace with your actual endpoint
+API_ENDPOINT = "https://oau6sljd4l.execute-api.ap-south-1.amazonaws.com/production/process"
+
+# DynamoDB configuration for session tracking
+DYNAMODB_REGION = "ap-south-1"
+SESSION_TABLE = "SessionTracking"
 
 # Set page configuration
 st.set_page_config(
@@ -46,6 +52,7 @@ st.markdown("""
         font-weight: 600;
         color: var(--header-color);
         margin-bottom: 1rem;
+        text-align: center;
     }
     
     .section-header {
@@ -82,17 +89,8 @@ st.markdown("""
         border-color: var(--border-color);
     }
     
-    .stTextInput > div > div > input {
-        background-color: var(--background-color);
-        color: var(--text-color);
-        border-color: var(--border-color);
-    }
-    
-    .stNumberInput > div > div > input {
-        background-color: var(--background-color);
-        color: var(--text-color);
-    }
-    
+    .stTextInput > div > div > input,
+    .stNumberInput > div > div > input,
     .stTextArea > div > div > textarea {
         background-color: var(--background-color);
         color: var(--text-color);
@@ -115,11 +113,6 @@ st.markdown("""
         border-color: var(--border-color);
     }
     
-    /* Footer */
-    footer {
-        color: #888888;
-    }
-    
     /* API Response Container */
     .api-response {
         background-color: #1A1A1A;
@@ -128,53 +121,139 @@ st.markdown("""
         border-left: 4px solid var(--accent-color);
     }
     
-    /* Custom label colors for Full Name and Gender */
-    [data-testid="stForm"] label:contains("Full Name"),
-    [data-testid="stForm"] label:contains("Gender") {
-        color: white !important;
-        font-weight: 500;
-    }
-    
-    /* Alternative approach if the above doesn't work */
-    .st-emotion-cache-1gulkj5 label,  /* For Full Name */
-    .st-emotion-cache-1aumxhf label { /* For Gender */
-        color: white !important;
+    .session-info {
+        background-color: var(--card-background);
+        padding: 1rem;
+        border-radius: 8px;
+        margin: 1rem 0;
+        border: 1px solid var(--border-color);
     }
 </style>
 """, unsafe_allow_html=True)
 
+# Initialize session state
+if 'session_ids' not in st.session_state:
+    st.session_state.session_ids = []
+
 # App Header
-col1, col2, col3 = st.columns([1, 3, 1])
-with col2:
-    st.markdown('<h1 class="main-header">📚 Base Course Personalization</h1>', unsafe_allow_html=True)
-    st.markdown("""
-    Complete this form to tailor your learning experience to your preferences.
-    Your personalized course will be generated based on the information you provide below.
-    """)
+st.markdown('<h1 class="main-header">📚 Base Course Personalization</h1>', unsafe_allow_html=True)
+st.markdown("""
+<div style="text-align: center; margin-bottom: 2rem;">
+Complete this form to tailor your learning experience to your preferences.<br>
+Your personalized course will be generated based on the information you provide below.
+</div>
+""", unsafe_allow_html=True)
 
 # Main Form
 with st.form("personalization_form", clear_on_submit=False):
-    # Personal Information Section
-    st.markdown('<div class="section-header">👤 Personal Details</div>', unsafe_allow_html=True)
+    # Course Information Section
+    st.markdown('<div class="section-header">📚 Course Information</div>', unsafe_allow_html=True)
     
     col1, col2 = st.columns(2)
     with col1:
-        user_name = st.text_input("Full Name", placeholder="Enter your full name")
-        user_age = st.number_input("Age", min_value=10, max_value=100, value=22)
+        course_id = st.number_input("Course ID", min_value=1, value=47, help="Enter the course ID")
+        user_id = st.number_input("User ID", min_value=1, value=30, help="Enter your user ID")
     
     with col2:
+        personalization_id = st.number_input("Personalization ID", min_value=1, value=100, help="Enter personalization ID")
+    
+    # Topics Section
+    st.markdown('<div class="section-header">📋 Topics Configuration</div>', unsafe_allow_html=True)
+    
+    # Option to choose single or multiple topics
+    topic_mode = st.radio(
+        "Topic Mode",
+        ["Single Topic", "Multiple Topics"],
+        horizontal=True,
+        help="Choose whether to process one topic or multiple topics"
+    )
+    
+    if topic_mode == "Single Topic":
+        st.subheader("Single Topic Details")
+        col1, col2 = st.columns(2)
+        with col1:
+            single_topic_id = st.number_input("Topic ID", min_value=1, value=10834)
+            single_topic_title = st.text_input("Topic Title", value="Setting Up Flask Environment")
+        with col2:
+            single_chapter_id = st.number_input("Chapter ID", min_value=1, value=647)
+            single_chapter_title = st.text_input("Chapter Title", value="Introduction to Flask")
+        
+        single_video_url = st.text_input(
+            "Video URL", 
+            value="https://techlearn-dev.s3.ap-south-1.amazonaws.com/course_videos/47/647/1729064365.mp4"
+        )
+        col1, col2 = st.columns(2)
+        with col1:
+            single_video_duration = st.number_input("Video Duration (seconds)", min_value=1, value=462)
+        with col2:
+            single_sequence_number = st.number_input("Sequence Number", min_value=1, value=1)
+    
+    else:
+        st.subheader("Multiple Topics")
+        num_topics = st.number_input("Number of Topics", min_value=1, max_value=10, value=3)
+        
+        topics_data = []
+        for i in range(num_topics):
+            with st.expander(f"Topic {i+1}"):
+                col1, col2 = st.columns(2)
+                with col1:
+                    topic_id = st.number_input(f"Topic ID {i+1}", min_value=1, value=10833+i, key=f"topic_id_{i}")
+                    topic_title = st.text_input(f"Topic Title {i+1}", 
+                                              value=["What is Flask", "Setting Up Flask Environment", "Understanding Flask Project Structure"][i] if i < 3 else f"Topic {i+1}",
+                                              key=f"topic_title_{i}")
+                with col2:
+                    chapter_id = st.number_input(f"Chapter ID {i+1}", min_value=1, value=647, key=f"chapter_id_{i}")
+                    chapter_title = st.text_input(f"Chapter Title {i+1}", value="Introduction to Flask", key=f"chapter_title_{i}")
+                
+                video_url = st.text_input(f"Video URL {i+1}", 
+                                        value=f"https://techlearn-dev.s3.ap-south-1.amazonaws.com/course_videos/47/647/172906{4322+i*50}.mp4",
+                                        key=f"video_url_{i}")
+                col1, col2 = st.columns(2)
+                with col1:
+                    video_duration = st.number_input(f"Video Duration {i+1} (seconds)", min_value=1, value=410+i*20, key=f"video_duration_{i}")
+                with col2:
+                    sequence_number = st.number_input(f"Sequence Number {i+1}", min_value=1, value=i+1, key=f"sequence_number_{i}")
+                
+                topics_data.append({
+                    "topic_id": topic_id,
+                    "topic_title": topic_title,
+                    "chapter_id": chapter_id,
+                    "chapter_title": chapter_title,
+                    "video_url": video_url,
+                    "video_duration": video_duration,
+                    "sequence_number": sequence_number,
+                    "course_id": course_id
+                })
+
+    # Personal Information Section
+    st.markdown('<div class="section-header">👤 User Profile</div>', unsafe_allow_html=True)
+    
+    col1, col2 = st.columns(2)
+    with col1:
+        user_name = st.text_input("Full Name", placeholder="Enter your full name", value="John Doe")
+        user_age = st.number_input("Age", min_value=10, max_value=100, value=22)
         user_gender = st.selectbox(
             "Gender", 
             ["male", "female", "other"], 
             index=0,
             format_func=lambda x: x.capitalize()
         )
+    
+    with col2:
         tech_knowledge = st.select_slider(
             "Technical Knowledge",
             options=["beginner", "intermediate", "advanced"],
             value="beginner"
         )
-    
+        learning_style = st.radio(
+            "Learning Style",
+            ["visual", "auditory", "kinesthetic"],
+            index=0,
+            horizontal=True,
+            format_func=lambda x: x.capitalize()
+        )
+
+    # Preferences Section
     st.markdown('<div class="section-header">🎯 Preferences</div>', unsafe_allow_html=True)
     col1, col2 = st.columns(2)
     
@@ -182,41 +261,25 @@ with st.form("personalization_form", clear_on_submit=False):
         preferred_activity = st.text_area(
             "Preferred Activities", 
             placeholder="e.g., movies, painting, shopping",
+            value="movies, painting, shopping",
             help="Enter activities separated by commas"
         )
-        physical_activities = st.text_area(
-            "Physical Activities", 
-            placeholder="e.g., badminton, cricket, yoga",
-            help="Enter activities separated by commas"
-        )
-    
-    with col2:
         food = st.text_area(
             "Food Preferences", 
             placeholder="e.g., vegetarian, paneer, pizza",
+            value="vegetarian, paneer, pizza",
             help="Enter food items separated by commas"
         )
     
-    # Learning Style Section
-    st.markdown('<div class="section-header">🎓 Learning Preferences</div>', unsafe_allow_html=True)
-    
-    learning_style = st.radio(
-        "Learning Style",
-        ["visual", "auditory", "kinesthetic"],
-        index=0,
-        help="Visual: learn by seeing, Auditory: learn by hearing, Kinesthetic: learn by doing",
-        horizontal=True,
-        format_func=lambda x: x.capitalize()
-    )
-    
-    with st.expander("Learn more about learning styles"):
-        st.info("""
-        - **Visual learners** prefer images, diagrams, and spatial understanding
-        - **Auditory learners** learn best through listening and verbal instructions
-        - **Kinesthetic learners** prefer hands-on activities and learning by doing
-        """)
-    
-    # Language Settings Section
+    with col2:
+        physical_activities = st.text_area(
+            "Physical Activities", 
+            placeholder="e.g., badminton, cricket, yoga",
+            value="badminton, cricket, yoga",
+            help="Enter activities separated by commas"
+        )
+
+    # Language & Voice Settings Section
     st.markdown('<div class="section-header">🗣️ Language & Voice Settings</div>', unsafe_allow_html=True)
     
     col1, col2, col3 = st.columns(3)
@@ -239,32 +302,45 @@ with st.form("personalization_form", clear_on_submit=False):
     with col3:
         tts_voice = st.selectbox(
             "Voice Style", 
-            ["echo", "soprano", "alto", "robotic"],
-            index=0,
+            ["echo", "soprano", "alto", "robotic", "onyx"],
+            index=4,  # Default to "onyx" as used in backend
             format_func=lambda x: x.capitalize()
         )
     
-    toggle_hinglish = st.toggle("Enable Hinglish", value=True, help="Mix of Hindi and English")
+    col1, col2 = st.columns(2)
+    with col1:
+        toggle_hinglish = st.toggle("Enable Hinglish", value=True, help="Mix of Hindi and English")
+    with col2:
+        run_visualization = st.toggle("Run Visualization", value=False, help="Enable visualization features")
 
-    # Course Settings Section
-    st.markdown('<div class="section-header">💻 Technical Preferences</div>', unsafe_allow_html=True)
+    # Technical Settings Section
+    st.markdown('<div class="section-header">💻 Technical Settings</div>', unsafe_allow_html=True)
     
-    programming_language = st.selectbox(
-        "Programming Language", 
-        ["python", "java", "javascript", "c++", "go"],
-        index=0,
-        format_func=lambda x: x.capitalize()
-    )
-    
-    # Show sample lesson structure
-    with st.expander("📋 Sample Course Structure"):
-        st.markdown("""
-        ### Introduction to Flask
-        1. What is Flask
-        2. Setting Up Flask Environment
-        3. Understanding Flask Project Structure
-        """)
+    col1, col2, col3 = st.columns(3)
+    with col1:
+        programming_language = st.selectbox(
+            "Programming Language", 
+            ["python", "java", "javascript", "c++", "go"],
+            index=0,
+            format_func=lambda x: x.capitalize()
+        )
+    with col2:
+        slide_colour = st.selectbox(
+            "Slide Color", 
+            ["blue", "red", "green", "purple", "orange"],
+            index=0,
+            format_func=lambda x: x.capitalize()
+        )
+    with col3:
+        video_type = st.selectbox(
+            "Video Type",
+            ["personalised_video", "standard_video"],
+            index=0,
+            format_func=lambda x: x.replace('_', ' ').title()
+        )
     
+    subtitle = st.text_input("Subtitle", placeholder="Optional subtitle for the video", value="")
+
     # Submit button
     st.markdown("<br>", unsafe_allow_html=True)
     col1, col2, col3 = st.columns([1, 2, 1])
@@ -276,111 +352,157 @@ if submitted:
     # Validate inputs
     if not user_name:
         st.error("Please enter your name")
-    elif not preferred_activity:
-        st.warning("Adding preferred activities will help us personalize your content better")
+    elif topic_mode == "Single Topic" and not single_topic_title:
+        st.error("Please enter a topic title")
     else:
         # Show loading state
         with st.spinner("Generating your personalized course..."):
             time.sleep(1.5)  # Simulating API call delay
             
-            # Create payload
-            payload = {
-                "personalization_id": 100,
-                "user_id": 30,
-                "course_id": 47,
-                "total_videos": 3,
-                "created_at": datetime.utcnow().isoformat(),
-                "user_profile": {
-                    "personalized": True,
-                    "user_name": user_name,
-                    "user_age": user_age,
-                    "user_gender": user_gender,
-                    "user_tech_knowledge": tech_knowledge,
-                    "user_preferred_activity": preferred_activity,
-                    "user_food": food,
-                    "user_physical_activities": physical_activities,
-                    "learning_style": learning_style,
-                    "target_language": target_language,
-                    "tts_gender": tts_gender,
-                    "tts_voice": tts_voice,
-                    "toggle_hinglish": toggle_hinglish,
-                    "run_visualization": False,
-                    "subtitle": "",
-                    "age_group": "18-25",
-                },
-                "topics": [
-                    {
-                        "topic_id": 10834,
-                        "topic_title": "Setting Up Flask Environment",
-                        "chapter_id": 647,
-                        "chapter_title": "Introduction to Flask",
-                        "course_id": 47,
-                        "video_url": "https://techlearn-dev.s3.ap-south-1.amazonaws.com/course_videos/47/647/1729064365.mp4",
-                        "video_duration": 462,
-                        "sequence_number": 2
-                    },
-                    {
-                        "topic_id": 10835,
-                        "topic_title": "Understanding Flask Project Structure",
-                        "chapter_id": 647,
-                        "chapter_title": "Introduction to Flask",
-                        "course_id": 47,
-                        "video_url": "https://techlearn-dev.s3.ap-south-1.amazonaws.com/course_videos/47/647/1729064412.mp4",
-                        "video_duration": 429,
-                        "sequence_number": 3
-                    },
-                    {
-                        "topic_id": 10833,
-                        "topic_title": "What is Flask",
-                        "chapter_id": 647,
-                        "chapter_title": "Introduction to Flask",
-                        "course_id": 47,
-                        "video_url": "https://techlearn-dev.s3.ap-south-1.amazonaws.com/course_videos/47/647/1729064322.mp4",
-                        "video_duration": 410,
-                        "sequence_number": 1
+            # Create user profile
+            user_profile = {
+                "personalized": True,
+                "user_name": user_name,
+                "user_age": user_age,
+                "user_gender": user_gender,
+                "user_tech_knowledge": tech_knowledge,
+                "user_preferred_activity": preferred_activity,
+                "user_food": food,
+                "user_physical_activities": physical_activities,
+                "learning_style": learning_style,
+                "target_language": target_language,
+                "tts_gender": tts_gender,
+                "tts_voice": tts_voice,
+                "toggle_hinglish": toggle_hinglish,
+                "run_visualization": run_visualization,
+                "subtitle": subtitle,
+                "age_group": "18-25" if user_age <= 25 else "26-35" if user_age <= 35 else "35+"
+            }
+            
+            # Create settings
+            settings = {
+                "target_language": target_language,
+                "tts_gender": tts_gender,
+                "tts_voice": tts_voice,
+                "toggle_hinglish": toggle_hinglish,
+                "run_visualization": run_visualization,
+                "subtitle": subtitle,
+                "programming_language": programming_language,
+                "slide_colour": slide_colour,
+                "video_type": video_type
+            }
+            
+            # Create payload based on topic mode
+            if topic_mode == "Single Topic":
+                payload = {
+                    "personalization_id": personalization_id,
+                    "user_profile": user_profile,
+                    "settings": settings,
+                    "topic_data": {
+                        "topic_id": single_topic_id,
+                        "topic_title": single_topic_title,
+                        "chapter_id": single_chapter_id,
+                        "chapter_title": single_chapter_title,
+                        "course_id": course_id,
+                        "video_url": single_video_url,
+                        "video_duration": single_video_duration,
+                        "sequence_number": single_sequence_number
                     }
-                ],
-                "settings": {
-                    "target_language": target_language,
-                    "tts_gender": tts_gender,
-                    "tts_voice": tts_voice,
-                    "toggle_hinglish": toggle_hinglish,
-                    "run_visualization": False,
-                    "subtitle": "",
-                    "programming_language": programming_language,
                 }
-            }
+            else:
+                payload = {
+                    "personalization_id": personalization_id,
+                    "user_id": user_id,
+                    "course_id": course_id,
+                    "total_videos": len(topics_data),
+                    "created_at": datetime.utcnow().isoformat(),
+                    "user_profile": user_profile,
+                    "topics": topics_data,
+                    "settings": settings
+                }
 
             # Make API call
             try:
-                response = requests.post(API_ENDPOINT, json=payload)
+                headers = {
+                    'Content-Type': 'application/json'
+                }
+                
+                response = requests.post(API_ENDPOINT, json=payload, headers=headers, timeout=30)
                 
                 if response.status_code == 200:
-                    st.success(f"Course personalized successfully for {user_name}!")
+                    response_data = response.json()
+                    session_ids = response_data.get("session_ids", [])
+                    
+                    st.success(f"Course personalization started successfully for {user_name}!")
+                    
+                    # Store session IDs in session state
+                    st.session_state.session_ids.extend(session_ids)
                     
                     # Display results in a clean format
-                    st.markdown("### 🎉 Your personalized course is ready!")
+                    st.markdown("### 🎉 Your personalized course generation has started!")
                     st.markdown(f"""
-                    **Course**: Introduction to Flask  
+                    **User**: {user_name}  
+                    **Course ID**: {course_id}  
                     **Programming Language**: {programming_language.capitalize()}  
                     **Learning Style**: {learning_style.capitalize()}  
-                    **Voice**: {tts_voice.capitalize()} ({tts_gender.capitalize()})
+                    **Voice**: {tts_voice.capitalize()} ({tts_gender.capitalize()})  
+                    **Topics**: {len(topics_data) if topic_mode == "Multiple Topics" else 1}
                     """)
                     
+                    # Display session IDs
+                    if session_ids:
+                        st.markdown("### 🔍 Session IDs for Tracking")
+                        for i, session_id in enumerate(session_ids, 1):
+                            st.markdown(f'<div class="session-info"><strong>Session {i}:</strong> <code>{session_id}</code></div>', unsafe_allow_html=True)
+                    
                     # Display API response
-                    st.markdown('<div class="section-header">🌍 API Response</div>', unsafe_allow_html=True)
+                    st.markdown('<div class="section-header">📊 API Response</div>', unsafe_allow_html=True)
                     st.markdown('<div class="api-response">', unsafe_allow_html=True)
-                    st.json(response.json())
+                    st.json(response_data)
                     st.markdown('</div>', unsafe_allow_html=True)
+                    
+                    # Information about tracking
+                    st.info(f"""
+                    💡 **Tracking Information**  
+                    You can track the progress of your video generation using the session IDs above.  
+                    The processing status will be updated in DynamoDB table: `{SESSION_TABLE}`  
+                    Region: `{DYNAMODB_REGION}`
+                    """)
+                    
                 else:
-                    st.error(f"API Error: {response.status_code} - {response.text}")
+                    st.error(f"API Error: {response.status_code}")
+                    if response.text:
+                        st.error(f"Error Details: {response.text}")
+                        
+            except requests.exceptions.Timeout:
+                st.error("Request timed out. Please try again later.")
+            except requests.exceptions.ConnectionError:
+                st.error("Connection error. Please check your internet connection.")
             except Exception as e:
-                st.error(f"API call failed: {e}")
+                st.error(f"API call failed: {str(e)}")
                 
-                # Show local payload as fallback
-                st.warning("Using local data as fallback:")
+                # Show payload for debugging
+                st.warning("Request payload for debugging:")
                 st.json(payload)
 
+# Display previous session IDs if any
+if st.session_state.session_ids:
+    st.markdown("---")
+    st.markdown("### 📋 Previous Session IDs")
+    for i, session_id in enumerate(st.session_state.session_ids, 1):
+        st.markdown(f'<div class="session-info"><strong>Session {i}:</strong> <code>{session_id}</code></div>', unsafe_allow_html=True)
+    
+    # Button to clear session history
+    if st.button("Clear Session History"):
+        st.session_state.session_ids = []
+        st.rerun()
+
 # Footer
 st.markdown("---")
-st.caption("© 2025 Base Course Personalization | All Rights Reserved")
+st.markdown(f"""
+<div style="text-align: center; color: #888888;">
+© 2025 Base Course Personalization | All Rights Reserved<br>
+<small>API Endpoint: {API_ENDPOINT}</small><br>
+<small>Session Tracking: {SESSION_TABLE} ({DYNAMODB_REGION})</small>
+</div>
+""", unsafe_allow_html=True)