Update app.py

app.py

@@ -37,7 +37,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def google_search(query):
+def google_search(query: str) -> str:
     """A tool that performs a Google search using a given query and retrieves search result titles and URLs.
     Args:
         query (str): The search term or phrase to look up on Google.
@@ -46,12 +46,20 @@ def google_search(query):
     response = requests.get(f'https://www.google.com/search?q={query}', headers=headers)
     if response.status_code == 200:
         soup = BeautifulSoup(response.text, 'html.parser')
+        results = []
+        
         for g in soup.find_all('div', class_='g'):
-            title = g.find('h3').text
-            link = g.find('a')['href']
-            print(f'Title: {title}\nLink: {link}\n')
+            title_tag = g.find('h3')
+            link_tag = g.find('a')
+            
+            if title_tag and link_tag:
+                title = title_tag.text
+                link = link_tag['href']
+                results.append(f"Title: {title}\nLink: {link}\n")
+
+        return "\n".join(results) if results else "No results found."
     else:
-        print('Failed to retrieve search results')
+        return f"Failed to retrieve search results. HTTP Status Code: {response.status_code}"
 
 
 final_answer = FinalAnswerTool()