modified search

Author: VIVPM

app.py

@@ -49,7 +49,6 @@ def handle_auth(supabase):
 def render_sidebar_config():
     render_settings(st)
     api_key = st.sidebar.text_input("SambaNova API Key", type="password", disabled=st.session_state.is_processing_docs)
-    st.sidebar.caption("Don't have an API key? [Get SambaNova API Key](https://cloud.sambanova.ai/)")
     model_choice = st.sidebar.selectbox("Choose model", options=[
         "Llama-3.3-Swallow-70B-Instruct-v0.4",
         "Llama-4-Maverick-17B-128E-Instruct",
@@ -65,10 +64,10 @@ def render_sidebar_config():
         "Qwen3-32B",
         "gpt-oss-120b"
     ], index=5, disabled=st.session_state.is_processing_docs)
-    firecrawl_api_key = st.sidebar.text_input("Firecrawl API Key", type="password", disabled=st.session_state.is_processing_docs)
-    render_web_search_toggle(st, firecrawl_api_key)
+    serper_api_key = st.sidebar.text_input("Serper API Key", type="password", disabled=st.session_state.is_processing_docs)
+    render_web_search_toggle(st, serper_api_key)
     render_auth_buttons(st)
-    return api_key, model_choice, firecrawl_api_key
+    return api_key, model_choice, serper_api_key
 
 def handle_file_upload(api_key, model_choice):
     uploaded_files = st.file_uploader("Upload up to 3 files each under 2MB", type=["pdf","txt","docx"], accept_multiple_files=True, disabled=st.session_state.is_processing_docs) or []
@@ -105,7 +104,7 @@ def render_chat_interface():
             st.session_state.memory = ensure_memory_from_chat(current_chat)
         render_chat_history(current_chat)
 
-def handle_chat_interaction(api_key, model_choice, firecrawl_api_key, supabase):
+def handle_chat_interaction(api_key, model_choice, serper_api_key, supabase):
     if not st.session_state.is_processing_docs:
         q = st.chat_input("Your question")
         if q:
@@ -131,8 +130,8 @@ def handle_chat_interaction(api_key, model_choice, firecrawl_api_key, supabase):
                 if not err:
                     history_text = render_history_text(st.session_state.memory)
 
-                    if st.session_state.use_web_search and firecrawl_api_key:
-                        final, sources = web_search_answer(llm, firecrawl_api_key, q)
+                    if st.session_state.use_web_search and serper_api_key:
+                        final, sources = web_search_answer(llm, serper_api_key, q)
                         source_block = "\n\n**Web Sources used:**\n" + "\n".join(f"- {s}" for s in sources) if sources else ""
                     else:
                         if "vectorstore" in st.session_state:
@@ -173,12 +172,12 @@ def main():
 
     handle_auth(supabase)
 
-    api_key, model_choice, firecrawl_api_key = render_sidebar_config()
+    api_key, model_choice, serper_api_key = render_sidebar_config()
     handle_file_upload(api_key, model_choice)
     render_chat_interface()
 
     if api_key:
-        handle_chat_interaction(api_key, model_choice, firecrawl_api_key, supabase)
+        handle_chat_interaction(api_key, model_choice, serper_api_key, supabase)
     else:
         st.info("Enter your SambaNova API key in the sidebar.")
 

services/ai_processor.py

@@ -8,7 +8,7 @@
 from langchain_pinecone import PineconeVectorStore
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.docstore.document import Document
-from langchain.docstore.document import Document
+from langchain_community.utilities import GoogleSerperAPIWrapper
 from config import PINECONE_API_KEY, PINECONE_INDEX_NAME
 from utils.common_utils import chunk_id, extract_text_from_file
 
@@ -72,41 +72,21 @@ def index_docs(vectorstore, docs):
     vectorstore.add_documents(docs, ids=ids)
     return len(docs)
 
-from firecrawl import FirecrawlApp
-
 # --- Search ---
-def web_search_answer(llm, firecrawl_api_key: str, question: str) -> tuple[str, list[str]]:
-    app = FirecrawlApp(api_key=firecrawl_api_key)
-    try:
-        # Using 'search' mode to get results similar to Google Search
-        sr = app.search(question,limit=5)
-        
-        # Firecrawl returns an object with a 'data' or 'web' attribute containing the list of results.
-        # Based on testing, it returns an object with a 'web' attribute.
-        if hasattr(sr, 'web'):
-            results_data = sr.web
-        elif hasattr(sr, 'data'):
-            results_data = sr.data
-        elif isinstance(sr, dict):
-            results_data = sr.get('data', sr.get('web', []))
-        else:
-            results_data = []
-        
-        search_results = "\n".join(
-            f"Title: {getattr(r, 'title', 'No Title')}\nLink: {getattr(r, 'url', getattr(r, 'link', ''))}\nSnippet: {getattr(r, 'description', getattr(r, 'snippet', ''))}\n"
-            for r in results_data
-        )
-        
-        prompt = PromptTemplate.from_template(
-            "Based on these web search results, answer concisely and include key sources.\n\n{results}\n\nQuestion: {q}\n\nAnswer:"
-        )
-        chain = LLMChain(llm=llm, prompt=prompt, output_key="ans")
-        ans = chain({"results": search_results, "q": question})["ans"].strip()
-        
-        sources = [getattr(r, 'url', getattr(r, 'link', '')) for r in results_data if getattr(r, 'url', None) or getattr(r, 'link', None)]
-        return ans, sources
-    except Exception as e:
-        return f"Error performing web search: {str(e)}", []
+def web_search_answer(llm, serper_api_key: str, question: str) -> tuple[str, list[str]]:
+    search_tool = GoogleSerperAPIWrapper(serper_api_key=serper_api_key, k=5)
+    sr = search_tool.results(question)
+    organic = sr.get("organic", [])
+    search_results = "\n".join(
+        f"Snippet: {o.get('snippet','')}\nLink: {o.get('link','')}" for o in organic
+    )
+    prompt = PromptTemplate.from_template(
+        "Based on these web search results, answer concisely and include key sources.\n\n{results}\n\nQuestion: {q}\n\nAnswer:"
+    )
+    chain = LLMChain(llm=llm, prompt=prompt, output_key="ans")
+    ans = chain({"results": search_results, "q": question})["ans"].strip()
+    sources = [o.get("link","") for o in organic if o.get("link")]
+    return ans, sources
 
 # --- Guardrails ---
 class InputGuardrails:

ui/components.py

@@ -22,13 +22,13 @@ def render_auth_buttons(st):
         st.session_state.show_signup = False
         st.rerun()
 
-def render_web_search_toggle(st, firecrawl_api_key: str | None):
-    enabled = st.session_state.use_web_search and bool(firecrawl_api_key)
+def render_web_search_toggle(st, serper_api_key: str | None):
+    enabled = st.session_state.use_web_search and bool(serper_api_key)
     ws_label = f"Web search: {'On' if enabled else 'Off'}"
-    new_val = st.sidebar.toggle(ws_label, value=enabled, disabled=not firecrawl_api_key, key="web_search_toggle")
+    new_val = st.sidebar.toggle(ws_label, value=enabled, disabled=not serper_api_key, key="web_search_toggle")
     st.session_state.use_web_search = bool(new_val)
-    
-    st.sidebar.caption("Don't have an API key? [Get Firecrawl API key](https://www.firecrawl.dev/)")
+    if not serper_api_key:
+        st.sidebar.caption("Add a Serper API key to enable web search.")
 
 def render_chat_search(st):
     st.sidebar.title("Search Chats")