traversaal-ai's picture
Update app.py
fb027f7 verified
raw
history blame contribute delete
885 Bytes
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)