Update app.py

app.py

@@ -388,13 +388,14 @@ QUESTION:
 MATERIALS:
 {search_block}
 
-Provide a direct and concise answer based on the materials above.
+Write ANSWER: <answer> on its own line.
 """
         try:
-            answer = self._call_llm(prompt, 300).strip()
+            raw = self._call_llm(prompt, 300)
+            answer = raw.split("ANSWER:")[-1].strip()
             
             # If first attempt fails or is empty, try a more direct prompt
-            if not answer or any(k in answer.lower() for k in ["cannot", "sorry", "don't know"]):
+            if not answer or "ANSWER:" not in raw or any(k in answer.lower() for k in ["cannot", "sorry", "don't know"]):
                 print("\nFirst attempt failed, trying direct prompt...")
                 direct_prompt = f"""
 Answer this question directly and concisely. Use the materials provided.
@@ -406,11 +407,13 @@ MATERIALS:
 {search_block}
 
 If you cannot find an exact answer, provide the most relevant information from the materials.
+Write ANSWER: <answer> on its own line.
 """
-                answer = self._call_llm(direct_prompt, 300).strip()
+                raw = self._call_llm(direct_prompt, 300)
+                answer = raw.split("ANSWER:")[-1].strip()
             
             # Final validation and fallback
-            if not answer:
+            if not answer or "ANSWER:" not in raw:
                 print("\nBoth attempts failed, using fallback answer...")
                 if materials:
                     # If we have materials but no answer, summarize what we know
@@ -418,8 +421,11 @@ If you cannot find an exact answer, provide the most relevant information from t
 Summarize the key information from these materials in one sentence:
 
 {search_block}
+
+Write ANSWER: <answer> on its own line.
 """
-                    answer = self._call_llm(summary_prompt, 150).strip()
+                    raw = self._call_llm(summary_prompt, 150)
+                    answer = raw.split("ANSWER:")[-1].strip()
                 else:
                     answer = "I cannot provide a definitive answer at this time."