Upload app.py

app.py

@@ -29,7 +29,14 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        return answer[14:]
+        print(f"Raw answer from agent: {answer}")
+        prefix = "Answer: "
+        if answer.startswith(prefix):
+            # Remove the prefix only if it exists
+            return answer[len(prefix):].strip()
+        else:
+            # Return the entire answer as is
+            return answer.strip()
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):