Update app.py

app.py

@@ -388,15 +388,15 @@ QUESTION:
 MATERIALS:
 {search_block}
 
-Return ONLY a JSON object in this format:
-{{"ANSWER": "your answer here"}}
+Return ONLY this exact JSON object:
+{{"ANSWER": "<answer text>"}}
 """
         try:
             raw = self._call_llm(prompt, 300)
             try:
-                result = json.loads(raw)
-                answer = result.get("ANSWER", "")
-            except json.JSONDecodeError:
+                data = json.loads(raw)
+                answer = data["ANSWER"]
+            except (json.JSONDecodeError, KeyError):
                 print("\nJSON parse error, trying direct prompt...")
                 # If first attempt fails, try a more direct prompt
                 direct_prompt = f"""
@@ -408,30 +408,31 @@ QUESTION:
 MATERIALS:
 {search_block}
 
-Return ONLY a JSON object in this format:
-{{"ANSWER": "your answer here"}}
+Return ONLY this exact JSON object:
+{{"ANSWER": "<answer text>"}}
 """
                 raw = self._call_llm(direct_prompt, 300)
                 try:
-                    result = json.loads(raw)
-                    answer = result.get("ANSWER", "")
-                except json.JSONDecodeError:
+                    data = json.loads(raw)
+                    answer = data["ANSWER"]
+                except (json.JSONDecodeError, KeyError):
                     print("\nBoth attempts failed, using fallback answer...")
                     if materials:
                         # If we have materials but no answer, summarize what we know
                         summary_prompt = f"""
 Summarize the key information from these materials in one sentence.
-Return ONLY a JSON object in this format:
-{{"ANSWER": "your summary here"}}
 
 MATERIALS:
 {search_block}
+
+Return ONLY this exact JSON object:
+{{"ANSWER": "<answer text>"}}
 """
                         raw = self._call_llm(summary_prompt, 150)
                         try:
-                            result = json.loads(raw)
-                            answer = result.get("ANSWER", "")
-                        except json.JSONDecodeError:
+                            data = json.loads(raw)
+                            answer = data["ANSWER"]
+                        except (json.JSONDecodeError, KeyError):
                             answer = "I cannot provide a definitive answer at this time."
                     else:
                         answer = "I cannot provide a definitive answer at this time."