Spaces:
Running
Running
File size: 2,629 Bytes
d61b6cb 5afd6f9 d61b6cb 5afd6f9 d61b6cb 5afd6f9 d61b6cb |
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 66 67 68 69 70 71 72 73 74 75 |
from annotated_text import annotation
from json import JSONDecodeError
import logging
from markdown import markdown
import requests
import streamlit as st
from utils.haystack import query, start_haystack
from utils.ui import reset_results, set_initial_state, sidebar
set_initial_state()
sidebar()
st.write("# Get the summaries of latest top Hacker News posts 🧡")
if st.session_state.get("model") == None:
mistral, openai = st.columns(2)
with mistral:
mistral_pressed = st.button("Mistral")
if mistral_pressed:
st.session_state["model"] = "Mistral"
with openai:
openai_pressed = st.button("OpenAI")
if openai_pressed:
st.session_state["model"] = "GPT-4"
if st.session_state.get("model") and (st.session_state.get("HF_TGI_TOKEN") or st.session_state.get("OPENAI_API_KEY")):
if st.session_state.get("HF_TGI_TOKEN"):
pipeline = start_haystack(st.session_state.get("HF_TGI_TOKEN"), st.session_state.get("model"))
st.session_state["api_key_configured"] = True
elif st.session_state.get("OPENAI_API_KEY"):
pipeline = start_haystack(st.session_state.get("OPENAI_API_KEY"), st.session_state.get("model"))
st.session_state["api_key_configured"] = True
search_bar, button = st.columns(2)
# Search bar
with search_bar:
top_k = st.slider('How many of the top posts should I summarize?', 0, 5, 0)
with button:
st.write("")
st.write("")
run_pressed = st.button("Get summaries")
else:
st.write("Please provide your Hugging Face or OpenAI key to start using the application")
st.write("If you are using a smaller screen, open the sidebar from the top left to provide your token 🙌")
if st.session_state.get("api_key_configured"):
run_query = (
run_pressed or top_k != st.session_state.top_k
)
# Get results for query
if run_query and top_k:
reset_results()
st.session_state.username = top_k
with st.spinner("🔎"):
try:
st.session_state.result = query(top_k, pipeline)
except JSONDecodeError as je:
st.error(
"👓 An error occurred reading the results. Is the document store working?"
)
except Exception as e:
logging.exception(e)
st.error("🐞 An error occurred during the request.")
if st.session_state.result:
summaries = st.session_state.result
st.write(summaries[0])
|