Update app.py

app.py

@@ -1,6 +1,28 @@
 # Multi-Student, Multi-Teacher Self-Improving LLM Training System with Gradio Dashboard
 # Requires: transformers, datasets, matplotlib, torch, gradio, sqlalchemy, scikit-learn
 
+import subprocess
+import sys
+
+def ensure_package(package_name, install_name=None):
+    try:
+        __import__(package_name)
+    except ImportError:
+        print(f"[!] Missing: {package_name} — attempting install...")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", install_name or package_name])
+
+# Ensure required packages are installed
+for pkg in [
+    ("torch", "torch"),
+    ("transformers", "transformers"),
+    ("datasets", "datasets"),
+    ("gradio", "gradio[oauth,mcp]==5.34.0"),
+    ("matplotlib", "matplotlib"),
+    ("sklearn", "scikit-learn"),
+    ("sqlite3", None),  # built-in, skip install
+]:
+    ensure_package(pkg[0], pkg[1])
+
 import os
 import random
 import torch