Update app.py

app.py

@@ -3,9 +3,12 @@ import gradio as gr
 import torch
 from transformers import pipeline
 
-@GPU  # 👈 REQUIRED to activate GPU on Hugging Face ZeroGPU
+@GPU  # 👈 REQUIRED for Hugging Face ZeroGPU Spaces
 def main():
+    # ✅ Select GPU if available
     device = 0 if torch.cuda.is_available() else -1
+
+    # ✅ Load model pipeline
     pipe = pipeline(
         "text-generation",
         model="Pisethan/khmer-lesson-model-v2",
@@ -13,14 +16,14 @@ def main():
         device=device
     )
 
-# ✅ Dropdown choices
-grade_options = ["1", "2", "3", "4", "5", "6"]
-topic_options = ["Addition", "Subtraction", "Counting", "Number Recognition", "Multiplication", "Division"]
-level_options = ["Beginner", "Intermediate", "Advanced"]
+    # ✅ Dropdown values
+    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 with the following structure:
+    # ✅ Generate one lesson
+    def generate_lesson(grade, topic, level):
+        prompt = f"""Generate a Khmer math lesson plan with the following structure:
 
 Lesson Title:
 Objective:
@@ -32,47 +35,48 @@ Grade: {grade}
 Topic: {topic}
 TaRL Level: {level}"""
 
-    output = pipe(prompt, max_new_tokens=300, temperature=0.7, do_sample=True)
-    return output[0]['generated_text']
+        output = pipe(prompt, max_new_tokens=300, temperature=0.7, do_sample=True)
+        return output[0]['generated_text']
 
-# ✅ Function: 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.
+    # ✅ 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
+                    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
-with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា")
-    gr.Markdown("ជ្រើសរើសថ្នាក់ ប្រធានបទ និងកម្រិតសិស្ស រួចចុចបង្កើតមេរៀន។ ឬចុចប៊ូតុងខាងក្រោមសម្រាប់បង្កើតមេរៀនទាំងអស់។")
+    # ✅ Gradio UI
+    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")
+        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)
+        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("🧹 សម្អាត")
+        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)
+        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)
 
-# ✅ Launch app with public link
-demo.launch(share=True)
+    # ✅ Run with public URL
+    demo.launch(share=True)
 
-main()
+# ✅ Call main
+main()