Update app.py

app.py

@@ -1,29 +1,26 @@
-from spaces import GPU
 import gradio as gr
-import torch
 from transformers import pipeline
+import torch
 
-@GPU  # 👈 REQUIRED for Hugging Face ZeroGPU Spaces
-def main():
-    # ✅ Select GPU if available
-    device = 0 if torch.cuda.is_available() else -1
+# Dropdown values
+grade_options = ["1", "2", "3", "4", "5", "6"]
+topic_options = ["Addition", "Subtraction", "Counting", "Number Recognition", "Multiplication", "Division"]
+level_options = ["Beginner", "Intermediate", "Advanced"]
 
-    # ✅ Load model pipeline
+def load_pipeline():
+    device = 0 if torch.cuda.is_available() else -1
     pipe = pipeline(
         "text-generation",
         model="Pisethan/khmer-lesson-model-v2",
         tokenizer="Pisethan/khmer-lesson-model",
         device=device
     )
+    return pipe
 
-    # ✅ Dropdown values
-    grade_options = ["1", "2", "3", "4", "5", "6"]
-    topic_options = ["Addition", "Subtraction", "Counting", "Number Recognition", "Multiplication", "Division"]
-    level_options = ["Beginner", "Intermediate", "Advanced"]
-
-    # ✅ Generate one lesson
-    def generate_lesson(grade, topic, level):
-        prompt = f"""Generate a Khmer math lesson plan with the following structure:
+# Function: generate one
+def generate_lesson(grade, topic, level):
+    pipe = load_pipeline()
+    prompt = f"""Generate a Khmer math lesson plan with the following structure:
 
 Lesson Title:
 Objective:
@@ -35,48 +32,45 @@ 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']
 
-    # ✅ 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.
+# Function: generate all combos
+def generate_all_lessons():
+    pipe = load_pipeline()
+    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
-    with gr.Blocks() as demo:
-        gr.Markdown("## 🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា")
-        gr.Markdown("ជ្រើសរើសថ្នាក់ ប្រធានបទ និងកម្រិតសិស្ស រួចចុចបង្កើតមេរៀន។ ឬចុចប៊ូតុងខាងក្រោមសម្រាប់បង្កើតមេរៀនទាំងអស់។")
+                output = pipe(prompt, max_new_tokens=200, temperature=0.7, do_sample=True)
+                results += f"🔹 ថ្នាក់ {grade} | {topic} | {level}\n{output[0]['generated_text']}\n\n{'-'*50}\n\n"
+    return results
 
-        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")
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា")
+    gr.Markdown("ជ្រើសរើសថ្នាក់ ប្រធានបទ និងកម្រិតសិស្ស រួចចុចបង្កើតមេរៀន។ ឬចុចប៊ូតុងខាងក្រោមសម្រាប់បង្កើតមេរៀនទាំងអស់។")
 
-        output_box = gr.Textbox(label="📘 Khmer Lesson Plan", lines=12)
+    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():
-            generate_btn = gr.Button("✅ បង្កើតមេរៀន")
-            generate_all_btn = gr.Button("🧠 បង្កើតមេរៀនទាំងអស់")
-            clear_btn = gr.Button("🧹 សម្អាត")
+    output_box = gr.Textbox(label="📘 Khmer Lesson Plan", lines=12)
 
-        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)
+    with gr.Row():
+        gen_btn = gr.Button("✅ បង្កើតមេរៀន")
+        gen_all_btn = gr.Button("🧠 បង្កើតមេរៀនទាំងអស់")
+        clear_btn = gr.Button("🧹 សម្អាត")
 
-    # ✅ Run with public URL
-    demo.launch(share=True)
+    gen_btn.click(fn=generate_lesson, inputs=[grade, topic, level], outputs=output_box)
+    gen_all_btn.click(fn=generate_all_lessons, outputs=output_box)
+    clear_btn.click(fn=lambda: "", outputs=output_box)
 
-# ✅ Call main
-main()
+# ❌ Don't use share=True inside HF Space
+demo.launch()