| import os | |
| import streamlit as st | |
| import requests | |
| ares_api = os.getenv("ARES_API") | |
| url = "https://api-ares.traversaal.ai/live/predict" | |
| headers = { | |
| "x-api-key": ares_api, | |
| "content-type": "application/json" | |
| } | |
| def inference(query): | |
| payload = { "query": [query] } | |
| response = requests.post(url, json=payload, headers=headers) | |
| response_text=response.json().get('data').get('response_text') | |
| urls=response.json().get('data').get('web_url') | |
| return response_text, urls | |
| st.title('Ares Demo') | |
| query = st.text_area('Find what you seek:', 'I am planning my 10th anniversary, provide me with a list of places in Boston that are quiet, private and climate-controlled so that we can have a great conversation') | |
| if query: | |
| response_text,urls = inference(query) | |
| st.write('Query Results:') | |
| st.write(response_text) | |
| st.write('Sources:') | |
| st.write(urls) | |