Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import os
 import gradio as gr
 from docx import Document
@@ -5,41 +6,35 @@ from docx.oxml.ns import qn
 from docx.oxml import OxmlElement
 import datetime
 from groq import Groq
-from huggingface_hub import InferenceClient
 
-# Load API keys (must set in Secrets / environment)
+# ✅ Load API key from Hugging Face Secrets
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-HF_TOKEN = os.getenv("HF_TOKEN")
 
-groq_client = Groq(api_key=GROQ_API_KEY)
-hf_client = InferenceClient(token=HF_TOKEN)
+# Initialize Groq client
+client = Groq(api_key=GROQ_API_KEY)
 
+
+# ✅ Text Generation Function
 def generate_lesson_plan(language, class_, subject, topic, students, duration, date, teacher):
     try:
         if language == "English":
             prompt = f"""
-You are a lesson plan expert.
+You are an expert lesson plan creator.
 Create a detailed English lesson plan using the BOPPPS model for the topic: "{topic}".
-Do not repeat the basic info (class, subject, topic, teacher etc.) again in headings or body.
-Include the following structured components only:
-1. Bridge-In: a short, engaging paragraph.
+Avoid repeating 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. One detailed Participatory Learning Activity.
+4. 1 Participatory Learning Activity (detailed).
 5. 2–3 Post-Assessment Questions.
 6. Summary paragraph by the teacher.
             """
-            response = groq_client.chat.completions.create(
-                model="llama-3.1-8b-instant",
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.7,
-            )
-            lesson_text = response.choices[0].message.content.strip()
 
         else:  # Urdu mode
             prompt = f"""
 آپ ایک ماہر اسباقی منصوبہ ساز ہیں۔
-ذیل کے موضوع "{topic}" پر BOPPPS ماڈل کے مطابق ایک مفصل اردو سبق منصوبہ تیار کریں۔
+موضوع "{topic}" پر BOPPPS ماڈل کے مطابق ایک مفصل اردو سبق منصوبہ تیار کریں۔
 براہ کرم درج ذیل حصے لازمی شامل کریں:
 1. تعارفی سرگرمی (Bridge-In)
 2. 3 واضح تعلیمی مقاصد
@@ -48,18 +43,21 @@ Include the following structured components only:
 5. 2 تا 3 بعد از تدریس سوالات (Post-assessment)
 6. سبق کا خلاصہ (Summary)
             """
-            urdu_response = hf_client.text_generation(
-                model="Imran1/gpt2-urdu-news",
-                prompt=prompt,
-                max_new_tokens=500,
-                temperature=0.8,
-            )
-            if isinstance(urdu_response, str):
-                lesson_text = urdu_response.strip()
-            else:
-                lesson_text = urdu_response.get("generated_text", "").strip()
-
-        # Create the Word document
+
+        # ✅ Use the Groq client with the openai/gpt-oss-120b model
+        response = client.chat.completions.create(
+            model="openai/gpt-oss-120b",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.8,
+            max_completion_tokens=2048,
+            top_p=1,
+            stream=False
+        )
+
+        # Extract generated text
+        lesson_text = response.choices[0].message.content.strip()
+
+        # ✅ Create the Word document
         doc = Document()
         doc.add_heading("BOPPPS Lesson Plan", 0)
 
@@ -79,15 +77,19 @@ Include the following structured components only:
             row_cells[1].text = str(value)
 
         doc.add_paragraph("\n")
+
         paragraph = doc.add_paragraph(lesson_text)
+
+        # Urdu formatting (right-aligned text)
         if language == "Urdu":
-            paragraph.alignment = 2  # right align
+            paragraph.alignment = 2
             rtl = OxmlElement('w:rtl')
             rtl.set(qn('w:val'), '1')
             paragraph._element.insert(0, rtl)
 
         doc.add_paragraph("\nDeveloped by Najaf Ali Sharqi")
 
+        # Save to temporary file
         filename = f"lesson_plan_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.docx"
         filepath = f"/tmp/{filename}"
         doc.save(filepath)
@@ -104,7 +106,7 @@ Include the following structured components only:
         return filepath
 
 
-# Gradio UI
+# ✅ Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("## 🌍 Bilingual BOPPPS Lesson Plan Generator (English + Urdu)")
 
@@ -118,7 +120,7 @@ with gr.Blocks() as demo:
             students = gr.Textbox(label="Number of Students", placeholder="e.g., 30")
         with gr.Row():
             duration = gr.Textbox(label="Duration (minutes)", placeholder="e.g., 40")
-            date = gr.Textbox(label="Date", placeholder="e.g., 2025-10-05")
+            date = gr.Textbox(label="Date", placeholder="e.g., 2025-10-07")
         teacher = gr.Textbox(label="Teacher Name", placeholder="e.g., Najaf Ali Sharqi")
 
         generate_button = gr.Button("Generate Lesson Plan")
@@ -134,3 +136,4 @@ with gr.Blocks() as demo:
     gr.Markdown("🔹 Developed by **Najaf Ali Sharqi**")
 
 demo.launch()
+