Update app.py

app.py

@@ -3,30 +3,45 @@ import gradio as gr
 from docx import Document
 import datetime
 from groq import Groq
+from huggingface_hub import InferenceClient
 
-# ✅ Get Groq API key from environment variable
+# ✅ Initialize Groq client for English
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+client_groq = Groq(api_key=GROQ_API_KEY)
 
-client = Groq(api_key=GROQ_API_KEY)
+# ✅ Initialize Falcon model client for Urdu
+falcon_client = InferenceClient(model="mlx-community/Falcon-H1-7B-Instruct-6bit")
 
 # ✅ Function to generate lesson plan
 def generate_lesson_plan(class_, subject, topic, students, duration, date, teacher, language):
-    if language == "Urdu":
-        prompt = f"""
+    try:
+        if language == "Urdu":
+            # Urdu prompt for Falcon model
+            prompt = f"""
 آپ ایک تجربہ کار استاد ہیں۔
-موضوع "{topic}" کے لیے BOPPPS ماڈل کے تحت ایک تفصیلی سبق کا منصوبہ تیار کریں۔
-یہ منصوبہ مکمل طور پر اردو زبان میں ہو۔
-مندرجہ ذیل حصے ضرور شامل کریں:
-1. Bridge-In: دلچسپ آغاز کے لیے ایک مختصر پیراگراف۔
+موضوع "{topic}" کے لیے BOPPPS ماڈل کے تحت ایک مکمل سبق کا منصوبہ تیار کریں۔
+یہ منصوبہ مکمل طور پر قدرتی اور درست اردو زبان میں ہو۔
+مندرجہ ذیل حصے شامل کریں:
+1. Bridge-In: دلچسپ آغاز کے لیے ایک پیراگراف۔
 2. تین واضح تعلیمی مقاصد۔
 3. دو پیشگی تشخیصی سوالات۔
 4. ایک تفصیلی تدریسی سرگرمی (Participatory Learning Activity)۔
 5. دو سے تین بعد از تدریس سوالات۔
 6. استاد کی طرف سے ایک خلاصہ پیراگراف۔
-نوٹ: بنیادی معلومات (کلاس، مضمون، موضوع، وغیرہ) کو دوبارہ سرخیوں یا متن میں شامل نہ کریں۔
+نوٹ: بنیادی معلومات (کلاس، مضمون، موضوع وغیرہ) دوبارہ متن میں شامل نہ کریں۔
 """
-    else:
-        prompt = f"""
+            # ✅ Use Falcon model for Urdu
+            response = falcon_client.text_generation(
+                prompt,
+                max_new_tokens=800,
+                temperature=0.7,
+                do_sample=True
+            )
+            lesson_text = response.strip()
+
+        else:
+            # English prompt for Groq model
+            prompt = f"""
 You are a lesson plan expert.
 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.
@@ -38,16 +53,13 @@ Include the following structured components only:
 5. 2–3 Post-Assessment Questions.
 6. Summary paragraph by the teacher.
 """
-
-    try:
-        # ✅ Generate response from Groq model
-        response = 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()
+            # ✅ Use Groq model for English
+            response = client_groq.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()
 
         # ✅ Create Word document
         doc = Document()
@@ -75,10 +87,7 @@ Include the following structured components only:
 
         doc.add_paragraph("\n")
         doc.add_paragraph(lesson_text)
-
-        # ✅ Add footer credit
-        doc.add_paragraph("\n")
-        doc.add_paragraph("Developed by Najaf Ali Sharqi")
+        doc.add_paragraph("\nDeveloped by Najaf Ali Sharqi")
 
         filename = f"lesson_plan_{language.lower()}_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.docx"
         filepath = f"/tmp/{filename}"
@@ -87,7 +96,6 @@ Include the following structured components only:
         return filepath
 
     except Exception as e:
-        # ✅ Return error document instead of crashing
         error_doc = Document()
         error_doc.add_heading("Error in Lesson Plan Generator", 0)
         error_doc.add_paragraph(str(e))
@@ -99,7 +107,7 @@ Include the following structured components only:
 
 # ✅ Build Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 📘 BOPPPS Lesson Plan Generator (English + Urdu)")
+    gr.Markdown("## 📘 BOPPPS Lesson Plan Generator (English + Urdu Improved)")
 
     with gr.Group():
         with gr.Row():
@@ -109,9 +117,9 @@ with gr.Blocks() as demo:
             topic = gr.Textbox(label="Topic / موضوع", placeholder="e.g., Newton’s First Law / نیوٹن کا پہلا قانون")
             students = gr.Textbox(label="Number of Students / طلبہ کی تعداد", placeholder="e.g., 30")
         with gr.Row():
-            duration = gr.Textbox(label="Duration (in minutes) / دورانیہ", placeholder="e.g., 40")
-            date = gr.Textbox(label="Date / تاریخ", placeholder="e.g., 2025-05-25")
-        teacher = gr.Textbox(label="Teacher Name / استاد کا نام", placeholder="e.g., Najaf Ali Sharqi")
+            duration = gr.Textbox(label="Duration / دورانیہ", placeholder="e.g., 40")
+            date = gr.Textbox(label="Date / تاریخ", placeholder="e.g., 2025-10-05")
+        teacher = gr.Textbox(label="Teacher / استاد کا نام", placeholder="e.g., Najaf Ali Sharqi")
 
         language = gr.Radio(["English", "Urdu"], label="Select Language / زبان منتخب کریں", value="English")
 
@@ -128,3 +136,4 @@ with gr.Blocks() as demo:
     gr.Markdown("🔹 Developed by **Najaf Ali Sharqi**")
 
 demo.launch()
+