Spaces:
Running
Running
File size: 695 Bytes
e92154d ec7ef1b 5c73d5c ec7ef1b e92154d 5c73d5c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import os
import requests
import streamlit as st
BE = os.getenv("be_url")
datasets = requests.get(f"{BE}/v1/datasets", timeout=500).json()
st.sidebar.title("Datasets")
ds = st.sidebar.selectbox(options=[ d["name"] for d in datasets],
label="Select your dataset")
print(ds)
if ds:
query = st.text_input("Enter your search query",
placeholder="Ask your question")
if query:
answer = requests.get(f"http://localhost:8080/v1/datasets/{ds}/answer?query={query}",
timeout=5000 )
print(answer.json()["answer"])
st.write(answer.json()["answer"])
else:
st.write("Choose your dataset!") |