Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import requests
 import json
 import os
+import re
+
 
 HF_API_URL = "https://router.huggingface.co/novita/v3/openai/chat/completions"
 HF_TOKEN = os.environ["HF_PROJECT_TOKEN"]
@@ -60,8 +62,13 @@ def evaluate_prompt(user_prompt):
         if isinstance(result, dict) and "choices" in result:
             raw_text = result["choices"][0]["message"]["content"]
             try:
-                json_part = raw_text[raw_text.index("{"):]
-                return json.loads(json_part)
+                # This regex finds the first complete JSON block
+                json_match = re.search(r'\{[\s\S]*\}', raw_text)
+                if json_match:
+                    json_str = json_match.group()
+                    return json.loads(json_str)
+                else:
+                    return {"error": "No valid JSON object found in model output."}
             except (ValueError, json.JSONDecodeError) as e:
                 return {"error": f"Failed to parse model output: {str(e)}"}
         return result