Update app.py

app.py

@@ -40,7 +40,6 @@ def respond(
     max_tokens: int,
     temperature: float,
     top_p: float,
-    search_query: str = ""  # Optional search query parameter for Yahoo scraping
 ) -> str:
     """
     Generates a response from the AI model based on the user's message, chat history, and optional Yahoo search results.
@@ -52,16 +51,21 @@ def respond(
         max_tokens (int): The maximum number of tokens for the output.
         temperature (float): Sampling temperature for controlling the randomness.
         top_p (float): Top-p (nucleus sampling) for controlling diversity.
-        search_query (str): Optional search query to scrape Yahoo results from.
     
     Returns:
         str: The AI's response as it is generated, including the source URL if applicable.
     """
     
-    # Scrape Yahoo search results if a query is provided
-    if search_query:
-        snippet, url = scrape_yahoo_search(search_query)
-        message = f"{message}\n\nWeb Content:\n{snippet}\nSource: {url}"
+    # Check for trigger word and activate search feature if present
+    trigger_word = "search"
+    if trigger_word in message.lower():
+        # Extract the query from the message
+        query = message.lower().split(trigger_word, 1)[-1].strip()
+        if query:
+            snippet, url = scrape_yahoo_search(query)
+            message = f"{message}\n\nWeb Content:\n{snippet}\nSource: {url}"
+        else:
+            message = "Please provide a search query after the trigger word."
 
     # Prepare the conversation history for the API call
     messages = [{"role": "system", "content": system_message}]
@@ -109,7 +113,6 @@ demo = gr.ChatInterface(
             step=0.05,
             label="Top-p (nucleus sampling)",
         ),
-        gr.Textbox(value="", label="Yahoo Search Query (optional)", placeholder="Enter a search query")
     ],
     title="Chatbot Interface",
     description="A customizable chatbot interface using Hugging Face's Inference API with Yahoo search scraping capabilities.",