Update app.py

app.py

@@ -4,8 +4,11 @@ from bs4 import BeautifulSoup
 import requests
 from typing import List, Tuple, Optional
 
-# Initialize the InferenceClient with the model ID from Hugging Face
-client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta")
+# Initialize the InferenceClient with the specified model and API key
+client = InferenceClient(
+    model="meta-llama/Meta-Llama-3.1-405B-Instruct",
+    token=""  # Replace with your API key
+)
 
 def scrape_yahoo_search(query: str, num_results: int = 3) -> Tuple[str, str]:
     """
@@ -34,8 +37,8 @@ def scrape_yahoo_search(query: str, num_results: int = 3) -> Tuple[str, str]:
         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']
+                snippet = result.find('div', {'class': 'compText aAbs'}).get_text(strip=True) if result.find('div', {'class': 'compText aAbs'}) else "No snippet"
+                url = result.find('a')['href'] if result.find('a') else "No URL"
                 results.append((title, snippet, url))
             
             formatted_results = "\n\n".join(
@@ -46,9 +49,9 @@ def scrape_yahoo_search(query: str, num_results: int = 3) -> Tuple[str, str]:
             return "No results found.", search_url
     
     except requests.RequestException as e:
-        return f"An error occurred while making the request: {str(e)}", search_url
+        return f"Request error: {str(e)}", search_url
     except Exception as e:
-        return f"An error occurred while processing the results: {str(e)}", search_url
+        return f"Processing error: {str(e)}", search_url
 
 def extract_search_query(message: str, trigger_word: str) -> Optional[str]:
     """
@@ -66,8 +69,7 @@ def extract_search_query(message: str, trigger_word: str) -> Optional[str]:
         parts = lower_message.split(trigger_word, 1)
         if len(parts) > 1:
             query = parts[1].strip()
-            if query:
-                return query
+            return query if query else None
     return None
 
 def respond(
@@ -96,9 +98,10 @@ def respond(
     # Check for trigger word and activate search feature if present
     trigger_word = "search"
     query = extract_search_query(message, trigger_word)
+    
     if query:
-        snippet, url = scrape_yahoo_search(query, num_results=3)  # Retrieve top 3 results
-        message = f"{message}\n\nWeb Content:\n{snippet}\nSource: {url}"
+        snippet, url = scrape_yahoo_search(query, num_results=3)
+        message += f"\n\nWeb Content:\n{snippet}\nSource: {url}"
     elif query is None:
         message = "Please provide a search query after the trigger word."
 
@@ -129,11 +132,10 @@ def respond(
             token = response_chunk.choices[0].delta.content
             response += token
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        return f"AI model error: {str(e)}"
 
     return response
 
-
 # Define the ChatInterface with additional input components for user customization
 demo = gr.ChatInterface(
     fn=respond,