def search_api(query):
    # Replace with your own search API (here's an example with a placeholder)
    # Use Serper or another search API
    response = requests.get(f"https://serper.dev/search?query={query}&api_key=614e992d87c496fb15a81a2039e00e6a42530f5c")
    return response.json()  # Modify based on the search API response format

if st.button("Send"):
    response = requests.post("http://127.0.0.1:8000/generate", json={"prompt": user_input}, stream=True)
    
    streaming_response = ""
    for line in response.iter_lines():
        if line:
            token = line.decode("utf-8")
            streaming_response += token
            st.write(token)
    
    # Example logic for insufficient LLM response
    if "I don't know" in streaming_response:
        search_results = search_api(user_input)
        st.write("Searching for recent information:")
        for result in search_results.get('results', []):
            st.write(result['title'], result['link'])