|
|
import streamlit as st |
|
|
import torch |
|
|
import uuid |
|
|
import nest_asyncio |
|
|
import asyncio |
|
|
import sys |
|
|
|
|
|
torch.classes.__path__ = [] |
|
|
|
|
|
if 'device_id' not in st.session_state: |
|
|
st.session_state.device_id = str(uuid.uuid4()) |
|
|
|
|
|
if "feedback_key" not in st.session_state: |
|
|
st.session_state.feedback_key = 0 |
|
|
|
|
|
|
|
|
async def launch_bot(): |
|
|
def reset(): |
|
|
st.session_state.messages = [{"role": "assistant", "content": initial_prompt, "avatar": "🦖"}] |
|
|
st.session_state.log_messages = [] |
|
|
st.session_state.prompt = None |
|
|
st.session_state.ex_prompt = None |
|
|
st.session_state.first_turn = True |
|
|
st.session_state.show_logs = False |
|
|
if 'agent' not in st.session_state: |
|
|
st.session_state.agent = initialize_agent(cfg, agent_progress_callback=agent_progress_callback) |
|
|
else: |
|
|
st.session_state.agent.clear_memory() |
|
|
if 'cfg' not in st.session_state: |
|
|
|
|
|
st.session_state.cfg = cfg |
|
|
st.session_state.ex_prompt = None |
|
|
example_messages = [example.strip() for example in cfg.examples.split(";")] if cfg.examples else [] |
|
|
st.session_state.example_messages = [em for em in example_messages if len(em)>0] |
|
|
reset() |
|
|
|
|
|
cfg = st.session_state.cfg |
|
|
sys.stdout.flush() |
|
|
|
|
|
if "feedback_key" not in st.session_state: |
|
|
st.session_state.feedback_key = 0 |
|
|
|
|
|
if __name__ == "__main__": |
|
|
st.set_page_config(page_title="Test Sticky Header", layout="wide") |
|
|
nest_asyncio.apply() |
|
|
asyncio.run(launch_bot()) |
|
|
st.image("images/SBS_Chapter_Index.png") |
|
|
for i in range(50): |
|
|
st.write(f"Settings: Scrollable content line {i+1}") |
