update app.py for henry first agent

app.py

@@ -3,10 +3,81 @@ import datetime
 import requests
 import pytz
 import yaml
+import os 
+import json
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+@tool
+def search_arxiv_papers(topic: str, max_results: int = 5) -> str:
+    """Searches arXiv for papers related to the given topic and returns results as JSON.
+    Args:
+        topic: The search query for arXiv.
+        max_results: Maximum number of results to return (default 5).
+    """
+    try:
+        url = 'http://export.arxiv.org/api/query'
+        params = {
+            'search_query': f'all:{topic}',
+            'start': 0,
+            'max_results': max_results,
+            'sortBy': 'submittedDate',
+            'sortOrder': 'descending'
+        }
+        response = requests.get(url, params=params)
+        response.raise_for_status()
+        
+        root = xml.etree.ElementTree.fromstring(response.content)
+        entries = root.findall('{http://www.w3.org/2005/Atom}entry')
+        papers = []
+        
+        for entry in entries:
+            title = entry.find('{http://www.w3.org/2005/Atom}title').text.strip()
+            authors = [author.find('{http://www.w3.org/2005/Atom}name').text 
+                      for author in entry.findall('{http://www.w3.org/2005/Atom}author')]
+            summary = entry.find('{http://www.w3.org/2005/Atom}summary').text.strip()
+            link = entry.find('{http://www.w3.org/2005/Atom}id').text
+            
+            papers.append({
+                'title': title,
+                'authors': authors,
+                'summary': summary,
+                'link': link
+            })
+            
+        return json.dumps(papers)
+    except Exception as e:
+        return f"Error searching arXiv: {str(e)}"
+
+@tool
+def save_papers_to_folder(papers_json: str, folder: str = "downloads") -> str:
+    """Saves papers data to specified folder as text files.
+    Args:
+        papers_json: JSON string containing papers data.
+        folder: Target folder path (default 'downloads').
+    """
+    try:
+        os.makedirs(folder, exist_ok=True)
+        papers = json.loads(papers_json)
+        
+        for paper in papers:
+            # Sanitize filename
+            title = "".join([c if c.isalnum() else "_" for c in paper['title']])[:50]
+            filename = os.path.join(folder, f"{title}.txt")
+            
+            with open(filename, 'w') as f:
+                f.write(f"Title: {paper['title']}\n")
+                f.write(f"Authors: {', '.join(paper['authors'])}\n")
+                f.write(f"Link: {paper['link']}\n")
+                f.write(f"Summary: {paper['summary']}\n\n")
+                
+        return f"Saved {len(papers)} papers to {folder} directory."
+    except Exception as e:
+        return f"Error saving papers: {str(e)}"
+
+
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -55,7 +126,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, search_arxiv_papers, save_papers_to_folder], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,