Update app.py

app.py

@@ -170,57 +170,76 @@ def analyze_math_work(image, grade_level, topic):
     if not topic:
         return "⚠️ Please select a topic first!"
     
-    system_prompt = f"""You are a patient and encouraging mathematics teacher for {grade_level} students.
-Your role is to:
-1. Carefully examine the student's written work in the image
-2. Identify what mathematical problem they are solving
-3. Check their working step-by-step
-4. Point out any errors with clear, kind explanations
-5. Provide the correct solution with detailed steps
-6. Encourage the student and give tips for improvement
+    system_prompt = f"""You are an exceptional mathematics teacher specializing in {grade_level} level students.
+Your student has submitted their work on: {topic}
 
-Current topic focus: {topic}
-Student level: {grade_level}
+Analyze their solution with EXTREME CLARITY and KINDNESS:
 
-Provide your feedback in this format:
+**📝 PROBLEM IDENTIFIED:**
+State exactly what mathematical problem they're solving
 
-**📝 Problem Identified:**
-[State what problem the student is trying to solve]
+**👀 STUDENT'S APPROACH:**
+Describe their working, steps, and methods clearly
 
-**👀 Student's Approach:**
-[Describe what the student did - their steps]
+**✅ CORRECTNESS EVALUATION:**
+Is it completely correct? Partially correct? Incorrect? Be clear and specific.
 
-**✅ Evaluation:**
-[Is it correct? If not, where exactly did they go wrong?]
+**🔍 DETAILED STEP-BY-STEP ANALYSIS:**
+For each step:
+- Is this step correct?
+- If incorrect, explain the error gently
+- Show the correct approach
 
-**💡 Detailed Feedback:**
-[Step-by-step explanation of mistakes and corrections]
+**📚 COMPLETE CORRECT SOLUTION:**
+Show the entire correct solution with:
+- All intermediate steps
+- Mathematical notation
+- Clear logic flow
 
-**📚 Correct Solution:**
-[Show the complete correct solution with all steps clearly explained]
+**💡 LEARNING INSIGHTS:**
+- What concept was this testing?
+- Key insight the student should remember
+- Common mistakes students make
+- Real-world connection (if applicable)
 
-**🌟 Encouragement & Tips:**
-[Positive feedback and 2-3 helpful tips for improvement]
+**🌟 ENCOURAGEMENT & IMPROVEMENT TIPS:**
+- Positive feedback on what they did well
+- 3 specific tips to improve
+- Next level of difficulty to attempt
 
-Be specific, kind, and educational. Focus on helping them understand their mistakes."""
+TONE: Be warm, encouraging, and supportive. Make them feel like learning is exciting! Use emojis to make feedback engaging."""
     
     try:
-        # Use Gemini with image
         response, source = ask_ai(system_prompt, temperature=0.5, image=image)
         
         if source == "error":
             return response
         
-        # Add source indicator if not primary
-        if source == "cohere":
-            response = f"🔵 *[Analyzed by Cohere - text fallback]*\n\n{response}"
-        elif source == "hf":
-            response = f"🟢 *[Analyzed by HF - text fallback]*\n\n{response}"
+        # Format the response nicely
+        formatted_response = f"""
+╔════════════════════════════════════════════════════════════════╗
+║           ✨ YOUR PERSONALIZED AI FEEDBACK ✨                  ║
+╚════════════════════════════════════════════════════════════════╝
+
+{response}
+
+╔════════════════════════════════════════════════════════════════╗
+║  🎯 NEXT STEPS: Try the next problem or ask your AI tutor!    ║
+╚════════════════════════════════════════════════════════════════╝
+"""
         
-        return response
+        return formatted_response
         
     except Exception as e:
