Fix tab switching bug - move all tabs inside gr.Tabs() container

app.py

@@ -816,160 +816,160 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
                     image_output = gr.Image(label="🖼️ Anatomy Diagram", type="pil")
         
         # VIVA MODE TAB (Defined before Book Mode to allow component access)
-        with gr.Tab("🎯 VIVA Training Mode") as viva_tab:
-            viva_status = gr.Markdown("Click 'Start VIVA Training' from Learning Mode after studying a topic!")
+            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)
+                # 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):
-                        viva_image = gr.Image(label="Reference Image", type="pil", interactive=False)
+                with gr.Column(visible=False) as viva_container:
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            viva_image = gr.Image(label="Reference Image", type="pil", interactive=False)
                     
-                    with gr.Column(scale=2):
-                        current_question_display = gr.Markdown("### Question will appear here")
-                        hint_display = gr.Markdown("💡 Hint will appear here")
+                        with gr.Column(scale=2):
+                            current_question_display = gr.Markdown("### Question will appear here")
+                            hint_display = gr.Markdown("💡 Hint will appear here")
                         
-                        # Audio player for question
-                        question_audio = gr.Audio(label="🔊 Listen to Question", autoplay=True, interactive=False)
+                            # Audio player for question
+                            question_audio = gr.Audio(label="🔊 Listen to Question", autoplay=True, interactive=False)
                         
-                        student_answer = gr.Textbox(
-                            label="Your Answer",
-                            placeholder="Type your answer here...",
-                            lines=4
-                        )
+                            student_answer = gr.Textbox(
+                                label="Your Answer",
+                                placeholder="Type your answer here...",
+                                lines=4
+                            )
                         
-                        submit_answer_btn = gr.Button("Submit Answer", variant="primary")
+                            submit_answer_btn = gr.Button("Submit Answer", variant="primary")
                         
-                        feedback_display = gr.Markdown("Feedback will appear here after you submit your answer")
+                            feedback_display = gr.Markdown("Feedback will appear here after you submit your answer")
 
-        # BOOK LEARNING MODE TAB
-        with gr.Tab("📖 Book Learning Mode") as book_tab:
-            # Upload PDF
-            pdf_upload = gr.File(label="Upload Anatomy Textbook (PDF)", file_types=[".pdf"], type="binary")
-            upload_status = gr.Markdown()
+            # BOOK LEARNING MODE TAB
+            with gr.Tab("📖 Book Learning Mode") as book_tab:
+                # Upload PDF
+                pdf_upload = gr.File(label="Upload Anatomy Textbook (PDF)", file_types=[".pdf"], type="binary")
+                upload_status = gr.Markdown()
             
-            # State to hold extracted images, captions, and text
-            book_images_state = gr.State([])
-            page_captions_state = gr.State([])
-            page_texts_state = gr.State([])
+                # State to hold extracted images, captions, and text
+                book_images_state = gr.State([])
+                page_captions_state = gr.State([])
+                page_texts_state = gr.State([])
             
-            # Dropdown to select a page after processing
-            page_dropdown = gr.Dropdown(label="Select Page", choices=[], interactive=False)
+                # Dropdown to select a page after processing
+                page_dropdown = gr.Dropdown(label="Select Page", choices=[], interactive=False)
             
-            # Display selected page image
-            selected_page_image = gr.Image(label="Selected Page", type="pil")
+                # Display selected page image
+                selected_page_image = gr.Image(label="Selected Page", type="pil")
             
-            # Analysis output
-            analysis_output = gr.Markdown(label="Page Analysis")
+                # Analysis output
+                analysis_output = gr.Markdown(label="Page Analysis")
             
-            # Button to start VIVA from this page
-            start_viva_book_btn = gr.Button("🎯 Start VIVA Training from this Page", variant="primary", visible=False)
+                # Button to start VIVA from this page
+                start_viva_book_btn = gr.Button("🎯 Start VIVA Training from this Page", variant="primary", visible=False)
             
-            # Process upload
-            def handle_book_upload(pdf_bytes):
-                extracted_data, status_msg = process_uploaded_book(pdf_bytes)
-                if not extracted_data:
-                    # No data extracted
-                    return [], status_msg, [], [], gr.update(choices=[], interactive=False), None, ""
+                # Process upload
+                def handle_book_upload(pdf_bytes):
+                    extracted_data, status_msg = process_uploaded_book(pdf_bytes)
+                    if not extracted_data:
+                        # No data extracted
+                        return [], status_msg, [], [], gr.update(choices=[], interactive=False), None, ""
                 
-                # Separate images, captions, and text
-                img_list = [item[0] for item in extracted_data]
-                caps = [item[1] for item in extracted_data]
-                texts = [item[2] for item in extracted_data]
+                    # Separate images, captions, and text
+                    img_list = [item[0] for item in extracted_data]
+                    caps = [item[1] for item in extracted_data]
+                    texts = [item[2] for item in extracted_data]
                 
-                # Update dropdown with captions and enable it
-                dropdown_update = gr.update(choices=caps, interactive=True)
-                return img_list, status_msg, caps, texts, dropdown_update, None, ""
+                    # Update dropdown with captions and enable it
+                    dropdown_update = gr.update(choices=caps, interactive=True)
+                    return img_list, status_msg, caps, texts, dropdown_update, None, ""
             
