Add VIVA button to Book Mode and update AI persona

app.py

@@ -186,30 +186,30 @@ def generate_anatomy_info(query: str) -> str:
             "Authorization": f"Bearer {HYPERBOLIC_API_KEY}"
         }
         
-        prompt = f"""You are an anatomy tutor for MBBS students. Provide key educational points about: {query}
+        prompt = f"""You are an expert anatomy professor teaching MBBS students. Provide a detailed, high-level educational summary about: {query}
 
 Format your response with clear sections using these exact emoji icons:
 
 📍 **Location & Definition:**
-[Brief description of what it is and where it's located]
+[Precise anatomical definition, location, and relations using standard medical terminology]
 
 🔍 **Key Anatomical Features:**
-- [Feature 1]
-- [Feature 2]
-- [Feature 3]
+- [Detailed feature 1 (e.g., attachments, blood supply, innervation)]
+- [Detailed feature 2]
+- [Detailed feature 3]
 
 🏥 **Clinical Significance:**
-- [Clinical point 1]
-- [Clinical point 2]
+- [Clinical correlation 1 (e.g., pathologies, surgical relevance)]
+- [Clinical correlation 2]
 
 🔗 **Related Structures:**
 - [Related structure 1]
 - [Related structure 2]
 
 💡 **Quick Memory Tip:**
-[A mnemonic or helpful tip to remember this anatomy]
+[A high-yield mnemonic or tip for exams]
 
-Keep it concise, clear, and educational. Use simple medical terminology."""
+Keep it professional, accurate, and suitable for medical school level study. Use proper anatomical terminology throughout."""
 
         payload = {
             "model": HYPERBOLIC_MODEL,
@@ -249,12 +249,12 @@ def generate_viva_questions(topic: str) -> list:
             "Authorization": f"Bearer {HYPERBOLIC_API_KEY}"
         }
         
-        prompt = f"""You are an anatomy professor conducting a VIVA exam on: {topic}
+        prompt = f"""You are a strict but fair anatomy professor conducting a VIVA exam for final year MBBS students on: {topic}
 
-Generate exactly 5 viva questions that would be asked during an MBBS anatomy practical exam. For each question, provide:
-1. The question
-2. A helpful hint (not the answer, but guides student thinking)
-3. The expected key points in the answer
+Generate exactly 5 viva questions that test deep anatomical understanding, clinical application, and relations. For each question, provide:
+1. The question (challenging, requiring synthesis of knowledge)
+2. A helpful hint (guides thinking without giving the answer)
+3. The expected key points in the answer (using proper terminology)
 
 Format your response EXACTLY as follows:
 Q1: [question]
@@ -267,7 +267,7 @@ ANSWER: [expected answer key points]
 
 ... and so on for all 5 questions.
 
-Make questions progressively harder. Start with identification, then structure, then clinical relevance."""
+Make questions progressively harder. Start with detailed relations/supply, then move to complex clinical scenarios."""
 
         payload = {
             "model": HYPERBOLIC_MODEL,
@@ -322,7 +322,7 @@ def evaluate_viva_answer(question: str, student_answer: str, expected_answer: st
             "Authorization": f"Bearer {HYPERBOLIC_API_KEY}"
         }
         
-        prompt = f"""You are an encouraging anatomy professor evaluating a student's VIVA answer.
+        prompt = f"""You are an anatomy professor evaluating an MBBS student's VIVA answer. Expect high standards and precise terminology.
 
 Question: {question}
 Expected key points: {expected_answer}
@@ -330,19 +330,19 @@ Student's answer: {student_answer}
 
 Provide feedback in this EXACT format:
 
-[First, write one encouraging sentence praising effort or any correct elements]
+[First, write one sentence evaluating the precision and depth of the answer]
 
 ✅ **What was correct:**
-[List what the student got right - be specific. If nothing, write "Unfortunately, the answer didn't cover the key anatomical points."]
+[List correct points. Praise use of proper terminology.]
 
 ❌ **What was missing:**
-[List what important points were missed - be specific. If nothing was missed, write "Nothing! You covered all the key points."]
+[List missing key points, relations, or clinical aspects. Be specific about missing terminology.]
 
-**Score:** [Choose: EXCELLENT, GOOD, FAIR, or NEEDS_IMPROVEMENT]
+**Score:** [Choose: DISTINCTION, PASS, BORDERLINE, or FAIL]
 
-[End with an encouraging message that motivates continued learning]
+[End with a constructive comment on how to improve to a professional medical standard]
 
-Be warm, supportive, and constructive. Celebrate correct answers enthusiastically!"""
+Be professional and constructive. Demand accuracy."""
 
         payload = {
             "model": HYPERBOLIC_MODEL,
@@ -359,18 +359,18 @@ Be warm, supportive, and constructive. Celebrate correct answers enthusiasticall
         
         # Determine emoji and encouragement based on feedback content
         feedback_upper = feedback.upper()
-        if "EXCELLENT" in feedback_upper:
+        if "DISTINCTION" in feedback_upper:
             emoji = "🌟"
-            encouragement = "\n\n🎉 **Outstanding!** You're mastering this topic! Keep up the excellent work!"
-        elif "GOOD" in feedback_upper:
+            encouragement = "\n\n🎉 **Outstanding!** Distinction level answer! You're mastering this topic!"
+        elif "PASS" in feedback_upper:
             emoji = "✅"
-            encouragement = "\n\n👏 **Well done!** You're doing great! A bit more practice and you'll ace this!"
-        elif "FAIR" in feedback_upper:
-            emoji = "👍"
-            encouragement = "\n\n💪 **Good effort!** You're on the right track. Review the key points and try again!"
+            encouragement = "\n\n👏 **Good Pass!** Solid understanding. Review the finer details to reach distinction level."
+        elif "BORDERLINE" in feedback_upper:
+            emoji = "⚠️"
+            encouragement = "\n\n💪 **Borderline.** You have the basics, but need more precision with terminology."
         else:
             emoji = "📚"
-            encouragement = "\n\n🌱 **Keep learning!** Don't worry, understanding anatomy takes practice. You've got this!"
+            encouragement = "\n\n🌱 **Keep studying.** Focus on the key anatomical relations and clinical points."
         
         # Format the complete feedback
         formatted_feedback = f"{emoji} **VIVA Feedback:**\n\n{feedback}{encouragement}\n\n---\n\n📖 **Reference Answer:**\n{expected_answer}"
@@ -493,34 +493,34 @@ def analyze_book_image(image, page_info, page_text=""):
         # Include extracted text in the prompt context
         context_text = f"Page Content:\n{page_text}" if page_text else "No text extracted from this page."
         
-        prompt = f"""You are an anatomy professor helping MBBS students learn from their textbook.
+        prompt = f"""You are an expert anatomy professor helping MBBS students analyze their textbook content.
 
 A student is looking at {page_info} of their anatomy textbook.
 {context_text}
 
-Based on the text content above (and typical anatomical diagrams found in such contexts), provide a comprehensive explanation:
+Based on the text content above, provide a high-level medical analysis:
 
 ## 📖 Page Overview
-[Summarize the key anatomical topic discussed on this page based on the text]
+[Summarize the key anatomical topic using standard medical terminology]
 
 ## 🔍 Key Concepts Explained
-[Explain the main concepts covered in the text in simple terms]
+[Explain the concepts in detail, focusing on relations, blood supply, nerve supply, and lymphatic drainage where applicable]
 
 ## 🏥 Clinical Relevance
-[Extract or infer clinical points mentioned or relevant to this topic]
+[Detailed clinical correlations, surgical landmarks, or pathological conditions mentioned or relevant]
 
 ## 💡 Study Tips
-[How to remember this specific information]
+[High-yield memory aids for medical exams]
 
-## ❓ Self-Test Questions (Answer these to test your understanding)
+## ❓ Self-Test Questions (MBBS Level)
 1. [Question based on the page text]
 2. [Question based on the page text]
 ...
 15. [Question based on the page text]
 
-(Provide at least 15 distinct questions covering identification, function, clinical application, and relationships)
+(Provide at least 15 distinct, challenging questions covering detailed anatomy, relations, and clinical application)
 
-Be educational and specific to the provided text content."""
+Be professional, accurate, and suitable for medical school level study."""
 
         payload = {
             "model": HYPERBOLIC_MODEL,
@@ -693,6 +693,32 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
                 with gr.Column(scale=1):
                     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.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")
+                        
+                        # 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
+                        )
+                        
+                        submit_answer_btn = gr.Button("Submit Answer", variant="primary")
+                        
+                        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
@@ -713,6 +739,9 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
             # 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)
+            
             # Process upload
             def handle_book_upload(pdf_bytes):
                 extracted_data, status_msg = process_uploaded_book(pdf_bytes)
@@ -749,35 +778,45 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
                 text = texts[idx] if idx < len(texts) else ""
                 
                 analysis = analyze_book_image(img, selected_caption, text)
-                return img, analysis
+                
+                # Construct a topic string for VIVA
+                viva_topic = f"Anatomy of {selected_caption} (from textbook)"
+                
+                return img, analysis, viva_topic, gr.update(visible=True)
             
-            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])
-        
-        # VIVA MODE TAB
-        with gr.Tab("🎯 VIVA Training Mode") as viva_tab:
-            viva_status = gr.Markdown("Click 'Start VIVA Training' from Learning Mode after studying a topic!")
+            # Hidden state to store current page topic for VIVA
+            current_book_topic = gr.State("")
             
-            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")
-                        
-                        # 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
-                        )
-                        
-                        submit_answer_btn = gr.Button("Submit Answer", variant="primary")
-                        
-                        feedback_display = gr.Markdown("Feedback will appear here after you submit your answer")
+            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
+            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]
+            )