chatbot_1 / Update_chatbot.py
Guhanselvam's picture
Rename chatbot.py to Update_chatbot.py
cf67e4d verified
raw
history blame contribute delete
978 Bytes
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'])