Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import json
+import ast
 import streamlit as st
 import torch
 import torch.nn.functional as F
@@ -8,11 +9,17 @@ import re
 import math
 import logging
 
-# --- Load heuristic weights from environment secrets ---
+# --- Load heuristic weights from environment secrets, with JSON→Python fallback ---
 @st.cache_resource
 def load_heuristic_weights():
-    ai = json.loads(os.environ["AI_WEIGHTS_JSON"])
-    og = json.loads(os.environ["OG_WEIGHTS_JSON"])
+    def _load(env_key):
+        raw = os.environ[env_key]
+        try:
+            return json.loads(raw)
+        except json.JSONDecodeError:
+            return ast.literal_eval(raw)
+    ai = _load("AI_WEIGHTS_JSON")
+    og = _load("OG_WEIGHTS_JSON")
     return ai, og
 
 AI_WEIGHTS, OG_WEIGHTS = load_heuristic_weights()