Add welcome screen for student name collection at app start with personalized VIVA greetings

app.py

@@ -25,9 +25,10 @@ ELEVENLABS_API_URL = "https://api.elevenlabs.io/v1/text-to-speech"
 # Voice ID for "Brian": nPczCjzI2devNBz1zQrb
 ELEVENLABS_VOICE_ID = "nPczCjzI2devNBz1zQrb" 
 
-def generate_audio(text: str) -> str:
+def generate_audio(text: str, student_name: str = None) -> str:
     """
     Generate audio from text using ElevenLabs API.
+    If student_name is provided, prepends a personalized greeting.
     Returns path to temporary audio file or None if failed.
     """
     if not ELEVENLABS_API_KEY:
@@ -37,6 +38,10 @@ def generate_audio(text: str) -> str:
     if not text:
         print("⚠️ No text provided for audio generation")
         return None
+    
+    # Add personalized greeting if student name is provided
+    if student_name:
+        text = f"Welcome to Viva, Doctor {student_name}, let's start. {text}"
         
     print(f"Generating audio for text: {text[:50]}...")
         
@@ -550,13 +555,13 @@ Be professional, accurate, and suitable for medical school level study."""
 
 
 # VIVA Mode Handler Functions
-def start_viva_mode(topic, image):
+def start_viva_mode(topic, image, student_name=""):
     """Initialize VIVA mode with questions."""
     if not topic or not image:
         return (
             gr.update(visible=False),  # viva_container
             "Please learn about a topic first before starting VIVA mode!",  # viva_status
-            None, None, None, None, None, None, []  # other outputs
+            None, None, None, None, None, None, [], None, student_name  # other outputs
         )
     
     questions = generate_viva_questions(topic)
@@ -565,33 +570,42 @@ def start_viva_mode(topic, image):
         return (
             gr.update(visible=False),
             "Error generating VIVA questions. Please try again.",
-            None, None, None, None, None, None, []
+            None, None, None, None, None, gr.update(interactive=False), [], None, student_name
         )
    
     # Start with question 1
     q1 = questions[0]
     
-    # Generate audio for first question
-    audio_path = generate_audio(q1['question'])
+    # Generate audio for first question with student name
+    audio_path = generate_audio(q1['question'], student_name if student_name else None)
     
     return (
         gr.update(visible=True),  # Show VIVA container
         f"**VIVA MODE ACTIVE** 📝\nTopic: {topic}",  # viva_status
         image,  # viva_image
-        f"### Question 1 of 5\n\n**{q1['question']}**",  # current_question
+        f"### Question 1 of 5\n\n**{q1['question']}**",  # current_question_display
         f"💡 **Hint:** {q1.get('hint', 'Think about the key anatomical features.')}",  # hint_display
         "",  # Clear answer input
         "",  # Clear feedback
         gr.update(interactive=True, value="Submit Answer"),  # Enable submit button
         questions,  # Store questions in state
-        audio_path  # Return audio path
+        audio_path,  # Return audio path
+        student_name  # Return student name to maintain in state
     )
 
+# Wrapper to start VIVA with personalized greeting
+def start_viva_with_name(name, topic, image):
+    viva_container_out, viva_status_out, viva_image_out, cur_q_disp, hint_disp, stu_ans, fb_disp, sub_btn, viva_q_state, q_audio, student_name_out = start_viva_mode(topic, image, name)
+    greeting = f"Doctor {name}, let's go to VIVA!"
+    # Add greeting as separate markdown component above question
+    viva_greeting_out = greeting
+    return viva_container_out, viva_status_out, viva_image_out, cur_q_disp, hint_disp, stu_ans, fb_disp, sub_btn, viva_q_state, q_audio, viva_greeting_out, student_name_out
+
 
-def submit_viva_answer(answer, questions, current_q_idx):
+def submit_viva_answer(answer, questions, current_q_idx, student_name=""):
     """Process student's answer and move to next question."""
     if not questions or current_q_idx >= len(questions):
-        return ("VIVA Complete!", "", "", gr.update(interactive=False), current_q_idx)
+        return ("VIVA Complete!", "", "", gr.update(interactive=False), current_q_idx, None)
     
     q = questions[current_q_idx]
     feedback_text, emoji = evaluate_viva_answer(q['question'], answer, q.get('answer', ''))
@@ -604,8 +618,8 @@ def submit_viva_answer(answer, questions, current_q_idx):
         next_question = f"### Question {next_idx + 1} of 5\n\n**{next_q['question']}**"
         next_hint = f"💡 **Hint:** {next_q.get('hint', 'Think carefully about the anatomical relationships.')}"
         
-        # Generate audio for next question
-        audio_path = generate_audio(next_q['question'])
+        # Generate audio for next question with student name
+        audio_path = generate_audio(next_q['question'], student_name if student_name else None)
         
         return (
             next_question,  # Show next question
@@ -632,21 +646,59 @@ def submit_viva_answer(answer, questions, current_q_idx):
 
 # Create Gradio interface
 with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
-    gr.Markdown(
-        """
-        # 🩺 AnatomyBot - Your MBBS Anatomy Tutor
-        
-        Master anatomy through AI-powered learning and interactive VIVA practice!
-        """
-    )
-    
-    # State variables for VIVA mode
+    # State variables
+    student_name_state = gr.State("")
     viva_questions_state = gr.State([])
     current_question_idx = gr.State(0)
     current_topic = gr.State("")
     current_image_state = gr.State(None)
     
-    with gr.Tabs() as tabs:
+    # Welcome Screen (shown first)
+    with gr.Column(visible=True) as welcome_screen:
+        gr.Markdown(
+            """
+            # 🩺 Welcome to AnatomyBot
+            ### Your Personal MBBS Anatomy Tutor
+            
+            Master anatomy through AI-powered learning and interactive VIVA practice!
+            """
+        )
+        gr.Markdown("---")
+        gr.Markdown("## 👨‍⚕️ Let's Get Started!")
+        gr.Markdown("Please enter your name to begin your personalized learning journey.")
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                pass  # Empty column for centering
+            with gr.Column(scale=2):
+                welcome_name_input = gr.Textbox(
+                    label="Your Name",
+                    placeholder="Enter your name (e.g., Ahmed, Sarah, etc.)",
+                    lines=1,
+                    scale=1
+                )
+                welcome_submit_btn = gr.Button(
+                    "🚀 Start Learning",
+                    variant="primary",
+                    size="lg"
+                )
+            with gr.Column(scale=1):
+                pass  # Empty column for centering
+    
+    # Main Application (hidden initially)
+    with gr.Column(visible=False) as main_app:
+        gr.Markdown(
+            """
+            # 🩺 AnatomyBot - Your MBBS Anatomy Tutor
+            
+            Master anatomy through AI-powered learning and interactive VIVA practice!
+            """
+        )
+        
+        # Display student name
+        student_name_display = gr.Markdown("")
+    
+    with gr.Tabs(visible=False) as tabs:
         # LEARNING MODE TAB
         with gr.Tab("📚 Learning Mode"):
             # Search and examples at the top
@@ -697,6 +749,9 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
         with gr.Tab("🎯 VIVA Training Mode") as viva_tab:
             viva_status = gr.Markdown("Click 'Start VIVA Training' from Learning Mode after studying a topic!")
             
+            # Additional greeting component (initially hidden)
+            viva_greeting = gr.Markdown("", visible=False)
+            
             with gr.Column(visible=False) as viva_container:
                 with gr.Row():
                     with gr.Column(scale=1):
@@ -793,33 +848,55 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
                 outputs=[selected_page_image, analysis_output, current_book_topic, start_viva_book_btn]
             )
             
