Update app.py

app.py

@@ -1,4 +1,5 @@
 # app.py
+# app.py
 import os
 import gradio as gr
 from docx import Document
@@ -17,21 +18,25 @@ client = Groq(api_key=GROQ_API_KEY)
 # ✅ Text Generation Function
 def generate_lesson_plan(language, class_, subject, topic, students, duration, date, teacher):
     try:
+        # English mode → Llama 3.1 model
         if language == "English":
+            model_name = "llama-3.1-8b-instant"
             prompt = f"""
-You are an expert lesson plan creator.
-Create a detailed English lesson plan using the BOPPPS model for the topic: "{topic}".
-Avoid repeating the basic info (class, subject, teacher, etc.) in headings.
+You are an expert lesson plan designer.
+Create a comprehensive English lesson plan using the BOPPPS model for the topic: "{topic}".
+Do NOT repeat the basic info (class, subject, teacher, etc.) in headings.
 Include only these sections:
 1. Bridge-In: short, engaging introduction.
 2. 3 Clear Learning Objectives.
 3. 2 Pre-Assessment Questions.
-4. 1 Participatory Learning Activity (detailed).
+4. 1 Detailed Participatory Learning Activity.
 5. 2–3 Post-Assessment Questions.
 6. Summary paragraph by the teacher.
             """
 
-        else:  # Urdu mode
+        # Urdu mode → OpenAI GPT-OSS-120B model
+        else:
+            model_name = "openai/gpt-oss-120b"
             prompt = f"""
 آپ ایک ماہر اسباقی منصوبہ ساز ہیں۔
 موضوع "{topic}" پر BOPPPS ماڈل کے مطابق ایک مفصل اردو سبق منصوبہ تیار کریں۔
@@ -44,9 +49,9 @@ Include only these sections:
 6. سبق کا خلاصہ (Summary)
             """
 
-        # ✅ Use the Groq client with the openai/gpt-oss-120b model
+        # ✅ Generate text using selected model
         response = client.chat.completions.create(
-            model="openai/gpt-oss-120b",
+            model=model_name,
             messages=[{"role": "user", "content": prompt}],
             temperature=0.8,
             max_completion_tokens=2048,
@@ -54,7 +59,6 @@ Include only these sections:
             stream=False
         )
 
-        # Extract generated text
         lesson_text = response.choices[0].message.content.strip()
 
         # ✅ Create the Word document
@@ -80,7 +84,7 @@ Include only these sections:
 
         paragraph = doc.add_paragraph(lesson_text)
 
-        # Urdu formatting (right-aligned text)
+        # Urdu formatting (right-aligned)
         if language == "Urdu":
             paragraph.alignment = 2
             rtl = OxmlElement('w:rtl')
@@ -89,7 +93,7 @@ Include only these sections:
 
         doc.add_paragraph("\nDeveloped by Najaf Ali Sharqi")
 
-        # Save to temporary file
+        # Save file
         filename = f"lesson_plan_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.docx"
         filepath = f"/tmp/{filename}"
         doc.save(filepath)
@@ -136,4 +140,3 @@ with gr.Blocks() as demo:
     gr.Markdown("🔹 Developed by **Najaf Ali Sharqi**")
 
 demo.launch()
-