Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 import logging
+import json
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -61,6 +62,16 @@ def query(payload, api_url):
         logger.error(f"Failed to decode JSON response: {response.text}")
         return None
 
+# Function to perform Google search using SerpHouse API
+def search_web(query):
+    search_url = f"https://serphouse.com/api/v1/google_custom_search?api_key=V38CNn4HXpLtynJQyOeoUensTEYoFy8PBUxKpDqAW1pawT1vfJ2BWtPQ98h6&q={query}&num=5"
+    response = requests.get(search_url)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        logger.error(f"Error while searching: {response.status_code}")
+        return None
+
 # Chat interface
 st.title("🤖 DeepSeek Chatbot")
 st.caption("Powered by Hugging Face Inference API - Configure in sidebar")
@@ -79,9 +90,19 @@ if prompt := st.chat_input("Type your message..."):
 
     try:
         with st.spinner("Generating response..."):
-            # Prepare the payload for the API
-            # Combine system message and user input into a single prompt
-            full_prompt = f"{system_message}\n\nUser: {prompt}\nAssistant:"
+            # Perform a web search based on the user input
+            search_results = search_web(prompt)
+
+            # Process search results if available
+            if search_results and "items" in search_results:
+                search_content = "\n".join([f"**{item['title']}**: {item['snippet']}" for item in search_results["items"]])
+                search_content = f"Here are some search results related to your question:\n\n{search_content}\n\n"
+
+                # Combine the system message, search results, and user input into a single prompt
+                full_prompt = f"{system_message}\n\n{search_content}User: {prompt}\nAssistant:"
+            else:
+                full_prompt = f"{system_message}\n\nUser: {prompt}\nAssistant:"
+
             payload = {
                 "inputs": full_prompt,
                 "parameters": {