frontend stuff elkay

utils/api.py

@@ -1,46 +1,53 @@
 # utils/api.py
 import os, requests
 
-BACKEND = (os.getenv("BACKEND_URL") or "").strip()
+BACKEND = (os.getenv("BACKEND_URL") or "").strip().rstrip("/")
 if not BACKEND:
     raise RuntimeError("BACKEND_URL is not set in Space secrets.")
-BACKEND = BACKEND.rstrip("/")
 
+TOKEN = (os.getenv("BACKEND_TOKEN") or "").strip()
 
-def start_agent(student_id:int, lesson_id:int, level_slug:str):
-    r = requests.post(f"{BACKEND}/agent/start", json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug})
-    r.raise_for_status()
-    return r.json()
+_session = requests.Session()
+_session.timeout = 30  # default timeout for all calls via this session
+if TOKEN:
+    _session.headers["Authorization"] = f"Bearer {TOKEN}"
+
+def _req(method: str, path: str, **kw):
+    url = f"{BACKEND}{path}"
+    r = _session.request(method, url, **kw)
+    try:
+        r.raise_for_status()
+    except requests.HTTPError as e:
+        # make errors readable in Streamlit
+        raise RuntimeError(f"{method} {path} failed [{r.status_code}]: {r.text[:500]}") from e
+    return r
 
 def health():
-    r = requests.get(f"{BACKEND}/", timeout=10)
-    r.raise_for_status()
-    return r.json()
+    return _req("GET", "/health").json()
+
+def start_agent(student_id:int, lesson_id:int, level_slug:str):
+    return _req("POST", "/agent/start",
+                json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug}).json()
 
 def get_quiz(student_id:int, lesson_id:int, level_slug:str):
-    r = requests.post(f"{BACKEND}/agent/quiz", json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug})
-    r.raise_for_status()
-    return r.json()["items"]
+    return _req("POST", "/agent/quiz",
+                json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug}).json()["items"]
 
 def grade_quiz(student_id:int, lesson_id:int, level_slug:str, answers:list[str], assignment_id:int|None=None):
-    r = requests.post(f"{BACKEND}/agent/grade", json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug,"answers":answers,"assignment_id":assignment_id})
-    r.raise_for_status()
-    d = r.json(); return d["score"], d["total"]
+    d = _req("POST", "/agent/grade",
+             json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug,
+                   "answers":answers,"assignment_id":assignment_id}).json()
+    return d["score"], d["total"]
 
 def next_step(student_id:int, lesson_id:int, level_slug:str, answers:list[str], assignment_id:int|None=None):
-    r = requests.post(f"{BACKEND}/agent/coach_or_celebrate", json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug,"answers":answers,"assignment_id":assignment_id})
-    r.raise_for_status()
-    return r.json()
+    return _req("POST", "/agent/coach_or_celebrate",
+                json={"student_id":student_id,"lesson_id":lesson_id,"level_slug":level_slug,
+                      "answers":answers,"assignment_id":assignment_id}).json()
 
 def login(email: str, password: str):
-    r = requests.post(f"{BACKEND}/auth/login", json={"email": email, "password": password})
-    r.raise_for_status()
-    return r.json()
+    return _req("POST", "/auth/login", json={"email": email, "password": password}).json()
 
 def signup_student(name: str, email: str, password: str, level_label: str, country_label: str):
-    r = requests.post(f"{BACKEND}/auth/signup/student", json={
-        "name": name, "email": email, "password": password,
-        "level_label": level_label, "country_label": country_label
-    })
-    r.raise_for_status()
-    return r.json()
+    return _req("POST", "/auth/signup/student",
+                json={"name": name, "email": email, "password": password,
+                      "level_label": level_label, "country_label": country_label}).json()