-        return f"❌ Error analyzing work: {str(e)}\n\nPlease ensure:\n1. Your image is clear and readable\n2. The mathematical work is visible\n3. Try again or check your internet connection"
+        return f"""❌ Error analyzing work: {str(e)}
+
+⚡ **Quick Fix:**
+1. Make sure your image is clear and readable
+2. The math work should be visible and in focus
+3. Check your internet connection
+4. Try uploading a cleaner photo
+
+💡 **Pro Tip:** Good lighting and a straight-on photo work best!"""
 
 # ---------- 5. Generate Practice Problems ----------
 def generate_practice_problem(grade_level, topic):
@@ -229,34 +248,51 @@ def generate_practice_problem(grade_level, topic):
     if not topic:
         return "⚠️ Please select a topic first!"
     
-    prompt = f"""Generate ONE practice problem for {grade_level} students on the topic: "{topic}"
+    prompt = f"""Generate ONE engaging practice problem for {grade_level} students on: "{topic}"
 
 Requirements:
-- Appropriate difficulty for {grade_level}
-- Clear problem statement
-- Include any necessary diagrams/context in text
-- Should take 5-10 minutes to solve
-- Show the correct solution process
+- Perfect difficulty for {grade_level}
+- Clear, interesting problem statement
+- 5-15 minutes to solve
+- Include any necessary context or diagrams (text description)
 
-Format your response as:
+Format:
 
-**📝 Practice Problem:**
-[Clear problem statement]
+**📝 PROBLEM:**
+[Clear problem statement - make it interesting!]
 
-**💭 Hints:**
-[2-3 helpful hints to get started]
+**💭 HELPFUL HINTS:**
+Hint 1: [First step guidance]
+Hint 2: [Process guidance]  
+Hint 3: [Concept check]
+
+**🎯 DIFFICULTY:** [Easy/Medium/Hard]
+
+**⏱️ EXPECTED TIME:** [5-10 minutes]
+
+**✅ COMPLETE SOLUTION (For After You Try):**
+[Step-by-step solution with explanations]
 
-**✅ Solution (Hidden - for teacher reference):**
-[Step-by-step solution]
+**📚 CONCEPT CHECK:**
+[What this problem teaches you]
 
-Make it engaging and educational!"""
+Make it FUN and ENCOURAGING!"""
     
     response, source = ask_ai(prompt, temperature=0.6)
     
-    if source in ["cohere", "hf"]:
-        response = f"*[Generated by {source.upper()}]*\n\n{response}"
+    formatted = f"""
+╔════════════════════════════════════════════════════════════════╗
+║             📝 YOUR PERSONALIZED PRACTICE PROBLEM             ║
+╚════════════════════════════════════════════════════════════════╝
+
+{response}
+
+╔════════════════════════════════════════════════════════════════╗
+║  📌 NEXT STEP: Solve this on paper, then upload to get help!  ║
+╚════════════════════════════════════════════════════════════════╝
+"""
     
-    return response
+    return formatted
 
 # ---------- 6. AI Tutor Chat ----------
 def ai_tutor_chat(message, history, grade_level, topic):
@@ -267,37 +303,46 @@ def ai_tutor_chat(message, history, grade_level, topic):
         history.append((message, "⚠️ Please select a topic first to get focused help!"))
         return history
     
-    system = f"""You are an expert mathematics tutor for {grade_level} students.
-Current topic: {topic}
+    system = f"""You are a world-class mathematics tutor specialized in teaching {grade_level} students.
+Current topic focus: {topic}
 
-Your role:
-- Provide clear, step-by-step explanations
+Your approach:
+1. LISTEN to their question and understand exactly what they're confused about
+2. EXPLAIN using analogies, visuals descriptions, and real examples
+3. BREAK DOWN complex concepts into simple steps
+4. ASK CLARIFYING questions to check understanding
+5. PROVIDE examples and practice opportunities
+6. ENCOURAGE and motivate them
+
+PERSONALITY:
+- Warm, friendly, and patient
+- Use emojis occasionally to stay engaging
+- Celebrate their progress
+- Make math FUN and RELEVANT
 - Use age-appropriate language for {grade_level}
