Update app.py

app.py

@@ -27,17 +27,16 @@ def my_custom_tool(arg1:str)-> str: #it's import to specify the return type
   
     try:
        response = requests.get(url, headers=headers)
-       response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
+       response.raise_for_status()
        soup = BeautifulSoup(response.content, "html.parser")
        results = soup.find_all("li", class_="booklink")
-       urls = []
-       for result in results[:1]: # grabs the top 1
-           link = result.find("a", class_="link")
+       if results: #Check if results are present.
+           link = results[0].find("a", class_="link") #Grabs the first result
            if link:
                full_url = "https://www.gutenberg.org" + link["href"]
-               urls.append(full_url)
-       return urls
-
+               return full_url
+       return None #Return None if no results, or no link.
+    
     except requests.exceptions.RequestException as e:
        print(f"Error: {e}")
        return None