zaz / traversaal.py
Aamod's picture
Upload 4 files
40ae8d6 verified
raw
history blame
646 Bytes
import requests
import streamlit as st
import json
import os
def getResponse(query):
url = "https://api-ares.traversaal.ai/live/predict"
payload = { "query": [query] }
headers = {
"x-api-key": st.secrets["TRAVERSAAL"],
"content-type": "application/json"
}
response = requests.post(url, json=payload, headers=headers)
if response.status_code == 200:
# Get the JSON content from the response
json_content = response.json()
# Specify the file path where you want to save the JSON content
return json_content
else:
print(response.status_code)
return " "