elkay

phase/Student_view/games/MoneyMatch.py

@@ -8,6 +8,11 @@ import streamlit as st
 from utils import db as dbapi
 import time
 from utils import db as db_util
+from utils import api   
+
+
+USE_LOCAL_DB = os.getenv("DISABLE_DB", "1") != "1"
+
 
 # ---------- paths ----------
 PROJECT_ROOT = Path(__file__).resolve().parents[3]
@@ -51,24 +56,35 @@ def _award_xp(gained: int):
         ss.mm_level += 1
 
 def _persist_success(gained_xp: int):
-    user = st.session_state.get("user")
-    if not user or "user_id" not in user:
+    user = st.session_state.get("user") or {}
+    user_id = int(user.get("user_id", 0))
+    if not user_id:
         st.error("Not saving. No logged-in user_id in session.")
         return
-    elapsed_ms = int((time.perf_counter() - st.session_state.mm_start_ts) * 1000)
+
+    payload = dict(
+        user_id=user_id,
+        target=int(st.session_state.mm_target),
+        total=int(st.session_state.mm_total),
+        elapsed_ms=int((time.perf_counter() - st.session_state.mm_start_ts) * 1000),
+        matched=True,
+        gained_xp=int(gained_xp),
+    )
+
     try:
-        dbapi.record_money_match_play(
-            user["user_id"],
-            target=int(st.session_state.mm_target),
-            total=int(st.session_state.mm_total),
-            elapsed_ms=elapsed_ms,
-            matched=True,
-            gained_xp=int(gained_xp),
-        )
-        st.toast(f"Saved to TiDB  +{gained_xp} XP")
+        if USE_LOCAL_DB and hasattr(dbapi, "record_money_match_play"):
+            # direct DB mode
+            dbapi.record_money_match_play(**payload)
+            st.toast(f"Saved to DB  +{gained_xp} XP")
+        else:
+            # backend mode (DISABLE_DB=1)
+            api.record_money_match_play(**payload)
+            st.toast(f"Saved via backend  +{gained_xp} XP")
+        st.session_state.mm_saved = True
     except Exception as e:
         st.error(f"Save failed: {e}")
 
+
 # --- CSS injection (run every render) ---
 def _inject_css():
     css_path = PROJECT_ROOT / "assets" / "styles.css"