Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from google import genai
+from groq import Groq
 from docx import Document
 from docx.shared import Pt, RGBColor, Inches
 from docx.enum.text import WD_ALIGN_PARAGRAPH
@@ -9,11 +9,13 @@ from datetime import datetime
 import os
 import re
 
-# Initialize Gemini client with error handling
+# Initialize Groq client with error handling
 try:
-    client = genai.Client()
+    client = Groq()
+    # Test the client
+    print("✓ Groq client initialized successfully")
 except Exception as e:
-    print(f"Warning: Gemini client initialization failed: {e}")
+    print(f"Warning: Groq client initialization failed: {e}")
     client = None
 
 # Comprehensive data dictionaries
@@ -147,12 +149,12 @@ def generate_lesson_plan(language, grade_range, grade_specific, subject, topic,
     # Validate inputs
     validation_errors = validate_inputs(language, grade_range, subject, topic, teacher_name, school_name)
     if validation_errors:
-        error_msg = "WARNING: Please fix the following errors:\n" + "\n".join(f"- {e}" for e in validation_errors)
+        error_msg = "⚠️ Please fix the following errors:\n" + "\n".join(f"• {e}" for e in validation_errors)
         return None, error_msg
     
     # Check if client is initialized
     if client is None:
-        return None, "ERROR: Gemini API client not initialized. Please check your API key."
+        return None, "❌ Error: Groq API client not initialized. Please check your API key."
     
     # Format date
     if date_input:
@@ -177,8 +179,7 @@ def generate_lesson_plan(language, grade_range, grade_specific, subject, topic,
     assessment_time = duration - (intro_time + main_time + conclusion_time)
     
     # Build enhanced prompt
-    prompt = f"""
-You are an expert Pakistani educator and curriculum specialist with deep knowledge of the National Curriculum Framework, modern pedagogical approaches, and classroom management in Gilgit-Baltistan context.
+    prompt = f"""You are an expert Pakistani educator and curriculum specialist with deep knowledge of the National Curriculum Framework, modern pedagogical approaches, and classroom management in Gilgit-Baltistan context.
 
 Create a COMPREHENSIVE, PRACTICAL, and STUDENT-CENTERED SMART LESSON PLAN in {language} language.
 
@@ -282,7 +283,7 @@ Create a COMPREHENSIVE, PRACTICAL, and STUDENT-CENTERED SMART LESSON PLAN in {la
    - How well were SLOs achieved?
 
 **FORMATTING & LANGUAGE GUIDELINES:**
-{"For English version:" if language == "English" else "For Urdu version:"}
+{"For English version:" if language == "English" else "اردو کے لیے:"}
 - Use clear, professional language appropriate for educational documentation
 - Write in complete sentences and paragraphs
 - Use bullet points ONLY for lists of items or steps
@@ -290,24 +291,47 @@ Create a COMPREHENSIVE, PRACTICAL, and STUDENT-CENTERED SMART LESSON PLAN in {la
 - Maintain logical flow between sections
 - Be specific and practical, not vague or generic
 - Include culturally relevant examples from Gilgit-Baltistan context
-- {"Proper English grammar and spelling" if language == "English" else "Proper Urdu script with right-to-left alignment"}
+- {"Proper English grammar and spelling" if language == "English" else "صحیح اردو رسم الخط اور املا، دائیں سے بائیں سمت"}
 
 **OUTPUT FORMAT:**
 Generate ONLY the lesson plan content starting from "SMART STUDENT LEARNING OUTCOMES (SLOs)". 
 Do NOT include any table or repeat the lesson details (Grade, Subject, etc.) as that will be added automatically.
 Do NOT add any developer credits, signatures, or metadata at the end.
 
-Now generate the complete, ready-to-implement lesson plan:
-"""
-    
+Now generate the complete, ready-to-implement lesson plan:"""
+
     try:
-        # Generate content using Gemini
-        response = client.models.generate_content(
-            model="gemini-2.0-flash-exp",
-            contents=prompt
+        # Generate content using Groq API with streaming
+        print("🔄 Generating lesson plan with Groq API...")
+        
+        completion = client.chat.completions.create(
+            model="openai/gpt-oss-120b",
+            messages=[
+                {
+                    "role": "system",
+                    "content": "You are an expert Pakistani educator and curriculum specialist. Generate comprehensive, professional lesson plans following the exact format and requirements provided."
+                },
+                {
+                    "role": "user",
+                    "content": prompt
+                }
+            ],
+            temperature=0.7,  # Slightly lower for more consistent educational content
+            max_completion_tokens=8192,
+            top_p=0.95,
+            reasoning_effort="high",  # Use high reasoning for better educational content
+            stream=True,
+            stop=None
         )
         
-        lesson_text = response.text.strip()
+        # Collect the streamed response
+        lesson_text = ""
+        for chunk in completion:
+            if chunk.choices[0].delta.content:
+                lesson_text += chunk.choices[0].delta.content
+        
+        lesson_text = lesson_text.strip()
+        print(f"✓ Generated {len(lesson_text)} characters")
         
         # Remove any accidental inclusion of lesson details from AI response
         lesson_text = re.sub(r'^.*?(SMART|SLO)', r'\1', lesson_text, flags=re.DOTALL | re.IGNORECASE)
@@ -384,7 +408,6 @@ Now generate the complete, ready-to-implement lesson plan:
             
             # Process lesson content with enhanced formatting
             lines = lesson_text.split("\n")
-            current_section = ""
             
             for line in lines:
                 line_stripped = line.strip()
@@ -439,7 +462,7 @@ Now generate the complete, ready-to-implement lesson plan:
             run.font.italic = True
             run.font.color.rgb = RGBColor(100, 100, 100)
             
-            run2 = footer.add_run("SMART Lesson Plan Generator v2.0")
+            run2 = footer.add_run("SMART Lesson Plan Generator v2.1 (Powered by Groq AI)")
             run2.font.size = Pt(8)
             run2.font.italic = True
             run2.font.color.rgb = RGBColor(100, 100, 100)
@@ -450,30 +473,31 @@ Now generate the complete, ready-to-implement lesson plan:
             file_path = f"/tmp/LP_{subject}_{safe_topic}_{full_grade}_{formatted_date}.docx"
             doc.save(file_path)
             
-            status_msg = f"""SUCCESS: Lesson Plan Generated Successfully!
+            status_msg = f"""✅ **Lesson Plan Generated Successfully!**
 
-FILE DETAILS:
+📄 **File Details:**
 - Subject: {subject}
 - Grade: {full_grade}
 - Topic: {topic}
 - Duration: {duration} minutes
 - Date: {formatted_date}
 
-FILE NAME: {os.path.basename(file_path)}
+💾 **File Name:** {os.path.basename(file_path)}
 
-NEXT STEPS:
+✨ **Next Steps:**
 1. Download the file using the button below
 2. Review and customize as needed
 3. Print or share with colleagues
 4. Use in your classroom!
 
-QUALITY ASSURANCE:
-- SMART SLOs aligned with Bloom's Taxonomy
-- Detailed lesson procedure with timings
-- Assessment strategies included
-- {'Teaching resources listed' if include_resources else 'Standard content included'}
-- {'Differentiation strategies added' if include_differentiation else 'Standard content included'}
-- Home assignment provided"""
+🎯 **Quality Assurance:**
+- ✓ SMART SLOs aligned with Bloom's Taxonomy
+- ✓ Detailed lesson procedure with timings
+- ✓ Assessment strategies included
+- ✓ {('Teaching resources listed' if include_resources else 'Differentiation strategies added' if include_differentiation else 'Complete lesson plan')}
+- ✓ Home assignment provided
+
+🤖 **Powered by:** Groq AI (openai/gpt-oss-120b)"""
             
         else:  # Urdu version
             # Create Urdu document
@@ -510,33 +534,41 @@ QUALITY ASSURANCE:
             file_path = f"/tmp/LP_Urdu_{safe_topic}_{formatted_date}.docx"
             doc.save(file_path)
             
-            status_msg = f"""SUCCESS: سبق کا منصوبہ تیار ہو گیا!
+            status_msg = f"""✅ **سبق کا منصوبہ تیار ہو گیا!**
 
-تفصیلات:
+📄 **تفصیلات:**
 - مضمون: {subject}
 - درجہ: {full_grade}
 - موضوع: {topic}
 
-فائل: {os.path.basename(file_path)}"""
+💾 فائل: {os.path.basename(file_path)}
+
+🤖 **تیار کردہ بذریعہ:** Groq AI"""
         
         return file_path, status_msg
         
     except Exception as e:
-        error_details = f"""ERROR: Error Generating Lesson Plan
+        error_details = f"""❌ **Error Generating Lesson Plan**
 
-Error Type: {type(e).__name__}
-Details: {str(e)}
+**Error Type:** {type(e).__name__}
+**Details:** {str(e)}
 
-Troubleshooting Steps:
+**Troubleshooting Steps:**
 1. Check your internet connection
-2. Verify Gemini API key is configured
+2. Verify Groq API key is configured (GROQ_API_KEY environment variable)
 3. Ensure all required fields are filled
 4. Try again with a simpler topic
+5. Check if you have sufficient API credits
+
+**API Configuration:**
+- Set your Groq API key: `export GROQ_API_KEY=your_api_key_here`
+- Get your API key from: https://console.groq.com/
 
-If problem persists: Contact technical support"""
+**If problem persists:** Contact technical support"""
         
         return None, error_details
 
+
 # Create Professional Gradio Interface
 custom_css = """
 .gradio-container {
@@ -571,12 +603,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
     with gr.Row():
         gr.HTML("""
         <div class="main-header">
-            <h1 style="margin:0; font-size: 2.5rem;">SMART Lesson Plan Generator</h1>
+            <h1 style="margin:0; font-size: 2.5rem;">📚 SMART Lesson Plan Generator</h1>
             <p style="margin: 0.5rem 0 0 0; font-size: 1.1rem; opacity: 0.9;">
                 Professional Lesson Planning Tool for Gilgit-Baltistan Schools
             </p>
             <p style="margin: 0.3rem 0 0 0; font-size: 0.9rem; opacity: 0.8;">
-                Powered by Google Gemini AI | Developed by Najaf Ali Sharqi
+                Powered by Groq AI (openai/gpt-oss-120b) | Developed by Najaf Ali Sharqi
             </p>
         </div>
         """)
@@ -585,31 +617,33 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
     with gr.Row():
         gr.Markdown("""
         <div class="info-box">
-        <strong>About This Tool:</strong> Generate comprehensive, curriculum-aligned SMART lesson plans in English or Urdu. 
+        <strong>🎯 About This Tool:</strong> Generate comprehensive, curriculum-aligned SMART lesson plans in English or Urdu. 
         All plans include SMART SLOs based on Bloom's Taxonomy, detailed procedures, assessments, and home assignments.
+        <br><br>
+        <strong>🤖 AI Model:</strong> Using Groq's openai/gpt-oss-120b for high-quality, fast lesson plan generation.
         </div>
         """)
     
     # Main form sections
-    gr.HTML('<div class="section-header"><h3>Basic Lesson Information</h3></div>')
+    gr.HTML('<div class="section-header"><h3>📋 Basic Lesson Information</h3></div>')
     
     with gr.Row():
         language = gr.Dropdown(
             choices=["English", "Urdu"], 
-            label="Language",
+            label="📝 Language",
             value="English",
             info="Select lesson plan language",
             interactive=True
         )
         grade_range = gr.Dropdown(
             choices=["Grade 1-5", "Grade 6-8", "Grade 9-10", "Grade 11-12"],
-            label="Grade Range",
+            label="🎓 Grade Range",
             info="Select grade level range",
             interactive=True
         )
         grade_specific = gr.Dropdown(
             choices=[f"Grade {i}" for i in range(1, 13)],
-            label="Specific Grade (Optional)",
+            label="📊 Specific Grade (Optional)",
             info="Select exact grade level",
             interactive=True
         )
@@ -617,51 +651,51 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
     with gr.Row():
         subject = gr.Dropdown(
             choices=[],
-            label="Subject",
+            label="📖 Subject",
             info="Will update based on grade selection",
             interactive=True
         )
         topic = gr.Textbox(
-            label="Lesson Topic",
+            label="📌 Lesson Topic",
             placeholder="e.g., Photosynthesis, Pythagorean Theorem, Parts of Speech...",
             info="Enter the specific topic for this lesson",
             lines=1
         )
     
-    gr.HTML('<div class="section-header"><h3>School & Teacher Details</h3></div>')
+    gr.HTML('<div class="section-header"><h3>👨‍🏫 School & Teacher Details</h3></div>')
     
     with gr.Row():
         school_name = gr.Textbox(
-            label="School Name",
+            label="🏫 School Name",
             placeholder="e.g., Government High School Gilgit",
             info="Enter complete school name",
             lines=1
         )
         teacher_name = gr.Textbox(
-            label="Teacher Name",
+            label="👤 Teacher Name",
             placeholder="e.g., Muhammad Ali Khan",
             info="Enter teacher's full name",
             lines=1
         )
         district = gr.Dropdown(
             choices=DISTRICTS_GB,
-            label="District",
+            label="📍 District",
             value="Gilgit",
             info="Select district in Gilgit-Baltistan",
             interactive=True
         )
     
-    gr.HTML('<div class="section-header"><h3>Class Details & Timing</h3></div>')
+    gr.HTML('<div class="section-header"><h3>⏰ Class Details & Timing</h3></div>')
     
     with gr.Row():
         date_input = gr.Textbox(
-            label="Lesson Date",
+            label="📅 Lesson Date",
             value=datetime.now().strftime("%Y-%m-%d"),
             placeholder="YYYY-MM-DD",
             info="Enter date in YYYY-MM-DD format"
         )
         students = gr.Number(
-            label="Number of Students",
+            label="👥 Number of Students",
             value=35,
             minimum=1,
             maximum=100,
@@ -670,42 +704,42 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
         )
         duration = gr.Dropdown(
             choices=[35, 40, 45, 50, 60, 70, 80, 90],
-            label="Duration (minutes)",
+            label="⏱️ Duration (minutes)",
             value=40,
             info="Total lesson duration",
             interactive=True
         )
     
-    gr.HTML('<div class="section-header"><h3>Teaching Methodology & Assessment</h3></div>')
+    gr.HTML('<div class="section-header"><h3>🎯 Teaching Methodology & Assessment</h3></div>')
     
     with gr.Row():
         teaching_method = gr.Dropdown(
             choices=TEACHING_METHODS,
-            label="Primary Teaching Method",
+            label="🎓 Primary Teaching Method",
             value="Lecture Method",
             info="Main instructional approach for this lesson",
             interactive=True
         )
         assessment_type = gr.Dropdown(
             choices=ASSESSMENT_TYPES,
-            label="Assessment Method",
+            label="📊 Assessment Method",
             value="Formative Assessment",
             info="How will you evaluate student learning?",
             interactive=True
         )
     
-    gr.HTML('<div class="section-header"><h3>Additional Options</h3></div>')
+    gr.HTML('<div class="section-header"><h3>⚙️ Additional Options</h3></div>')
     
     with gr.Row():
         with gr.Column(scale=1):
             include_resources = gr.Checkbox(
-                label="Include Teaching Resources List",
+                label="📦 Include Teaching Resources List",
                 value=True,
                 info="Add detailed list of required materials and resources"
             )
         with gr.Column(scale=1):
             include_differentiation = gr.Checkbox(
-                label="Include Differentiation Strategies",
+                label="🎯 Include Differentiation Strategies",
                 value=True,
                 info="Add strategies for diverse learners (advanced, struggling, special needs)"
             )
@@ -714,14 +748,14 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
     
     # Generate button with styling
     generate_btn = gr.Button(
-        "Generate SMART Lesson Plan", 
+        "🚀 Generate SMART Lesson Plan", 
         variant="primary", 
         size="lg",
         scale=2
     )
     
     # Instructions
-    with gr.Accordion("Instructions & Tips", open=False):
+    with gr.Accordion("📖 Instructions & Tips", open=False):
         gr.Markdown("""
         ### How to Use This Generator:
         
@@ -734,7 +768,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
         7. **Choose Methods**: Select teaching and assessment approaches
         8. **Click Generate**: Wait 10-20 seconds for AI to create your plan
         
-        ### Tips for Best Results:
+        ### 💡 Tips for Best Results:
         
         - **Be Specific**: Instead of "Math", use "Linear Equations" or "Fractions"
         - **Realistic Duration**: Match duration to your actual class period
@@ -742,7 +776,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
         - **Review & Edit**: Always review and customize the generated plan
         - **Save Multiple**: Generate variations to compare approaches
         
-        ### What You'll Get:
+        ### ✨ What You'll Get:
         
         - **3 SMART SLOs** aligned with Bloom's Taxonomy (Knowledge, Understanding, Application)
         - **Comprehensive Content Knowledge** with key concepts and misconceptions
@@ -752,7 +786,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
         - **SMART Home Assignment** aligned with SLOs
         - **Reflection Section** for post-lesson notes
         
-        ### About SMART SLOs:
+        ### 🎓 About SMART SLOs:
         
         SMART stands for:
         - **S**pecific: Clear and precise learning outcomes
@@ -760,24 +794,37 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
         - **A**chievable: Realistic within lesson time
         - **R**elevant: Aligned with curriculum standards
         - **T**ime-bound: Achievable in one lesson period
+        
+        ### 🔑 API Configuration:
+        
+        **Before using this tool**, you need to set up your Groq API key:
+        
+        1. Get your free API key from: https://console.groq.com/
+        2. Set it as an environment variable:
+           ```bash
+           export GROQ_API_KEY=your_api_key_here
+           ```
+        3. Or add it to your `.env` file
+        
+        The app uses the **openai/gpt-oss-120b** model for high-quality educational content generation.
         """)
     
     gr.Markdown("---")
     
     # Output section with enhanced styling
-    gr.HTML('<div class="section-header"><h3>Your Generated Lesson Plan</h3></div>')
+    gr.HTML('<div class="section-header"><h3>📥 Your Generated Lesson Plan</h3></div>')
     
     with gr.Row():
         with gr.Column(scale=2):
             status_output = gr.Textbox(
-                label="Generation Status", 
+                label="📊 Generation Status", 
                 interactive=False,
-                lines=10,
+                lines=12,
                 placeholder="Status messages will appear here..."
             )
         with gr.Column(scale=1):
             output_file = gr.File(
-                label="Download File",
+                label="💾 Download File",
                 file_count="single"
             )
     
@@ -802,47 +849,73 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
     # Footer with additional information
     gr.Markdown("---")
     
-    with gr.Accordion("About & Support", open=False):
+    with gr.Accordion("ℹ️ About & Support", open=False):
         gr.Markdown("""
-        ### About This Application
+        ### 📱 About This Application
         
-        **Version:** 2.0 (Production Release)  
+        **Version:** 2.1 (Groq AI Edition)  
         **Developer:** Najaf Ali Sharqi  
         **Organization:** School Education Department, Gilgit-Baltistan  
-        **AI Model:** Google Gemini 2.0 Flash  
+        **AI Model:** Groq openai/gpt-oss-120b  
+        **Speed:** Ultra-fast generation (10-20 seconds)
+        
+        ### 🎯 Features:
+        
+        - ✅ SMART Learning Outcomes (Bloom's Taxonomy aligned)
+        - ✅ Comprehensive lesson procedures with timings
+        - ✅ National Curriculum Framework compliance
+        - ✅ Bilingual support (English & Urdu)
+        - ✅ Professional Word document formatting
+        - ✅ Differentiation strategies
+        - ✅ Assessment guidelines
+        - ✅ Home assignments
+        - ✅ Powered by Groq's ultra-fast AI inference
         
-        ### Features:
+        ### 🤖 About Groq AI:
         
-        - SMART Learning Outcomes (Bloom's Taxonomy aligned)
-        - Comprehensive lesson procedures with timings
-        - National Curriculum Framework compliance
-        - Bilingual support (English & Urdu)
-        - Professional Word document formatting
-        - Differentiation strategies
-        - Assessment guidelines
-        - Home assignments
+        This application uses Groq's **openai/gpt-oss-120b** model, which provides:
+        - Lightning-fast inference speed
+        - High-quality educational content
+        - Reasoning capabilities for better lesson structure
+        - Support for both English and Urdu languages
         
-        ### Support & Feedback:
+        ### 📞 Support & Feedback:
         
         For technical support, feature requests, or to report issues:
         - **Email:** Contact School Education Department
         - **Feedback:** Use the feedback option in your district office
+        - **API Issues:** Check https://console.groq.com/ for API status
         
-        ### Privacy & Data:
+        ### 🔒 Privacy & Data:
         
         - All lesson plans are generated in real-time
         - No data is stored on external servers
         - Your lesson plans remain confidential
         - Files are temporarily created and deleted after download
         
+        ### 📚 Resources:
+        
+        - **Groq Console:** https://console.groq.com/
+        - **Groq Documentation:** https://console.groq.com/docs
+        - **National Curriculum:** http://www.mofept.gov.pk/
+        - **Bloom's Taxonomy Guide:** Refer to teacher training materials
+        
+        ### 🙏 Acknowledgments:
+        
+        This tool was developed to support teachers in Gilgit-Baltistan in creating high-quality,
+        standards-aligned lesson plans efficiently. Special thanks to:
+        - All educators who provided feedback during development
+        - Groq AI for providing powerful, fast AI inference
+        - School Education Department, Gilgit-Baltistan
+        
         ---
         
-        **Copyright 2024 School Education Department, Gilgit-Baltistan**  
+        **© 2024 School Education Department, Gilgit-Baltistan**  
         *Empowering Teachers, Enriching Learning*
         """)
     
     # Quick reference guide
-    with gr.Accordion("Quick Reference: Bloom's Taxonomy", open=False):
+    with gr.Accordion("📋 Quick Reference: Bloom's Taxonomy", open=False):
         gr.Markdown("""
         ### Understanding Bloom's Taxonomy Levels
         
@@ -855,7 +928,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="SMART Lesson Plan
         | **Evaluation** (Evaluate) | Evaluate, Judge, Critique, Justify | "Students will be able to **evaluate** the effectiveness of different solutions" |
         | **Creation** (Create) | Create, Design, Develop, Compose | "Students will be able to **design** an experiment to test their hypothesis" |
         
-        ### Tips:
+        ### 💡 Tips:
         - Start with lower levels (Knowledge, Understanding) for new topics
         - Progress to higher levels (Application, Analysis) as students master basics
         - Each lesson should target 2-3 adjacent levels