-- Give practical examples
-- Be encouraging and patient
-- Break down complex concepts
-- Check for understanding
 
-Answer the student's question about {topic} in a way that helps them truly understand."""
+Remember: You're not just answering - you're building understanding and confidence!"""
     
     # Build conversation context
     conversation = ""
-    for user_msg, bot_msg in history[-3:]:
+    for user_msg, bot_msg in history[-4:]:
         if user_msg:
             conversation += f"Student: {user_msg}\n"
         if bot_msg:
-            clean_msg = bot_msg.replace("🔵 ", "").replace("🟢 ", "")
+            clean_msg = bot_msg.replace("🔵 ", "").replace("🟢 ", "").replace("📌 ", "")
             conversation += f"Tutor: {clean_msg}\n"
     
-    full_prompt = f"{system}\n\n{conversation}Student: {message}\nTutor:"
+    full_prompt = f"{system}\n\n{conversation}\nStudent: {message}\n\nRespond as a caring, expert tutor:"
     
     bot_response, source = ask_ai(full_prompt, temperature=0.7)
     
-    # Add indicators for non-primary models
+    # Add source indicator
     if source == "cohere":
         bot_response = f"🔵 {bot_response}"
     elif source == "hf":
         bot_response = f"🟢 {bot_response}"
+    else:
+        bot_response = f"📌 {bot_response}"
     
     history.append((message, bot_response))
     return history
@@ -310,55 +355,209 @@ def update_topics(grade_level):
     topics = syllabus_topics["math"].get(grade_level, [])
     return gr.Dropdown(choices=topics, value=None)
 
-# ---------- 8. Gradio UI ----------
-with gr.Blocks(theme=gr.themes.Soft(), title="Math Learning Assistant", css="""
+# ---------- 8. Session & Progress Tracking ----------
+import json
+from pathlib import Path
+
+class StudentSession:
+    def __init__(self, student_name="Student"):
+        self.student_name = student_name
+        self.start_time = datetime.now()
+        self.problems_attempted = 0
+        self.correct_solutions = 0
+        self.total_points = 0
+        self.session_history = []
+        self.achievements = []
+        
+    def log_attempt(self, grade, topic, result):
+        """Log a learning attempt"""
+        self.problems_attempted += 1
+        self.session_history.append({
+            'timestamp': datetime.now().isoformat(),
+            'grade': grade,
+            'topic': topic,
+            'result': result
+        })
+    
+    def get_stats(self):
+        """Get session statistics"""
+        accuracy = (self.correct_solutions / max(self.problems_attempted, 1)) * 100
+        return {
+            'problems': self.problems_attempted,
+            'correct': self.correct_solutions,
+            'accuracy': accuracy,
+            'points': self.total_points,
+            'achievements': len(self.achievements)
+        }
+
+session = StudentSession()
+
+# ---------- 9. Gradio UI with Enhanced Design ----------
+with gr.Blocks(theme=gr.themes.Soft(), title="✨ Smart Math Learning Assistant", css="""
+    :root {
+        --primary-color: #6366f1;
+        --success-color: #10b981;
+        --warning-color: #f59e0b;
+        --danger-color: #ef4444;
+    }
+    
     .gradio-container {
-        font-family: 'Inter', sans-serif;
+        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
         max-width: 1400px;
         margin: auto;
+        background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
+        padding: 20px;
+    }
+    
+    .header-section {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        padding: 30px;
+        border-radius: 15px;
+        color: white;
+        margin-bottom: 30px;
+        box-shadow: 0 10px 30px rgba(0,0,0,0.2);
+    }
+    
+    .header-section h1 {
+        margin: 0;
+        font-size: 2.5em;
+        font-weight: 700;
+    }
+    
+    .stats-card {
+        background: white;
+        padding: 20px;
+        border-radius: 12px;
+        border-left: 5px solid var(--primary-color);
+        box-shadow: 0 4px 15px rgba(0,0,0,0.08);
     }
-    .tab-nav button {font-weight: 600;}
+    
     .feedback-box {
-        background: #f8f9fa;
-        border-left: 4px solid #4CAF50;
+        background: linear-gradient(135deg, #f0fdf4 0%, #dcfce7 100%);
+        border-left: 5px solid var(--success-color);
+        padding: 20px;
+        border-radius: 12px;
+        font-size: 14px;
+        line-height: 1.8;
+    }
+    
+    .error-box {
+        background: linear-gradient(135deg, #fef2f2 0%, #fee2e2 100%);
+        border-left: 5px solid var(--danger-color);
+        padding: 20px;
+        border-radius: 12px;
+    }
+    
+    .problem-box {
+        background: linear-gradient(135deg, #fef9e7 0%, #fef3c7 100%);
+        border-left: 5px solid var(--warning-color);
+        padding: 20px;
+        border-radius: 12px;
+    }
+    
+    .tab-button {
+        font-weight: 600;
+        font-size: 16px;
+        padding: 12px 24px !important;
+    }
+    
+    .achievement-box {
+        background: linear-gradient(135deg, #fef3c7 0%, #fcd34d 100%);
         padding: 15px;
+        border-radius: 10px;
+        text-align: center;
+        margin: 10px;
+    }
+    
+    button.primary-btn {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        color: white;
+        font-weight: 600;
+        border: none;
+        padding: 12px 30px;
         border-radius: 8px;
+        cursor: pointer;
+        transition: all 0.3s ease;
+    }
+    
+    button.primary-btn:hover {
+        transform: translateY(-2px);
+        box-shadow: 0 8px 20px rgba(102, 126, 234, 0.4);
+    }
+    
+    .dropdown-label {
+        font-weight: 600;
+        font-size: 15px;
+        color: #1f2937;
+    }
+    
+    .textbox-container textarea {
+        border-radius: 10px;
+        border: 2px solid #e5e7eb;
+        font-size: 14px;
     }
 """) as app:
     
     gr.Markdown("""
-    # ✍️ Math Learning Assistant
-    ### AI-Powered Handwriting Recognition & Tutoring System
-    **Write your math solutions naturally - Get instant AI feedback!**
+    # ✨ Smart Math Learning Assistant
+    ### AI-Powered Adaptive Learning Platform
+    **Learn math the smart way - Write naturally, get instant AI feedback!**
     
-    _🤖 Powered by Gemini AI (Vision) with intelligent fallback system (Cohere → Hugging Face)_
+    _🚀 Personalized learning for every student | 🤖 Advanced AI-powered analysis | 📊 Real-time progress tracking_
     """)
     
