| import streamlit as st |
| from multipage import MultiPage |
| from pages import ask, settings, info |
|
|
|
|
| def init_session_key_value(key, value): |
| if key not in st.session_state: |
| st.session_state[key] = value |
|
|
|
|
| lfqa_api = "HuggingFace" if "api_lfqa_selector" not in st.secrets else st.secrets["api_lfqa_selector"] |
| session_values = {"api_lfqa_selector": lfqa_api, |
| "tts": "Google", |
| "min_length": 64, |
| "max_length": 256, |
| "do_sample": False, |
| "early_stopping": True, |
| "num_beams": 8, |
| "temperature": 1.0, |
| "top_k": None, |
| "top_p": None, |
| "no_repeat_ngram_size": 3, |
| "num_return_sequences": 1} |
|
|
| for k, v in session_values.items(): |
| init_session_key_value(k, v) |
|
|
| app = MultiPage() |
| st.set_page_config( |
| page_title="Wikipedia Assistant", |
| initial_sidebar_state="expanded", |
| ) |
| |
| app.add_page("Home", "house", ask.app) |
| app.add_page("Settings", "gear", settings.app) |
| app.add_page("Info", "info", info.app) |
|
|
| |
| app.run() |
|
|