Update app.py

app.py

@@ -2,34 +2,46 @@ import os
 import gradio as gr
 from docx import Document
 import datetime
+from groq import Groq
 from huggingface_hub import InferenceClient
 
-# ✅ Load your Hugging Face token from environment (set it in HF Spaces or locally)
+# ✅ Load API Keys from environment variables (set in Hugging Face Secrets)
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# ✅ Initialize Hugging Face client
-client = InferenceClient(
-    model="tiiuae/falcon-7b-instruct", 
-    token=HF_TOKEN
-)
+# ✅ Initialize clients
+groq_client = Groq(api_key=GROQ_API_KEY)
+hf_client = InferenceClient(model="tiiuae/falcon-7b-instruct", token=HF_TOKEN)
 
+
+# ✅ Core Function
 def generate_lesson_plan(language, class_, subject, topic, students, duration, date, teacher):
-    if language == "Urdu":
-        prompt = f"""
+    try:
+        if language == "Urdu":
+            prompt = f"""
 آپ ایک تجربہ کار استاد ہیں۔
 براہ کرم درج ذیل موضوع پر ایک مکمل BOPPPS لیسن پلان اردو زبان میں تیار کریں۔
 موضوع: "{topic}"
-منصوبہ میں درج ذیل حصے شامل ہوں:
+منصوبے میں درج ذیل حصے شامل ہوں:
 1. Bridge-In: ایک مختصر دلچسپ تمہیدی پیراگراف۔
 2. 3 واضح تعلیمی مقاصد۔
 3. 2 قبل از جائزہ (Pre-Assessment) سوالات۔
 4. ایک مفصل تدریسی سرگرمی (Participatory Learning)۔
 5. 2 تا 3 بعد از جائزہ (Post-Assessment) سوالات۔
 6. سبق کا خلاصہ (Summary)۔
-براہ کرم متن کو قدرتی اور تعلیمی انداز میں اردو میں لکھیں۔
-        """
-    else:
-        prompt = f"""
+براہ کرم مکمل متن قدرتی اور تعلیمی انداز میں اردو میں لکھیں۔
+            """
+
+            response = hf_client.text_generation(
+                prompt,
+                max_new_tokens=900,
+                temperature=0.7,
+            )
+
+            lesson_text = response.strip()
+
+        else:
+            prompt = f"""
 You are a professional 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.
@@ -40,23 +52,20 @@ Include only the following:
 4. One detailed Participatory Learning Activity.
 5. 2–3 Post-Assessment Questions.
 6. A short Summary paragraph.
-        """
+            """
 
-    try:
-        # ✅ Generate lesson text
-        response = client.text_generation(
-            prompt,
-            max_new_tokens=800,
-            temperature=0.7,
-        )
+            response = groq_client.chat.completions.create(
+                model="llama-3.1-8b-instant",
+                messages=[{"role": "user", "content": prompt}],
+                temperature=0.7,
+            )
 
-        lesson_text = response.strip()
+            lesson_text = response.choices[0].message.content.strip()
 
         # ✅ Create Word document
         doc = Document()
         doc.add_heading("BOPPPS Lesson Plan", 0)
 
-        # Metadata table
         table = doc.add_table(rows=0, cols=2)
         data = {
             "Language": language,
@@ -85,7 +94,6 @@ Include only the following:
         return filepath
 
     except Exception as e:
-        # Error handling
         error_doc = Document()
         error_doc.add_heading("Error in Lesson Plan Generator", 0)
         error_doc.add_paragraph(str(e))
@@ -127,4 +135,3 @@ with gr.Blocks() as demo:
 
 demo.launch()
 
-