Update app.py

app.py

@@ -10,13 +10,13 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def search_articles(text2search: str, start: int, end: int, output_md: str = "arxiv_articles.md") -> str:
+def search_articles(search_term: str, start: int, end: int, output_md: str = "arxiv_articles.md") -> str:
     """
     Searches for academic articles on arxiv.org using a given search term, retrieves 
     results within a given range, and saves the output as a Markdown file.
 
     Args:
-        text2search (str): The keyword or phrase to search for in ArXiv articles.
+        search_term (str): The keyword or phrase to search for in ArXiv articles.
         start (int): The starting index of the search results (used for pagination).
         end (int): The number of articles to retrieve from the API.
         output_md (str, optional): The name of the output Markdown file. Defaults to "arxiv_articles.md".
@@ -31,8 +31,8 @@ def search_articles(text2search: str, start: int, end: int, output_md: str = "ar
     try:
         # Construct the ArXiv API query
         search_url = (
-            f"http://export.arxiv.org/api/query?search_query=all:{text2search}"
-            f"&start={start}&max_results=5&sortBy=submittedDate&sortOrder=descending"
+            f"http://export.arxiv.org/api/query?search_query=all:{search_term}"
+            f"&start={start}&max_results={end}&sortBy=submittedDate&sortOrder=descending"
         )
 
         response = requests.get(search_url)
@@ -58,11 +58,11 @@ def search_articles(text2search: str, start: int, end: int, output_md: str = "ar
             articles.append(f"### [{title}]({link})\n**Published Date:** {published_date}\n")
 
         if not articles:
-            return f"No articles found for '{text2search}'."
+            return f"No articles found for '{search_term}'."
 
         # Save results to a Markdown file
         with open(output_md, "w", encoding="utf-8") as md_file:
-            md_file.write(f"# ArXiv Articles on '{text2search}'\n\n")
+            md_file.write(f"# ArXiv Articles on '{search_term}'\n\n")
             md_file.writelines("\n".join(articles))
 
         return f"Markdown file created: {os.path.abspath(output_md)}"