Update app.py

app.py

@@ -1,14 +1,19 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the hosted model and tokenizer from Hugging Face Hub
+# Load model from Hugging Face Hub
 pipe = pipeline(
     "text-generation",
     model="Pisethan/khmer-lesson-model",
     tokenizer="Pisethan/khmer-lesson-model"
 )
 
-# Generation function
+# Define dropdown options
+grade_options = ["1", "2", "3", "4", "5", "6"]
+topic_options = ["Addition", "Subtraction", "Counting", "Number Recognition", "Multiplication", "Division"]
+level_options = ["Beginner", "Intermediate", "Advanced"]
+
+# Function: single lesson
 def generate_lesson(grade, topic, level):
     prompt = f"""Generate a Khmer math lesson plan.
 
@@ -18,17 +23,41 @@ TaRL Level: {level}"""
     output = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)
     return output[0]['generated_text']
 
-# Gradio UI
-iface = gr.Interface(
-    fn=generate_lesson,
-    inputs=[
-        gr.Dropdown(["1", "2", "3", "4", "5", "6"], label="ថ្នាក់ (Grade)"),
-        gr.Dropdown(["Addition", "Subtraction", "Counting", "Number Recognition", "Multiplication", "Division"], label="ប្រធានបទ (Topic)"),
-        gr.Dropdown(["Beginner", "Intermediate", "Advanced"], label="កម្រិតសិស្ស (TaRL Level)")
-    ],
-    outputs=gr.Textbox(label="Khmer Lesson Plan"),
-    title="🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា",
-    description="ជ្រើសរើសថ្នាក់ ប្រធានបទ និងកម្រិតសិស្ស រួចចុចបង្កើតមេរៀន។ នេះជាម៉ូឌែលកែសម្រួលជាសកលសម្រាប់បង្រៀនគណិតជាភាសាខ្មែរ។"
-)
+# Function: auto generate all combinations
+def generate_all_lessons():
+    results = ""
+    for grade in grade_options:
+        for topic in topic_options:
+            for level in level_options:
+                prompt = f"""Generate a Khmer math lesson plan.
+
+Grade: {grade}
+Topic: {topic}
+TaRL Level: {level}"""
+                output = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)
+                lesson = output[0]["generated_text"]
+                results += f"🔹 ថ្នាក់ {grade} | {topic} | {level}\n{lesson}\n\n{'-'*50}\n\n"
+    return results
+
+# Gradio UI layout
+with gr.Blocks() as demo:
+    gr.Markdown("## 🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា")
+    gr.Markdown("ជ្រើសរើសថ្នាក់ ប្រធានបទ និងកម្រិតសិស្ស រួចចុចបង្កើតមេរៀន។ ឬចុចប៊ូតុងខាងក្រោមសម្រាប់បង្កើតមេរៀនទាំងអស់។")
+
+    with gr.Row():
+        grade = gr.Dropdown(choices=grade_options, label="ថ្នាក់ (Grade)", value="1")
+        topic = gr.Dropdown(choices=topic_options, label="ប្រធានបទ (Topic)", value="Addition")
+        level = gr.Dropdown(choices=level_options, label="កម្រិតសិស្ស (TaRL Level)", value="Beginner")
+
+    output_box = gr.Textbox(label="📘 Khmer Lesson Plan", lines=12)
+
+    with gr.Row():
+        generate_btn = gr.Button("✅ បង្កើតមេរៀន")
+        generate_all_btn = gr.Button("🧠 បង្កើតមេរៀនទាំងអស់")
+        clear_btn = gr.Button("🧹 សម្អាត")
+
+    generate_btn.click(fn=generate_lesson, inputs=[grade, topic, level], outputs=output_box)
+    generate_all_btn.click(fn=generate_all_lessons, outputs=output_box)
+    clear_btn.click(fn=lambda: "", outputs=output_box)
 
-iface.launch()
+demo.launch()