changes v20

question.py

@@ -48,6 +48,14 @@ def is_relevant(article, keywords):
     text = f"{article.get('title', '')} {article.get('content', '')}".lower()
     return any(kw.lower() in text for kw in keywords)
 
+def extract_answer_after_label(text: str) -> str:
+    """
+    Extracts everything after the first 'Answer:' label.
+    Assumes 'Answer:' appears once and is followed by the relevant content.
+    """
+    if "Answer:" in text:
+        return text.split("Answer:", 1)[1].strip()
+    return text.strip()  
 
 def mistral_generate(prompt: str, max_new_tokens=128):
     payload = {
@@ -62,7 +70,6 @@ def mistral_generate(prompt: str, max_new_tokens=128):
         response.raise_for_status()
         result = response.json()
         if isinstance(result, list) and len(result) > 0:
-            print("Mistral Result", result);
             return result[0].get("generated_text", "").strip()
         else:
             return ""
@@ -133,13 +140,15 @@ async def ask_question(input: QuestionInput):
     )
     answer = mistral_generate(answer_prompt, max_new_tokens=256)
     if not answer:
-        answer = "Cannot answer – model did not return a valid response."
+        final_answer = "Cannot answer – model did not return a valid response."
+
+    final_answer = extract_answer_after_label(answer)
 
     print("Mistral Answer:", answer)
 
     return {
         "question": question,
-        "answer": answer.strip(),
+        "answer": final_answer.strip(),
         "sources": [
             {"title": a["title"], "url": a["url"]}
             for a in relevant_articles