-            pdf_upload.upload(
-                fn=handle_book_upload,
-                inputs=[pdf_upload],
-                outputs=[book_images_state, upload_status, page_captions_state, page_texts_state, page_dropdown, selected_page_image, analysis_output]
-            )
+                pdf_upload.upload(
+                    fn=handle_book_upload,
+                    inputs=[pdf_upload],
+                    outputs=[book_images_state, upload_status, page_captions_state, page_texts_state, page_dropdown, selected_page_image, analysis_output]
+                )
             
-            # When a page is selected, show image and analysis
-            def show_page_analysis(selected_caption, images, captions, texts):
-                if not selected_caption:
-                    return None, ""
-                # Find index
-                try:
-                    idx = captions.index(selected_caption)
-                except ValueError:
-                    return None, ""
+                # When a page is selected, show image and analysis
+                def show_page_analysis(selected_caption, images, captions, texts):
+                    if not selected_caption:
+                        return None, ""
+                    # Find index
+                    try:
+                        idx = captions.index(selected_caption)
+                    except ValueError:
+                        return None, ""
                 
-                img = images[idx]
-                text = texts[idx] if idx < len(texts) else ""
+                    img = images[idx]
+                    text = texts[idx] if idx < len(texts) else ""
                 
-                analysis = analyze_book_image(img, selected_caption, text)
+                    analysis = analyze_book_image(img, selected_caption, text)
                 
-                # Construct a topic string for VIVA
-                viva_topic = f"Anatomy of {selected_caption} (from textbook)"
+                    # Construct a topic string for VIVA
+                    viva_topic = f"Anatomy of {selected_caption} (from textbook)"
                 
-                return img, analysis, viva_topic, gr.update(visible=True)
+                    return img, analysis, viva_topic, gr.update(visible=True)
             
-            # Hidden state to store current page topic for VIVA
-            current_book_topic = gr.State("")
+                # Hidden state to store current page topic for VIVA
+                current_book_topic = gr.State("")
             
-            page_dropdown.change(
-                fn=show_page_analysis, 
-                inputs=[page_dropdown, book_images_state, page_captions_state, page_texts_state], 
-                outputs=[selected_page_image, analysis_output, current_book_topic, start_viva_book_btn]
-            )
+                page_dropdown.change(
+                    fn=show_page_analysis, 
+                    inputs=[page_dropdown, book_images_state, page_captions_state, page_texts_state], 
+                    outputs=[selected_page_image, analysis_output, current_book_topic, start_viva_book_btn]
+                )
             
-            # 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]
-        )
+                # 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]
+            )
         
-        # ADMIN PANEL TAB
-        with gr.Tab("🔐 Admin Panel") as admin_tab:
-            gr.Markdown("## Admin Panel - Student Database")
-            gr.Markdown("Enter the admin password to view registered students.")
+            # ADMIN PANEL TAB
+            with gr.Tab("🔐 Admin Panel") as admin_tab:
+                gr.Markdown("## Admin Panel - Student Database")
+                gr.Markdown("Enter the admin password to view registered students.")
             
-            # Password input
-            with gr.Row():
-                admin_password_input = gr.Textbox(
-                    label="Admin Password",
-                    placeholder="Enter admin password",
-                    type="password",
-                    scale=2
-                )
-                admin_login_btn = gr.Button("🔓 Login", variant="primary", scale=1)
+                # Password input
+                with gr.Row():
+                    admin_password_input = gr.Textbox(
+                        label="Admin Password",
+                        placeholder="Enter admin password",
+                        type="password",
+                        scale=2
+                    )
+                    admin_login_btn = gr.Button("🔓 Login", variant="primary", scale=1)
             
-            admin_status = gr.Markdown("")
+                admin_status = gr.Markdown("")
             
-            # Admin content (hidden until authenticated)
-            with gr.Column(visible=False) as admin_content:
-                gr.Markdown("### 📊 Registered Students")
+                # Admin content (hidden until authenticated)
+                with gr.Column(visible=False) as admin_content:
+                    gr.Markdown("### 📊 Registered Students")
                 
-                admin_stats = gr.Markdown("")
+                    admin_stats = gr.Markdown("")
                 
-                with gr.Row():
-                    refresh_btn = gr.Button("🔄 Refresh Data", variant="secondary")
-                    logout_btn = gr.Button("🚪 Logout", variant="secondary")
+                    with gr.Row():
+                        refresh_btn = gr.Button("🔄 Refresh Data", variant="secondary")
+                        logout_btn = gr.Button("🚪 Logout", variant="secondary")
                 
-                students_table = gr.Dataframe(
-                    headers=["ID", "Name", "Medical School", "Year", "Registration Date"],
-                    label="Students Database",
-                    interactive=False,
-                    wrap=True
-                )
+                    students_table = gr.Dataframe(
+                        headers=["ID", "Name", "Medical School", "Year", "Registration Date"],
+                        label="Students Database",
+                        interactive=False,
+                        wrap=True
+                    )