Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from langchain_community.document_loaders import ArxivLoader
 
 from Gradio_UI import GradioUI
 
@@ -18,6 +19,29 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+
+@tool
+def arxiv_fetch_paper_tool(arxiv_id : str) -> str:
+    """ An Tool that would fetch research papers from ArXiV collection and parse it and return the PDF text"""
+    loader = ArxivLoader(query=arxiv_id)
+    docs = loader.load()
+    title = docs[0].metadata.get("title", "this paper")
+    return "\n".join([doc.page_content for doc in docs]), title
+
+
+
+
+# @tool
+# def write_python_code(coding_problem:str) -> str:
+#     """A tool that would code to solve user request in Python on a jupyter notebook environment
+#     Args:
+#         coding_problem: the problem statement needed to be solved by writing python code.
+#     Returns the code that has to be replaced in the notebook cell.
+#     """
+
+
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +79,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,arxiv_fetch_paper_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,