Update app.py

app.py

@@ -129,13 +129,14 @@ ask_wiki(query: str) -> str:
         A single string containing the retrieved article from Wikipedia.
 
 youtube_transcript(url: str) -> str:
+    Fetch the transcript of a youtube video.
     Args:
         url: input youtube url.
     Returns:
         A single string containing the transcript of the youtube videos.
 """
     sys_msg = SystemMessage(
-        content=f"You are a helpful assistant at answering user questions. You can access provided tools:\n{textual_description_of_tool}\n"
+        content=f"You are a helpful assistant at answering user questions. Your final answer will be between <answers> tags. You can access provided tools:\n{textual_description_of_tool}\n"
     )
 
     return {
@@ -161,6 +162,11 @@ builder.add_conditional_edges(
 builder.add_edge("tools", "assistant")
 agent_graph = builder.compile()
 
+def extract_answer(text):
+    match = re.search(r'<answer>(.*?)</answer>', text, re.DOTALL)
+    if match:
+        return match.group(1).strip()
+    return 'None'
 
 class BasicAgent:
     def __init__(self):
@@ -182,7 +188,8 @@ class BasicAgent:
         ]
         response = await agent_graph.ainvoke({"messages": messages}, config={"recursion_limit": 5})
         response_text = response['messages'][-1].content
-        return response_text.split('</think>')[-1]
+        # return response_text.split('</think>')[-1]
+        return extract_answer(response_text)
 
 
 async def run_and_submit_all( profile: gr.OAuthProfile | None):