Spaces:
Runtime error
Runtime error
File size: 1,656 Bytes
8df121c a831acd 70b2fc9 05feb2b 70b2fc9 05feb2b 70b2fc9 0b1473d 70b2fc9 8df121c 05feb2b ea67dd2 8df121c 2e9668f a831acd 0b1473d 05feb2b 0b1473d 8df121c 05feb2b 8df121c |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import glob
import os
import logging
import sys
import streamlit as st
from haystack import Pipeline
logging.basicConfig(
level=logging.DEBUG,
format="%(levelname)s %(asctime)s %(name)s:%(message)s",
handlers=[logging.StreamHandler(sys.stdout)],
force=True,
)
p_1 = None
p_2 = None
def app_init():
# indexing_pipeline = Pipeline.load_from_yaml("pipeline.yaml", pipeline_name="indexing")
# file_paths = glob.glob("data/*")
# ds = indexing_pipeline.get_node("DocumentStore")
# ds.delete_all_documents()
# indexing_pipeline.run(file_paths=file_paths)
# ds.update_embeddings(indexing_pipeline.get_node("Retriever"))
# ds.save(config_path="my_faiss_config.json", index_path="my_faiss_index.faiss")
os.environ["OPENAI_API_KEY"] = st.secrets["OPENAI_API_KEY"]
global p_1
p_1 = Pipeline.load_from_yaml("pipeline.yaml", pipeline_name="query_1")
global p_2
p_2 = Pipeline.load_from_yaml("pipeline.yaml", pipeline_name="query_2")
def main():
app_init()
st.title("Haystack Demo")
input = st.text_input("Query ...")
query_type = st.radio("Type",
("Retrieval Augmented", "Retrieval Augmented with Sources",
"Retrieval Augmented with Web Search"))
col_1, col_2 = st.columns(2)
with col_1:
st.text("PLAIN")
answers = p_1.run(input)["answers"]
for ans in answers:
st.text(ans.answer)
with col_2:
st.write(query_type.upper())
answers = p_2.run(input)["answers"]
for ans in answers:
st.text(ans.answer)
if __name__ == "__main__":
main()
|