Update app.py

app.py

@@ -200,7 +200,8 @@ class BasicAgent:
                 messages=[
                     {"role": "user", "content": prompt},
                 ],
-                temperature=0.3,
+                temperature=0,
+                top_p=0.1,
                 max_tokens=max_tokens,
             )
             return resp.choices[0].message.content.strip()
@@ -208,6 +209,20 @@ class BasicAgent:
             print(f"\nLLM Error: {str(e)}")
             raise
 
+    def _safe_parse(self, raw: str) -> str:
+        try:
+            return json.loads(raw)["ANSWER"]
+        except Exception:
+            # grab the first {...} in the text
+            match = re.search(r'\{.*?\}', raw, re.S)
+            if match:
+                try:
+                    return json.loads(match.group())["ANSWER"]
+                except Exception:
+                    pass
+            # as a last resort, strip everything before the first colon
+            return raw.split(':', 1)[-1].strip()
+
     def _generate_answer(self, state: AgentState) -> AgentState:
         prompt = f"""
 Answer this question using the materials provided.
@@ -220,8 +235,7 @@ Return ONLY this exact JSON object:
 """
         try:
             raw = self._call_llm(prompt, 300)
-            data = json.loads(raw)
-            answer = data["ANSWER"]
+            answer = self._safe_parse(raw)
             state["final_answer"] = answer
         except Exception as e:
             print(f"\nLLM Error in answer generation: {str(e)}")