Spaces:
Sleeping
Sleeping
| 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']) |