Update app.py

app.py

@@ -13,12 +13,37 @@ if 'device_id' not in st.session_state:
 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:
+        cfg = get_agent_config()
+        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}")
-    
-    nest_asyncio.apply()
-    sys.stdout.flush()
-    #asyncio.run(main())
+        st.write(f"Settings: Scrollable content line {i+1}")