Update app.py

app.py

@@ -5,27 +5,29 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 import urllib, urllib.request
-import xml.etree.ElementTree as ET
+
 from Gradio_UI import GradioUI
+import re
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_arxiv_tool(keyword:str, n:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
+def my_arxiv_tool(keyword: str, n: int) -> str:
     """A tool that returns "n" research paper links for given keyword on arXiv.
     Args:
-        keyword:a keyword related to research paper.
+        keyword: a keyword related to research paper.
         n: number of research papers.
     """
-    url = f'http://export.arxiv.org/api/query?search_query=all:{keyword}&year=2002&start=0&max_results={n}'
+    url = f'http://export.arxiv.org/api/query?search_query=all:{keyword}&start=0&max_results={n}'
     data = urllib.request.urlopen(url)
     xml_string = data.read().decode('utf-8')
-
-    # Parse the XML string
-    root = ET.fromstring(xml_string)
-
-    # Convert the XML to a string
-    result_string = ET.tostring(root, encoding='unicode', method='xml')
+    
+    # Extract links using regular expressions
+    links = re.findall(r'<id>(http://arxiv\.org/abs/.*?)</id>', xml_string)
+    
+    # Format the result as a string
+    result_string = f"Research paper links for keyword '{keyword}':\n\n"
+    for i, link in enumerate(links, 1):
+        result_string += f"{i}. {link}\n"
+    
     return result_string
 
 @tool