Update app.py

app.py

@@ -112,6 +112,45 @@ Answer:"""
         except Exception:
             return f"Error executing code:\n{traceback.format_exc()}"
 
+import requests
+from smolagents import Tool
+
+class ArxivSearchTool(Tool):
+    name = "arxiv_search"
+    description = "Search Arxiv for papers matching a query and return titles and links."
+    inputs = {
+        "query": {"type": "string", "description": "Search query for Arxiv papers"}
+    }
+    output_type = "string"
+
+    def forward(self, query: str) -> str:
+        url = "http://export.arxiv.org/api/query"
+        params = {
+            "search_query": query,
+            "start": 0,
+            "max_results": 3,
+            "sortBy": "relevance",
+            "sortOrder": "descending"
+        }
+        try:
+            response = requests.get(url, params=params, timeout=10)
+            response.raise_for_status()
+            # Simple parse titles and links (basic, for demo)
+            import xml.etree.ElementTree as ET
+            root = ET.fromstring(response.content)
+            ns = {"atom": "http://www.w3.org/2005/Atom"}
+
+            entries = root.findall("atom:entry", ns)
+            results = []
+            for entry in entries:
+                title = entry.find("atom:title", ns).text.strip().replace('\n', ' ')
+                link = entry.find("atom:id", ns).text.strip()
+                results.append(f"{title}\n{link}")
+            return "\n\n".join(results) if results else "No results found."
+        except Exception as e:
+            return f"Error during Arxiv search: {e}"
+
+
 
 from transformers import pipeline
 from smolagents import Tool