-            # Start VIVA from Book Mode
-            start_viva_book_btn.click(
-                fn=lambda: gr.update(value="⏳ Processing VIVA Question...", interactive=False),
-                outputs=[start_viva_book_btn]
-            ).then(
-                fn=start_viva_mode,
-                inputs=[current_book_topic, selected_page_image],
-                outputs=[
-                    viva_container, viva_status, viva_image,
-                    current_question_display, hint_display,
-                    student_answer, feedback_display, submit_answer_btn,
-                    viva_questions_state,
-                    question_audio
-                ]
-            ).then(
-                fn=lambda: gr.update(selected=1),  # Switch to VIVA tab (index 1 is VIVA tab)
-                outputs=[tabs]
-            ).then(
-                fn=lambda: gr.update(value="🎯 Start VIVA Training from this Page", interactive=True),
-                outputs=[start_viva_book_btn]
-            ).then(
-                fn=lambda: 0,
-                outputs=[current_question_idx]
-            )
+            # Start VIVA from Book Mode - Use pre-collected name
+        start_viva_book_btn.click(
+            fn=lambda name, topic, image: start_viva_with_name(name, topic, image),
+            inputs=[student_name_state, current_book_topic, selected_page_image],
+            outputs=[
+                viva_container, viva_status, viva_image,
+                current_question_display, hint_display,
+                student_answer, feedback_display, submit_answer_btn,
+                viva_questions_state,
+                question_audio, viva_greeting, student_name_state
+            ]
+        ).then(
+            fn=lambda: gr.update(selected=1),  # Switch to VIVA tab
+            outputs=[tabs]
+        ).then(
+            fn=lambda: 0,
+            outputs=[current_question_idx]
+        )
         
 
     
+    # Welcome Screen Handler
+    def handle_welcome_submit(name):
+        """Handle welcome screen submission."""
+        if not name or not name.strip():
+            return gr.update(), gr.update(), gr.update(), gr.update(), ""  # Don't proceed if name is empty
+        
+        greeting = f"**Welcome, Doctor {name}!** 👋"
+        return (
+            gr.update(visible=False),  # Hide welcome screen
+            gr.update(visible=True),   # Show main app
+            gr.update(visible=True),   # Show tabs
+            greeting,                   # Display greeting
+            name                        # Store name in state
+        )
+    
+    welcome_submit_btn.click(
+        fn=handle_welcome_submit,
+        inputs=[welcome_name_input],
+        outputs=[welcome_screen, main_app, tabs, student_name_display, student_name_state]
+    )
+    
+    # Also allow Enter key to submit
+    welcome_name_input.submit(
+        fn=handle_welcome_submit,
+        inputs=[welcome_name_input],
+        outputs=[welcome_screen, main_app, tabs, student_name_display, student_name_state]
+    )
+    
     # Event handlers for Learning Mode
     def handle_query(query):
         """Handle learning mode query and store topic/image."""
@@ -840,19 +917,20 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
         outputs=[image_output, info_output, error_output, current_topic, current_image_state, start_viva_btn]
     )
     
-    # Start VIVA Mode
+    # Start VIVA Mode - Directly start with pre-collected name
     start_viva_btn.click(
         fn=lambda: gr.update(value="⏳ Processing VIVA Question...", interactive=False),
         outputs=[start_viva_btn]
     ).then(
-        fn=start_viva_mode,
-        inputs=[current_topic, current_image_state],
+        fn=lambda name, topic, image: start_viva_mode(topic, image, name),
+        inputs=[student_name_state, current_topic, current_image_state],
         outputs=[
             viva_container, viva_status, viva_image,
             current_question_display, hint_display,
             student_answer, feedback_display, submit_answer_btn,
             viva_questions_state,
-            question_audio  # Output audio
+            question_audio,  # Output audio
+            student_name_state  # Return student name (unchanged)
         ]
     ).then(
         fn=lambda: gr.update(selected=1),  # Switch to VIVA tab
@@ -860,9 +938,6 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
     ).then(
         fn=lambda: gr.update(value="🎯 Start VIVA Training", interactive=True),  # Reset button
         outputs=[start_viva_btn]
-    ).then(
-        fn=lambda: gr.update(selected=1),  # Switch to VIVA tab
-        outputs=[tabs]
     ).then(
         fn=lambda: 0,  # Reset question index
         outputs=[current_question_idx]
@@ -871,7 +946,7 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
     # Submit VIVA Answer
     submit_answer_btn.click(
         fn=submit_viva_answer,
-        inputs=[student_answer, viva_questions_state, current_question_idx],
+        inputs=[student_answer, viva_questions_state, current_question_idx, student_name_state],
         outputs=[
             current_question_display, hint_display, student_answer,
             feedback_display, submit_answer_btn, current_question_idx,