This is a minimal user interface for STORMWikiRunner
which includes the following features:
- Allowing user to create a new article through the "Create New Article" page.
- Showing the intermediate steps of STORMWikiRunner in real-time when creating an article.
- Displaying the written article and references side by side.
- Allowing user to view previously created articles through the "My Articles" page.
- Make sure you have installed
knowledge-storm
or set up the source code correctly. - Install additional packages required by the user interface:
pip install -r requirements.txt
- Make sure you set up the API keys following the instructions in the main README file. Create a copy of
secrets.toml
and place it under.streamlit/
. - Run the following command to start the user interface:
The user interface will create a
streamlit run storm.py
DEMO_WORKING_DIR
directory in the current directory to store the outputs.
You can customize the STORMWikiRunner
powering the user interface according to the guidelines in the main README file.
The STORMWikiRunner
is initialized in set_storm_runner()
in demo_util.py. You can change STORMWikiRunnerArguments
, STORMWikiLMConfigs
, or use a different retrieval model according to your need.