Update app.py

app.py

@@ -21,54 +21,25 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
 from bs4 import BeautifulSoup
 import json
 
-@tool
-def get_hugging_face_top_daily_paper() -> str:
-    """
-    This is a tool that returns the most upvoted paper on Hugging Face daily papers.
-    It returns the title of the paper
-    """
-    try:
-      url = "<https://huggingface.co/papers>"
-      response = requests.get(url)
-      response.raise_for_status()  # Raise an exception for bad status codes (4xx or 5xx)
-      soup = BeautifulSoup(response.content, "html.parser")
-
-      # Extract the title element from the JSON-like data in the "data-props" attribute
-      containers = soup.find_all('div', class_='SVELTE_HYDRATER contents')
-      top_paper = ""
-
-      for container in containers:
-          data_props = container.get('data-props', '')
-          if data_props:
-              try:
-                  # Parse the JSON-like string
-                  json_data = json.loads(data_props.replace('&quot;', '"'))
-                  if 'dailyPapers' in json_data:
-                      top_paper = json_data['dailyPapers'][0]['title']
-              except json.JSONDecodeError:
-                  continue
-
-      return top_paper
-    except requests.exceptions.RequestException as e:
-      print(f"Error occurred while fetching the HTML: {e}")
-      return None
+from transformers import tool
 
 @tool
-def get_paper_id_by_title(title: str) -> str:
+def is_prime(n: int) -> bool:
     """
-    This is a tool that returns the arxiv paper id by its title.
-    It returns the title of the paper
+    Check if a number is a prime number.
 
     Args:
-        title: The paper title for which to get the id.
+        n: The number to check.
+
+    Returns:
+        True if n is a prime number, False otherwise.
     """
-    api = HfApi()
-    papers = api.list_papers(query=title)
-    if papers:
-        paper = next(iter(papers))
-        return paper.id
-    else:
-        return None
+    if n < 2:
+        return False
+    for i in range(2, int(n**0.5) + 1):
+        if n % i == 0:
+            return False
+    return True
 
 
 @tool