Update app.py

app.py

@@ -82,6 +82,50 @@ SORENESS_SEVERITY_OPTIONS = ["none", "mild", "moderate", "severe"]
 
 COMPLETION_OPTIONS = ["full", "partial"]
 
+# ─────────────────────────────────────────────────────────────
+# NUMERICAL ENCODING MAPS
+# Must stay in sync with your training notebook label maps
+# ─────────────────────────────────────────────────────────────
+
+MOOD_MAP = {
+    "accomplished": 0,
+    "anxious":      1,
+    "distracted":   2,
+    "energized":    3,
+    "fatigued":     4,
+    "frustrated":   5,
+    "neutral":      6,
+    "positive":     7,
+}
+
+EXERTION_MAP = {
+    "low":      0,
+    "moderate": 1,
+    "high":     2,
+}
+
+SORENESS_REGION_MAP = {
+    "none":     0,
+    "back":     1,
+    "biceps":   2,
+    "chest":    3,
+    "legs":     4,
+    "shoulder": 5,
+    "triceps":  6,
+}
+
+SORENESS_SEVERITY_MAP = {
+    "none":     0,
+    "mild":     1,
+    "moderate": 2,
+    "severe":   3,
+}
+
+COMPLETION_MAP = {
+    "full":    1,
+    "partial": 0,
+}
+
 
 # ─────────────────────────────────────────────────────────────
 # SUBMISSION HANDLER
@@ -120,15 +164,15 @@ def submit_entry(
     if soreness_region != "none" and soreness_severity == "none":
         return "⚠️ Please select a severity level for your soreness region."
 
-    # ── Build row ────────────────────────────────────────────
+    # ── Encode labels to integers ─────────────────────────────
     row = [
         datetime.utcnow().strftime("%Y-%m-%d %H:%M:%S"),
         user_text.strip(),
-        mood,
-        exertion,
-        soreness_region,
-        soreness_severity,
-        completion_status,
+        MOOD_MAP[mood],
+        EXERTION_MAP[exertion],
+        SORENESS_REGION_MAP[soreness_region],
+        SORENESS_SEVERITY_MAP[soreness_severity],
+        COMPLETION_MAP[completion_status],
     ]
 
     # ── Write to Google Sheets ───────────────────────────────