| --- |
| title: Haystack Search Pipeline with Streamlit |
| emoji: π |
| colorFrom: indigo |
| colorTo: indigo |
| sdk: streamlit |
| sdk_version: 1.23.0 |
| app_file: app.py |
| pinned: false |
| --- |
| |
| # Template Streamlit App for Haystack Search Pipelines |
|
|
| This template [Streamlit](https://docs.streamlit.io/) app set up for simple [Haystack search applications](https://docs.haystack.deepset.ai/docs/semantic_search). The template is ready to do QA with **Retrievel Augmented Generation**, or **Ectractive QA** |
|
|
| See the ['How to use this template'](#how-to-use-this-template) instructions below to create a simple UI for your own Haystack search pipelines. |
|
|
| Below you will also find instructions on how you could [push this to Hugging Face Spaces π€](#pushing-to-hugging-face-spaces-). |
|
|
| ## Installation and Running |
| To run the bare application which does _nothing_: |
| 1. Install requirements: `pip install -r requirements.txt` |
| 2. Run the streamlit app: `streamlit run app.py` |
|
|
| This will start up the app on `localhost:8501` where you will find a simple search bar. Before you start editing, you'll notice that the app will only show you instructions on what to edit. |
|
|
| ### Optional Configurations |
|
|
| You can set optional cofigurations to set the: |
| - `--task` you want to start the app with: `rag` or `extractive` (default: rag) |
| - `--store` you want to use: `inmemory`, `opensearch`, `weaviate` or `milvus` (default: inmemory) |
| - `--name` you want to have for the app. (default: 'My Search App') |
|
|
| E.g.: |
|
|
| ```bash |
| streamlit run app.py -- --store opensearch --task extractive --name 'My Opensearch Documentation Search' |
| ``` |
|
|
| In a `.env` file, include all the config settings that you would like to use based on: |
| - The DocumentStore of your choice |
| - The Extractive/Generative model of your choice |
|
|
| While the `/utils/config.py` will create default values for some configurations, others have to be set in the `.env` such as the `OPENAI_KEY` |
|
|
| Example `.env` |
|
|
| ``` |
| OPENAI_KEY=YOUR_KEY |
| EMBEDDING_MODEL=sentence-transformers/all-MiniLM-L12-v2 |
| GENERATIVE_MODEL=text-davinci-003 |
| ``` |
|
|
|
|
| ## How to use this template |
| 1. Create a new repository from this template or simply open it in a codespace to start playing around π |
| 2. Make sure your `requirements.txt` file includes the Haystack and Streamlit versions you would like to use. |
| 3. Change the code in `utils/haystack.py` if you would like a different pipeline. |
| 4. Create a `.env`file with all of your configuration settings. |
| 5. Make any UI edits you'd like to and [share with the Haystack community](https://haystack.deepeset.ai/community) |
| 6. Run the app as show in [installation and running](#installation-and-running) |
|
|
| ### Repo structure |
| - `./utils`: This is where we have 3 files: |
| - `config.py`: This file extracts all of the configuration settings from a `.env` file. For some config settings, it uses default values. An example of this is in [this demo project](https://github.com/TuanaCelik/should-i-follow/blob/main/utils/config.py). |
| - `haystack.py`: Here you will find some functions already set up for you to start creating your Haystack search pipeline. It includes 2 main functions called `start_haystack()` which is what we use to create a pipeline and cache it, and `query()` which is the function called by `app.py` once a user query is received. |
| - `ui.py`: Use this file for any UI and initial value setups. |
| - `app.py`: This is the main Streamlit application file that we will run. In its current state it has a simple search bar, a 'Run' button, and a response that you can highlight answers with. |
|
|
| ### What to edit? |
| There are default pipelines both in `start_haystack_extractive()` and `start_haystack_rag()` |
|
|
| - Change the pipelines to use the embedding models, extractive or generative models as you need. |
| - If using the `rag` task, change the `default_prompt_template` to use one of our available ones on [PromptHub](https://prompthub.deepset.ai) or create your own `PromptTemplate` |
|
|
|
|
| ## Pushing to Hugging Face Spaces π€ |
|
|
| Below is an example GitHub action that will let you push your Streamlit app straight to the Hugging Face Hub as a Space. |
|
|
| A few things to pay attention to: |
|
|
| 1. Create a New Space on Hugging Face with the Streamlit SDK. |
| 2. Create a Hugging Face token on your HF account. |
| 3. Create a secret on your GitHub repo called `HF_TOKEN` and put your Hugging Face token here. |
| 4. If you're using DocumentStores or APIs that require some keys/tokens, make sure these are provided as a secret for your HF Space too! |
| 5. This readme is set up to tell HF spaces that it's using streamlit and that the app is running on `app.py`, make any changes to the frontmatter of this readme to display the title, emoji etc you desire. |
| 6. Create a file in `.github/workflows/hf_sync.yml`. Here's an example that you can change with your own information, and an [example workflow](https://github.com/TuanaCelik/should-i-follow/blob/main/.github/workflows/hf_sync.yml) working for the [Should I Follow demo](https://huggingface.co/spaces/deepset/should-i-follow) |
|
|
| ```yaml |
| name: Sync to Hugging Face hub |
| on: |
| push: |
| branches: [main] |
| |
| # to run this workflow manually from the Actions tab |
| workflow_dispatch: |
| |
| jobs: |
| sync-to-hub: |
| runs-on: ubuntu-latest |
| steps: |
| - uses: actions/checkout@v2 |
| with: |
| fetch-depth: 0 |
| lfs: true |
| - name: Push to hub |
| env: |
| HF_TOKEN: ${{ secrets.HF_TOKEN }} |
| run: git push --force https://{YOUR_HF_USERNAME}:$HF_TOKEN@{YOUR_HF_SPACE_REPO} main |
| ``` |
|
|