Update app.py

app.py

@@ -69,13 +69,17 @@ class MathRetrievalQA(dspy.Program):
         print("Generating answer...")
         answer = qa_pipeline(prompt, max_new_tokens=100)[0]["generated_text"]
         print("Generated answer:", answer)
-        return dspy.Output(answer=answer, retrieved_context=context)
+        return {"answer": answer, "retrieved_context": context}
+
+       # return dspy.Output(answer=answer, retrieved_context=context)
 
 class WebFallbackQA(dspy.Program):
     def forward(self, question):
         print("Fallback to Tavily...")
         answer = web_search_tavily(question)
-        return dspy.Output(answer=answer, retrieved_context="Tavily")
+       # return dspy.Output(answer=answer, retrieved_context="Tavily")
+        return {"answer": answer, "retrieved_context": "Tavily"}
+
 
 class MathRouter(dspy.Program):
     def forward(self, question):
@@ -83,8 +87,8 @@ class MathRouter(dspy.Program):
         if not is_valid_math_question(question):
             return dspy.Output(answer="❌ Only math questions are accepted. Please rephrase.", retrieved_context="")
         result = MathRetrievalQA().forward(question)
-        return result if result.answer else WebFallbackQA().forward(question)
-
+        #return result if result.answer else WebFallbackQA().forward(question)
+        return result if result["answer"] else WebFallbackQA().forward(question)
 router = MathRouter()
 
 # === Feedback Storage ===
@@ -105,7 +109,9 @@ def ask_question(question):
     print("ask_question() called with:", question)
     result = router.forward(question)
     print("Result:", result)
-    return result.answer, question, result.answer
+    #return result.answer, question, result.answer
+    return result["answer"], question, result["answer"]
+
 
 def submit_feedback(question, model_answer, feedback, correct_answer):
     store_feedback(question, model_answer, feedback, correct_answer)