-    # Grade Level and Topic Selection (Global)
-    with gr.Row():
-        grade_level = gr.Dropdown(
-            choices=list(syllabus_topics["math"].keys()),
-            label="📚 Select Your Grade Level",
-            value="Primary 3-4",
-            info="Choose your current grade level"
-        )
-        
-        topic_select = gr.Dropdown(
-            choices=syllabus_topics["math"]["Primary 3-4"],
-            label="📖 Select Topic",
-            info="Choose the topic you want to practice"
-        )
+    # Grade Level and Topic Selection (Global) with Student Name
+    with gr.Group():
+        gr.Markdown("### 👤 Your Learning Profile")
+        with gr.Row():
+            student_name_input = gr.Textbox(
+                label="📝 Your Name (Optional)",
+                placeholder="Enter your name for personalized learning",
+                value="Student"
+            )
+            grade_level = gr.Dropdown(
+                choices=list(syllabus_topics["math"].keys()),
+                label="📚 Grade Level",
+                value="Primary 3-4",
+                info="Select your current grade"
+            )
+            
+            topic_select = gr.Dropdown(
+                choices=syllabus_topics["math"]["Primary 3-4"],
+                label="📖 Topic",
+                info="Choose what to practice"
+            )
+    
+    # Progress Stats Bar
+    with gr.Group():
+        gr.Markdown("### 📊 Your Progress Today")
+        with gr.Row():
+            problems_stat = gr.Textbox(value="Problems: 0", interactive=False, scale=1, container=False)
+            correct_stat = gr.Textbox(value="Correct: 0", interactive=False, scale=1, container=False)
+            accuracy_stat = gr.Textbox(value="Accuracy: 0%", interactive=False, scale=1, container=False)
+            points_stat = gr.Textbox(value="Points: 0", interactive=False, scale=1, container=False)
     
     # Update topics when grade changes
     grade_level.change(update_topics, grade_level, topic_select)
     
