Update app.py

app.py

@@ -9,14 +9,27 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def find_website(site:str) -> str:
-    """A tool that finds websites.
+def find_website(site: str) -> str:
+    """A tool that fetches a website and returns its text content.
     Args:
-        site: website to find
-        
+        site: website URL to fetch (must include https:// or http://)
     """
-    webpage = requests.get(site)
-    return f"The webpage {webpage} has been found."
+    try:
+        # Add https:// if not provided
+        if not site.startswith(('http://', 'https://')):
+            site = 'https://' + site
+        
+        response = requests.get(site, timeout=10)
+        response.raise_for_status()  # Raises error for bad status codes
+        
+        # Return first 1000 characters of content
+        content = response.text[:1000]
+        return f"Website {site} fetched successfully!\n\nContent preview:\n{content}..."
+        
+    except requests.exceptions.Timeout:
+        return f"Error: Timeout while fetching {site}"
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching {site}: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: