Update app.py

app.py

@@ -1,13 +1,40 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer
 import torch
-import spaces  # required for ZeroGPU
+import spaces
+import datetime
+import json
+import os
+from datasets import Dataset
+from huggingface_hub import HfFolder
+
+# ✅ Authenticate Hugging Face account (save token only once)
+HfFolder.save_token("hf_your_token_here")  # ← Replace with your real HF token
 
 # 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"]
 
+# ✅ Save lesson to Hugging Face Datasets
+def save_to_hf_dataset(prompt, output, repo_id="Pisethan/khmer-lesson-history"):
+    timestamp = datetime.datetime.now().isoformat()
+    record = {
+        "timestamp": timestamp,
+        "prompt": prompt,
+        "lesson": output
+    }
+
+    # Save locally
+    os.makedirs("history", exist_ok=True)
+    local_path = f"history/lesson_{timestamp.replace(':', '-')}.json"
+    with open(local_path, "w", encoding="utf-8") as f:
+        json.dump(record, f, ensure_ascii=False, indent=2)
+
+    # Push to HF Dataset
+    dataset = Dataset.from_list([record])
+    dataset.push_to_hub(repo_id)
+
 @spaces.GPU
 def generate_lesson(grade, topic, level):
     device = 0 if torch.cuda.is_available() else -1
@@ -34,7 +61,6 @@ Grade: {grade}
 Topic: {topic}
 TaRL Level: {level}
 """
-
     output = pipe(
         prompt,
         max_new_tokens=300,
@@ -43,8 +69,10 @@ TaRL Level: {level}
         eos_token_id=tokenizer.eos_token_id
     )
 
-    return output[0]['generated_text']
+    # ✅ Save lesson
+    save_to_hf_dataset(prompt, output[0]["generated_text"])
 
+    return output[0]['generated_text']
 
 @spaces.GPU
 def generate_all_lessons():
@@ -84,9 +112,9 @@ TaRL Level: {level}
                     eos_token_id=tokenizer.eos_token_id
                 )
                 results += f"🔹 ថ្នាក់ {grade} | {topic} | {level}\n{output[0]['generated_text']}\n\n{'-'*50}\n\n"
+                save_to_hf_dataset(prompt, output[0]["generated_text"])  # ✅ Save each one
     return results
 
-
 # Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 អ្នកជំនួយបង្កើតមេរៀនគណិតវិទ្យា")
@@ -114,5 +142,5 @@ with gr.Blocks() as demo:
     gen_all_btn.click(fn=generate_all_lessons, outputs=output_box)
     clear_btn.click(fn=lambda: "", outputs=output_box)
 
-demo.queue()  # Required for ZeroGPU
+demo.queue()
 demo.launch()