update

app.py

@@ -2,27 +2,63 @@ import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 from dotenv import load_dotenv
+from tavily import TavilyClient
 
 # Load environment variables
 load_dotenv()
 
-# Initialize Hugging Face client
+# Initialize clients
 client = InferenceClient(
     provider="novita",
     api_key=os.getenv("HF_TOKEN"),
     bill_to="huggingface"
 )
 
+tavily_client = TavilyClient(api_key=os.getenv("TAVILY_API_KEY"))
+
+def get_web_context(query):
+    """
+    Get relevant web search results using Tavily
+    """
+    try:
+        response = tavily_client.search(
+            query=query,
+            search_depth="advanced",
+            max_results=3
+        )
+        
+        # Format the search results
+        context = "Web Search Results:\n\n"
+        for result in response['results']:
+            context += f"Title: {result['title']}\n"
+            context += f"URL: {result['url']}\n"
+            context += f"Content: {result['content']}\n\n"
+        
+        return context
+    except Exception as e:
+        return f"Error getting web context: {str(e)}"
+
 def chat(message, history):
     """
-    Process chat messages using Hugging Face's Inference Provider
+    Process chat messages using Hugging Face's Inference Provider with web context
     """
     try:
+        # Get web context
+        web_context = get_web_context(message)
+        
         # Format the conversation history
         messages = []
         for human, assistant in history:
             messages.append({"role": "user", "content": human})
             messages.append({"role": "assistant", "content": assistant})
+        
+        # Add system message with web context
+        messages.append({
+            "role": "system",
+            "content": f"You are a helpful AI assistant. Use the following web search results to inform your response:\n\n{web_context}"
+        })
+        
+        # Add user message
         messages.append({"role": "user", "content": message})
 
         # Get response from the model

requirements.txt

@@ -1,4 +1,5 @@
 gradio>=4.0.0
 huggingface-hub>=0.20.0
 python-dotenv>=1.0.0
-requests>=2.31.0 
+requests>=2.31.0
+tavily-python>=0.2.0