Update app.py

app.py

@@ -1,17 +1,19 @@
 import gradio as gr
-import os
 from huggingface_hub import InferenceClient
 from datasets import load_dataset
 import random
 import re
 
-# Load math datasets for sample problems
-print("🔄 Loading datasets...")
-fw = load_dataset("HuggingFaceFW/fineweb-edu", name="sample-10BT", split="train", streaming=True)
-ds = load_dataset("HuggingFaceH4/ultrachat_200k", streaming=True)
+# Global datasets - load lazily
+math_samples = None
 
-def load_sample_problems():
-    """Load sample problems from math datasets"""
+def load_datasets_lazy():
+    """Load datasets only when needed - prevents startup crashes"""
+    global math_samples
+    if math_samples is not None:
+        return math_samples
+    
+    print("🔄 Loading datasets...")
     try:
         gsm8k = load_dataset("openai/gsm8k", "main", streaming=True)
         samples = []
@@ -20,21 +22,21 @@ def load_sample_problems():
             if i >= 50:
                 break
         print(f"✅ Loaded {len(samples)} GSM8K samples")
+        math_samples = samples
         return samples
     except Exception as e:
         print(f"⚠️ Dataset error: {e}, using fallback")
-        return [
+        math_samples = [
             "What is the derivative of f(x) = 3x² + 2x - 1?",
             "A triangle has sides of length 5, 12, and 13. What is its area?",
             "If log₂(x) + log₂(x+6) = 4, find the value of x.",
             "Find the limit: lim(x->0) (sin(x)/x)",
             "Solve the system: x + 2y = 7, 3x - y = 4"
         ]
-
-math_samples = load_sample_problems()
+        return math_samples
 
 def create_math_system_message():
-    """Create specialized system prompt for mathematics with LaTeX"""
+    """Specialized system prompt for mathematics with LaTeX"""
     return """You are Mathetics AI, an advanced mathematics tutor and problem solver.
 
 🧮 **Your Expertise:**
@@ -144,7 +146,10 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         yield error_msg
 
 def get_random_sample():
-    """Get a random sample problem"""
+    """Get a random sample problem - loads datasets if needed"""
+    global math_samples
+    if math_samples is None:
+        math_samples = load_datasets_lazy()
     if math_samples:
         return random.choice(math_samples)
     return "Solve for x: 2x² + 5x - 3 = 0"