Spaces:
Running
Running
| import streamlit as st | |
| from PIL import Image | |
| def set_state_if_absent(key, value): | |
| if key not in st.session_state: | |
| st.session_state[key] = value | |
| def set_initial_state(): | |
| set_state_if_absent("top_k", "How many of the top posts would you like a summary for?") | |
| set_state_if_absent("result", None) | |
| set_state_if_absent("haystack_started", False) | |
| set_state_if_absent("model", None) | |
| def reset_results(*args): | |
| st.session_state.result = None | |
| st.session_state.top_k = None | |
| def set_hf_token(api_key: str): | |
| st.session_state["HF_TGI_TOKEN"] = api_key | |
| def set_openai_key(api_key: str): | |
| st.session_state["OPENAI_API_KEY"] = api_key | |
| def sidebar(): | |
| with st.sidebar: | |
| # image = Image.open('logo/haystack-logo-colored.png') | |
| st.markdown("Thanks for coming to this 🤗 Space.\n\n" | |
| "This is a project for fun, and is not a final product." | |
| " There's a lot that can be improved to make this app better.\n\n" | |
| "**Take results with a grain of** 🧂\n\n" | |
| "For more on how this was built, instructions to run locally and to contribute: [visit GitHub](https://github.com/TuanaCelik/hackernews-summaries#readme)") | |
| st.markdown( | |
| "## How to use\n" | |
| "1. Enter your Hugging Face Token below\n" | |
| "2. Select the number of summaries you want\n" | |
| "3. Enjoy 🤗\n" | |
| ) | |
| if st.session_state.model == "Mistral": | |
| api_key_input = st.text_input( | |
| "Hugging Face Token", | |
| type="password", | |
| placeholder="Paste your Hugging Face TGI Token", | |
| help="You can get your API key from https://platform.openai.com/account/api-keys.", | |
| value=st.session_state.get("HF_TGI_TOKEN", ""), | |
| ) | |
| if api_key_input: | |
| set_hf_token(api_key_input) | |
| elif st.session_state.model == "GPT-4": | |
| api_key_input = st.text_input( | |
| "OpenAI API Key", | |
| type="password", | |
| placeholder="Paste your OpenAI API Key", | |
| help="You can get your API key from https://platform.openai.com/account/api-keys.", | |
| value=st.session_state.get("OPENAI_API_KEY", ""), | |
| ) | |
| if api_key_input: | |
| set_openai_key(api_key_input) | |
| st.markdown("---") | |
| st.markdown( | |
| "## How this works\n" | |
| "This app was built with [Haystack 2.0-Beta](https://haystack.deepset.ai) using the" | |
| " [`HuggingFaceTGIGenerator`](https://docs.haystack.deepset.ai/v2.0/docs/huggingfacetgigenerator) and [`PromptBuilder`](https://docs.haystack.deepset.ai/v2.0/docs/promptbuilder).\n\n" | |
| " The source code is also on [GitHub](https://github.com/TuanaCelik/hackernews-summaries)" | |
| " with instructions to run locally.\n" | |
| "You can see how the `PromptBuilder` was set up [here](https://github.com/TuanaCelik/hackernews-summaries/blob/main/utils/haystack.py)") | |
| st.markdown("---") | |
| st.markdown("Made by [tuanacelik](https://twitter.com/tuanacelik)") |