Update app.py

app.py

@@ -1,116 +1,93 @@
 import os
-import requests
 import gradio as gr
 from docx import Document
-from docx.shared import Pt
 import datetime
+from groq import Groq
 
-# Groq API Key (replace with your own if not using environment variable)
-GROQ_API_KEY = os.getenv("GROQ_API_KEY", "your_groq_api_key_here")  # replace this before deploying
-GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
+# Load Groq API key
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Set this in Hugging Face secrets
 
-def generate_lesson_plan(class_level, subject, topic, students, duration, date, teacher):
+client = Groq(api_key=GROQ_API_KEY)
+
+def generate_lesson_plan(class_, subject, topic, students, duration, date, teacher):
+    # Construct the prompt — instruct model NOT to repeat inputs
     prompt = f"""
-Create a complete lesson plan using the BOPPPS model.
-
-Details:
-- Class: {class_level}
-- Subject: {subject}
-- Topic: {topic}
-- Number of students: {students}
-- Duration: {duration} minutes
-- Date: {date}
-- Teacher Name: {teacher}
-
-Include the following in proper sequence:
-1. **Bridge-In**: Introduction to grab students' attention related to the topic.
-2. **Objectives**: Three specific learning objectives.
-3. **Pre-Assessment**: Two questions to assess prior knowledge related to the topic.
-4. **Participatory Learning**: Detailed instruction for interactive/participatory learning.
-5. **Post-Assessment**: Questions or activities to evaluate understanding.
-6. **Summary**: Wrap-up by the teacher, summarizing key points of the lesson.
-
-Output must be in proper English and suitable for inclusion in a Word document.
-"""
+You are an expert lesson plan developer.
+
+Write ONLY the **lesson body** using the BOPPPS model for the topic: "{topic}".
 
-    headers = {
-        "Authorization": f"Bearer {GROQ_API_KEY}",
-        "Content-Type": "application/json"
-    }
+Do NOT restate these inputs (class, subject, topic, date, teacher name, etc.) — only write the lesson content.
 
-    body = {
-        "model": "llama3-8b-8192",
-        "messages": [{"role": "user", "content": prompt}],
-        "temperature": 0.7
-    }
+The plan must be in English and include the following:
+
+1. **Bridge-In**: A short paragraph that introduces the topic engagingly.
+2. **Learning Objectives**: List 3 clear objectives.
+3. **Pre-Assessment Questions**: Write 2 questions to assess prior knowledge.
+4. **Participatory Learning**: Describe an engaging activity.
+5. **Post-Assessment**: 2–3 questions to evaluate student understanding.
+6. **Summary**: A concluding paragraph by the teacher.
+"""
 
     try:
-        response = requests.post(GROQ_API_URL, headers=headers, json=body)
-        response.raise_for_status()
-        result = response.json()
-        lesson_plan = result["choices"][0]["message"]["content"]
+        response = client.chat.completions.create(
+            model="llama3-8b-8192",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.7,
+        )
+        lesson_text = response.choices[0].message.content
 
-        # Create Word document
         doc = Document()
-        doc.add_heading("Lesson Plan (BOPPPS Model)", 0)
-
-        # Add table at the top
-        table = doc.add_table(rows=6, cols=2)
-        table.style = 'Table Grid'
-
-        entries = [
-            ("Class Level", class_level),
-            ("Subject", subject),
-            ("Topic", topic),
-            ("Number of Students", str(students)),
-            ("Duration", f"{duration} minutes"),
-            ("Date & Teacher", f"{date} | {teacher}")
-        ]
-
-        for i, (label, value) in enumerate(entries):
-            table.cell(i, 0).text = label
-            table.cell(i, 1).text = value
-
-            # Optional styling
-            for cell in table.row_cells(i):
-                for paragraph in cell.paragraphs:
-                    for run in paragraph.runs:
-                        run.font.size = Pt(11)
-
-        doc.add_paragraph("\n")  # spacing
-        doc.add_paragraph(lesson_plan)
-
-        filename = f"LessonPlan_{topic.replace(' ', '_')}_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
-        filepath = os.path.join("/tmp", filename)
+        doc.add_heading(f"BOPPPS Lesson Plan - {topic}", 0)
+
+        # Table for metadata (not repeated in content)
+        table = doc.add_table(rows=0, cols=2)
+        fields = {
+            "Class": class_,
+            "Subject": subject,
+            "Topic": topic,
+            "No. of Students": students,
+            "Duration": duration,
+            "Date": date,
+            "Teacher Name": teacher
+        }
+        for key, value in fields.items():
+            row_cells = table.add_row().cells
+            row_cells[0].text = key
+            row_cells[1].text = value
+
+        doc.add_paragraph("\n")  # space after table
+        doc.add_paragraph(lesson_text)
+
+        filename = f"lesson_plan_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.docx"
+        filepath = f"/tmp/{filename}"
         doc.save(filepath)
-
         return filepath
-
-    except requests.exceptions.RequestException as e:
-        return f"API request failed: {str(e)}"
-    except Exception as ex:
-        return f"Error generating lesson plan: {str(ex)}"
+    except Exception as e:
+        return f"Error: {e}"
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 📘 BOPPPS Lesson Plan Generator\nCreate complete lesson plans powered by Groq + LLaMA3.")
+    gr.Markdown("## 📘 BOPPPS Lesson Plan Generator (English Only)")
 
     with gr.Row():
-        class_level = gr.Textbox(label="Class Level (e.g. Grade 6)")
-        subject = gr.Textbox(label="Subject (e.g. Science)")
-        topic = gr.Textbox(label="Topic (e.g. Water Cycle)")
-        students = gr.Number(label="Number of Students", precision=0)
-        duration = gr.Number(label="Duration (minutes)", precision=0)
-        date = gr.Textbox(label="Date (e.g. 2025-05-25)")
+        class_ = gr.Textbox(label="Class")
+        subject = gr.Textbox(label="Subject")
+        topic = gr.Textbox(label="Topic")
+
+    with gr.Row():
+        students = gr.Textbox(label="Number of Students")
+        duration = gr.Textbox(label="Duration (in minutes)")
+        date = gr.Textbox(label="Date")
         teacher = gr.Textbox(label="Teacher Name")
 
-    generate_button = gr.Button("Generate Lesson Plan")
-    output_file = gr.File(label="⬇️ Download Lesson Plan (Word)")
+    generate_btn = gr.Button("Generate Lesson Plan")
+    output_file = gr.File(label="Download Lesson Plan")
 
-    generate_button.click(
+    generate_btn.click(
         fn=generate_lesson_plan,
-        inputs=[class_level, subject, topic, students, duration, date, teacher],
+        inputs=[class_, subject, topic, students, duration, date, teacher],
         outputs=output_file
     )
 
 demo.launch()
+