Update app.py

app.py

@@ -7,16 +7,16 @@ from typing import List, Tuple, Optional
 # Initialize the InferenceClient with the model ID from Hugging Face
 client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta")
 
-def scrape_yahoo_search(query: str, num_results: int = 1) -> Tuple[str, str]:
+def scrape_yahoo_search(query: str, num_results: int = 3) -> Tuple[str, str]:
     """
-    Scrapes Yahoo search results for the given query and returns the top results' snippets and URLs.
+    Scrapes Yahoo search results for the given query and returns detailed snippets and URLs for the top results.
     
     Args:
         query (str): The search query.
         num_results (int): Number of results to retrieve.
     
     Returns:
-        Tuple[str, str]: The snippets and URLs of the top search results.
+        Tuple[str, str]: The formatted snippets and URLs of the top search results.
     """
     search_url = f"https://search.yahoo.com/search?p={query}"
     headers = {
@@ -29,19 +29,26 @@ def scrape_yahoo_search(query: str, num_results: int = 1) -> Tuple[str, str]:
         soup = BeautifulSoup(response.content, 'html.parser')
         
         results = []
-        for result in soup.find_all('div', {'class': 'dd algo algo-sr Sr'}, limit=num_results):
-            snippet = result.find('div', {'class': 'compText aAbs'}).get_text(strip=True)
-            url = result.find('a')['href']
-            results.append((snippet, url))
+        result_elements = soup.find_all('div', {'class': 'dd algo algo-sr Sr'}, limit=num_results)
         
-        if results:
-            snippets = "\n\n".join(f"Snippet: {snippet}\nSource: {url}" for snippet, url in results)
-            return snippets, search_url
+        if result_elements:
+            for result in result_elements:
+                title = result.find('h3').get_text(strip=True) if result.find('h3') else "No title"
+                snippet = result.find('div', {'class': 'compText aAbs'}).get_text(strip=True)
+                url = result.find('a')['href']
+                results.append((title, snippet, url))
+            
+            formatted_results = "\n\n".join(
+                f"Title: {title}\nSnippet: {snippet}\nURL: {url}" for title, snippet, url in results
+            )
+            return formatted_results, search_url
         else:
             return "No results found.", search_url
     
+    except requests.RequestException as e:
+        return f"An error occurred while making the request: {str(e)}", search_url
     except Exception as e:
-        return f"An error occurred while scraping Yahoo: {str(e)}", search_url
+        return f"An error occurred while processing the results: {str(e)}", search_url
 
 def extract_search_query(message: str, trigger_word: str) -> Optional[str]:
     """