+    def update_student_name(name):
+        session.student_name = name if name else "Student"
+        return f"Welcome, {session.student_name}! 👋"
+    
+    student_name_input.change(update_student_name, student_name_input, None)
+    
     with gr.Tabs():
         # ===== TAB 1: HANDWRITING ANALYSIS =====
         with gr.Tab("✍️ Write & Check Work"):
             gr.Markdown("""
-            ### Write Your Math Solution
-            Draw your solution using the canvas below OR upload a photo of your written work on paper.
-            The AI will analyze your steps, identify errors, and help you improve!
+            ### 📝 Smart Solution Analyzer
+            
+            **Three ways to submit your work:**
+            1. ✏️ **Draw** your solution using the canvas
+            2. 📸 **Upload** a photo of your written work on paper
+            3. 🖥️ **Type** if you prefer digital input
+            
+            The AI Teacher will analyze every step and provide personalized feedback!
             """)
             
             with gr.Row():
@@ -366,7 +565,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Math Learning Assistant", css="""
                     with gr.Tabs():
                         with gr.Tab("🖊️ Draw Here"):
                             sketchpad = gr.Sketchpad(
-                                label="Write your solution (use finger/mouse/stylus)",
+                                label="✏️ Draw or write your solution",
                                 type="pil",
                                 brush=gr.Brush(
                                     colors=["#000000", "#0000FF", "#FF0000", "#00AA00"],
@@ -377,34 +576,74 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Math Learning Assistant", css="""
                         
                         with gr.Tab("📸 Upload Photo"):
                             image_upload = gr.Image(
-                                label="Upload photo of your written work",
+                                label="📷 Upload a clear photo of your work",
                                 type="pil",
                                 height=500
                             )
+                        
+                        with gr.Tab("📝 Type Solution"):
+                            text_input = gr.Textbox(
+                                label="Or describe/type your solution here",
+                                lines=10,
+                                placeholder="Type your mathematical working, steps, or solution...",
+                            )
                     
                     with gr.Row():
                         check_btn = gr.Button("🔍 Check My Work", variant="primary", size="lg", scale=2)
                         clear_canvas_btn = gr.Button("🗑️ Clear", size="lg", scale=1)
                 
                 with gr.Column(scale=1):
-                    gr.Markdown("### 📊 AI Teacher's Feedback")
+                    gr.Markdown("### 🎯 AI Teacher's Detailed Feedback")
                     feedback_output = gr.Textbox(
-                        label="Detailed Analysis & Feedback",
+                        label="Your Personalized Analysis",
                         lines=25,
-                        placeholder="Your personalized feedback will appear here...\n\n✓ Problem identification\n✓ Step-by-step analysis\n✓ Error detection\n✓ Correct solution\n✓ Learning tips",
+                        placeholder="📋 Your detailed feedback will appear here:\n\n✓ Problem you solved\n✓ Your approach analysis\n✓ Correct/incorrect assessment\n✓ Step-by-step explanation\n✓ Correct solution\n✓ Learning tips",
                         elem_classes="feedback-box"
                     )
             
             # Event handlers for handwriting check
-            def check_work_handler(sketch, upload, grade, topic):
+            def check_work_handler(sketch, upload, text, grade, topic):
+                if sketch is None and upload is None and (not text or not text.strip()):
+                    return "⚠️ Please draw, upload an image, or type your solution!"
+                
                 image_to_analyze = sketch if sketch is not None else upload
-                if image_to_analyze is None:
-                    return "⚠️ Please draw your solution or upload an image first!"
-                return analyze_math_work(image_to_analyze, grade, topic)
+                
+                if image_to_analyze:
+                    feedback = analyze_math_work(image_to_analyze, grade, topic)
+                else:
+                    # Use text-based analysis
+                    system_prompt = f"""You are an encouraging math teacher for {grade} students learning {topic}.
+The student has submitted: {text}
+
+Analyze their solution like this:
+
+**📝 Problem Identified:**
+[What they're solving]
+
+**👀 Analysis of Their Work:**
+[What they did - is it correct?]
+
+**✅ Evaluation:**
+[Correct? If not, where exactly is the error?]
+
+**💡 Detailed Explanation:**
+[Step-by-step correct approach]
+
+**🌟 Encouragement & Tips:**
+[Positive feedback and 2-3 improvement tips]
+
+Be kind, specific, and educational!"""
+                    feedback, _ = ask_ai(system_prompt, temperature=0.5)
+                
+                # Update stats
+                session.problems_attempted += 1
+                session.log_attempt(grade, topic, "checked")
+                
+                return feedback
             
             check_btn.click(
                 fn=check_work_handler,
-                inputs=[sketchpad, image_upload, grade_level, topic_select],
+                inputs=[sketchpad, image_upload, text_input, grade_level, topic_select],
                 outputs=feedback_output
             )
             
@@ -413,50 +652,97 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Math Learning Assistant", css="""
         # ===== TAB 2: PRACTICE PROBLEMS =====
         with gr.Tab("📝 Practice Questions"):
             gr.Markdown("""
-            ### Generate Practice Problems
-            Get custom practice problems for your level and topic. Solve them on paper or digitally!
+            ### 🎲 Get Custom Practice Problems
+            
+            Generated problems are **perfectly suited** to your level and topic!
+            - Solve them step-by-step on paper or digitally
+            - Upload your solution above to get feedback
+            - Challenge yourself with different problems
             """)
             
             with gr.Row():
-                generate_btn = gr.Button("🎲 Generate New Problem", variant="primary", size="lg")
+                generate_btn = gr.Button("🎲 Generate New Problem", variant="primary", size="lg", scale=1)
+                difficulty_level = gr.Radio(
+                    choices=["Easy 😊", "Medium 😐", "Hard 😤"],
+                    value="Medium 😐",
+                    label="Difficulty Level",
+                    scale=1
+                )
             
             practice_output = gr.Textbox(
-                label="Practice Problem",
-                lines=15,
-                placeholder="Click 'Generate New Problem' to get started!",
-                interactive=False
+                label="📖 Your Practice Problem",
+                lines=18,
+                placeholder="Click 'Generate New Problem' to get started!\n\nYou'll get:\n✓ Clear problem statement\n✓ Helpful hints\n✓ Full solution for reference",
+                elem_classes="problem-box"
             )
             
+            attempt_info = gr.Textbox(
+                label="How to use",
+                value="1️⃣ Read the problem carefully\n2️⃣ Solve it on paper or in the canvas\n3️⃣ Go to 'Write & Check Work' tab\n4️⃣ Get instant AI feedback!",
+                interactive=False,
+                lines=4
+            )
+            
+            def generate_with_difficulty(grade, topic, diff):
+                diff_text = diff.split()[0].lower()
+                prompt = f"""Generate ONE practice problem for {grade} students on "{topic}"
+Difficulty: {diff_text}
+
+Format:
+**📝 Problem:**
+[Clear statement]
+
+**💭 Hints:**
+[2-3 hints]
+
+**✅ Solution:**
+[Step-by-step]"""
+                response, _ = ask_ai(prompt, temperature=0.6)
+                session.problems_attempted += 1
+                return response
+            
             generate_btn.click(
-                fn=generate_practice_problem,
-                inputs=[grade_level, topic_select],
+                fn=generate_with_difficulty,
+                inputs=[grade_level, topic_select, difficulty_level],
                 outputs=practice_output
             )
         
         # ===== TAB 3: AI TUTOR CHAT =====
         with gr.Tab("💬 Ask AI Tutor"):
             gr.Markdown("""
-            ### Chat with Your AI Math Tutor
-            Ask questions about any math concept. Get clear, step-by-step explanations!
+            ### 🤖 24/7 Interactive AI Tutor
+            
+            Ask anything about your topic! Your AI tutor will:
+            - 📚 Explain complex concepts clearly
+            - 📐 Show step-by-step solutions
+            - ❓ Answer your questions patiently
+            - 💡 Provide real-world examples
             """)
             
             chatbot = gr.Chatbot(
                 height=500,
                 show_label=False,
-                placeholder="👋 Hi! I'm your AI math tutor. Ask me anything about the topic you selected!"
+                placeholder="👋 Hi! I'm your personal math tutor. Ask me anything about the topic you selected! Examples: 'How do I solve quadratic equations?', 'Can you explain fractions?'",
+                bubble_full_width=False
             )
             
             with gr.Row():
                 msg_input = gr.Textbox(
-                    placeholder="Ask your question here... (e.g., 'How do I solve 3x + 5 = 20?')",
+                    placeholder="Ask your question... (e.g., 'What is the formula for area?')",
                     label="Your Question",
                     lines=2,
                     scale=4
                 )
-                send_btn = gr.Button("📤 Send", variant="primary", scale=1)
+                send_btn = gr.Button("📤 Send", variant="primary", scale=1, size="lg")
             
             with gr.Row():
-                clear_chat_btn = gr.Button("🗑️ Clear Chat")
+                clear_chat_btn = gr.Button("🗑️ Clear Chat", scale=1)
+                hint_text = gr.Textbox(
+                    value="💡 Tip: Be specific! Ask about one concept at a time.",
+                    interactive=False,
+                    scale=3,
+                    container=False
+                )
             
             send_btn.click(
                 ai_tutor_chat,
@@ -470,30 +756,123 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Math Learning Assistant", css="""
             )
             clear_chat_btn.click(clear_chat, outputs=chatbot)
     
-    # Footer with tips
+        # ===== TAB 4: PROGRESS & ACHIEVEMENTS =====
+        with gr.Tab("🏆 Progress & Achievements"):
+            gr.Markdown("""
+            ### 📈 Your Learning Journey
+            
+            Track your improvement, earn badges, and celebrate your wins!
+            """)
+            
+            with gr.Row():
+                with gr.Column(scale=1):
+                    gr.Markdown("### 📊 Session Statistics")
+                    total_problems = gr.Textbox(
+                        label="Total Problems Attempted",
+                        value="0",
+                        interactive=False
+                    )
+                    session_time = gr.Textbox(
+                        label="Time Spent Learning",
+                        value="0 min",
+                        interactive=False
+                    )
+                    topics_practiced = gr.Textbox(
+                        label="Topics Practiced",
+                        value="0",
+                        interactive=False
+                    )
+                
+                with gr.Column(scale=1):
+                    gr.Markdown("### 🎯 Your Achievements")
+                    achievement_display = gr.Textbox(
+                        label="Unlocked Badges",
+                        value="🎯 Welcome Badge (Just joined!)\n🔥 Keep learning to unlock more!",
+                        lines=8,
+                        interactive=False
+                    )
+            
+            def get_session_stats():
+                stats = session.get_stats()
+                time_spent = (datetime.now() - session.start_time).seconds // 60
+                return {
+                    'total': str(stats['problems']),
+                    'time': f"{time_spent} min",
+                    'topics': str(len(set([h['topic'] for h in session.session_history]))),
+                }
+            
+            with gr.Row():
+                refresh_btn = gr.Button("🔄 Refresh Stats", variant="primary")
+            
+            def refresh_handler():
+                s = get_session_stats()
+                return s['total'], s['time'], s['topics']
+            
+            refresh_btn.click(
+                fn=refresh_handler,
+                inputs=[],
+                outputs=[total_problems, session_time, topics_practiced]
+            )
+    
+    # ===== TAB 5: TIPS & GUIDE =====
+        with gr.Tab("📚 Learning Guide"):
+            gr.Markdown("""
+            ### 🎓 How to Get the Most from This App
+            
+            #### 🖊️ **Write & Check Work**
+            - Write your solution clearly, showing every step
+            - Include the problem statement
+            - Use proper mathematical notation
+            - Upload a clear photo if handwriting
+            
+            #### 📝 **Practice Questions**
+            - Start with the difficulty that challenges you
+            - Try solving BEFORE looking at hints
+            - Review the solution to understand concepts
+            - Repeat different problems on the same topic
+            
+            #### 💬 **Ask AI Tutor**
+            - Ask one specific question at a time
+            - Use mathematical terminology when possible
+            - Ask "why" questions to understand concepts deeper
+            - Request step-by-step explanations
+            
+            #### ✅ **Tips for Success**
+            1. **Practice Regularly** - Consistency beats cramming
+            2. **Understand, Don't Memorize** - Ask "why" for every step
+            3. **Review Mistakes** - Your errors are your best teachers
+            4. **Work Progressively** - Easy → Medium → Hard
+            5. **Take Breaks** - Learning in focused 25-minute sessions works best
+            
+            #### 🚀 **Pro Tips**
+            - Use the eraser before submitting to fix small errors
+            - Request hints before checking the full solution
+            - Create a notebook to track your improvements
+            - Try to explain solutions to someone else
+            - Come back to difficult problems the next day
+            """)
+    
+    # Footer
     gr.Markdown("""
     ---
-    ### 💡 Tips for Best Results:
+    ### 🌟 **Why Choose This Learning Platform?**
     
-    **For Handwriting Analysis:**
-    - ✓ Write clearly and show all your working steps
-    - ✓ Use proper mathematical notation
-    - ✓ If uploading a photo, ensure good lighting and focus
-    - ✓ Include the problem statement at the top
+    | Feature | Benefit |
+    |---------|---------|
+    | 🤖 **AI-Powered Analysis** | Get intelligent, personalized feedback on every solution |
+    | 🎨 **Multiple Input Methods** | Draw, upload photos, or type - choose what works for you |
+    | 📊 **Progress Tracking** | See your improvement over time |
+    | 💬 **24/7 AI Tutor** | Get help anytime, anywhere |
+    | 🎲 **Infinite Practice** | Never run out of problems to solve |
+    | ✨ **Adaptive Learning** | Personalized to your level and pace |
     
-    **For Learning:**
-    - ✓ Don't worry about making mistakes - they help you learn!
-    - ✓ Try to solve problems on your own first
-    - ✓ Review the AI feedback carefully
-    - ✓ Practice regularly for best results
+    ---
     
-    **AI System Status:**
-    - 🟢 **Gemini** (Primary): Vision-enabled analysis
-    - 🔵 **Cohere** (Secondary): Text-based fallback
-    - 🟢 **Hugging Face** (Tertiary): Final fallback
+    **Made with ❤️ for students who want to truly understand mathematics**
     
-    ---
-    _Made with ❤️ for students learning mathematics_
+    _Current AI System: Gemini (Primary) → Cohere (Secondary) → Hugging Face (Backup)_
+    
+    **Version 2.0 - Enhanced for Better Learning** ✨
     """)
 
 # Launch the app