Update app.py

app.py

@@ -276,8 +276,9 @@ def run_vlm_and_get_features(face_path: str, eye_path: str, prompt: Optional[str
        (parsed_features_dict_or_None, raw_text_response_str)
 
     We attempt to parse JSON as before, but always return the original raw text so it can be
-    forwarded verbatim to the LLM if desired. This function now logs the raw VLM output and
-    the parsed features for debugging.
+    forwarded verbatim to the LLM if desired. This function now also tries the robust
+    regex extractor (extract_json_via_regex) on the raw text if json.loads fails, and logs
+    the extracted values.
     """
     prompt = prompt or DEFAULT_VLM_PROMPT
     if not os.path.exists(face_path) or not os.path.exists(eye_path):
@@ -319,31 +320,29 @@ def run_vlm_and_get_features(face_path: str, eye_path: str, prompt: Optional[str
     except Exception:
         logger.info("VLM raw output (could not pretty print)")
 
-    # Try to parse JSON but remember raw text always
+    # Try to parse JSON first (as before)
     parsed_features = None
     try:
         parsed_features = json.loads(text_out)
         if not isinstance(parsed_features, dict):
             parsed_features = None
     except Exception:
+        parsed_features = None
+
+    # If json.loads failed, try regex-based extraction (robust)
+    if parsed_features is None:
         try:
-            s = text_out
-            first = s.find("{")
-            last = s.rfind("}")
-            if first != -1 and last != -1 and last > first:
-                maybe = s[first:last+1]
-                parsed_features = json.loads(maybe)
-                if not isinstance(parsed_features, dict):
-                    parsed_features = None
-            else:
-                parsed_features = None
-        except Exception:
+            parsed_features = extract_json_via_regex(text_out)
+            logger.info("VLM regex-extracted features:\n%s", json.dumps(parsed_features, indent=2, ensure_ascii=False))
+        except Exception as e:
+            # No JSON-like block or extraction failed: keep parsed_features as None and log
+            logger.info("VLM regex extraction did not find structured JSON (this may be fine): %s", str(e))
             parsed_features = None
 
     # Log parsed features if available
     if parsed_features is not None:
         try:
-            logger.info("VLM parsed features:\n%s", json.dumps(parsed_features, indent=2, ensure_ascii=False))
+            logger.info("VLM parsed features (final):\n%s", json.dumps(parsed_features, indent=2, ensure_ascii=False))
         except Exception:
             logger.info("VLM parsed features (raw): %s", str(parsed_features))
     else: