Create app.py

app.py

@@ -0,0 +1,53 @@
+import os
+
+import streamlit as st
+
+from chatbot.data import download_test_data
+from chatbot.data import load_data
+
+# add all secrets into environmental variables
+try:
+    for key, value in st.secrets.items():
+        os.environ[key] = value
+except FileNotFoundError as e:
+    print(e)
+    print("./streamlit/secrets.toml not found. Assuming secrets are already available" "as environmental variables...")
+
+# Initialize message history
+st.header("Chat with André's research 💬 📚")
+
+if "messages" not in st.session_state.keys():  # Initialize the chat message history
+    st.session_state.messages = [{"role": "assistant", "content": "Ask me a question about André's research!"}]
+
+
+def main():
+    # setup logger sidebar
+    # st.sidebar.text("Standard output log:")
+    # _sidebar_out = st.sidebar.empty()
+    # with rd.stdout(to=_sidebar_out, format='text'):
+    #    print("test")
+
+    # setup dataset
+    download_test_data()
+    index = load_data()
+    chat_engine = index.as_chat_engine(chat_mode="condense_question", verbose=True)
+
+    if prompt := st.chat_input("Your question"):  # Prompt for user input and save to chat history
+        st.session_state.messages.append({"role": "user", "content": prompt})
+
+    for message in st.session_state.messages:  # Display the prior chat messages
+        with st.chat_message(message["role"]):
+            st.write(message["content"])
+
+    # If last message is not from assistant, generate a new response
+    if st.session_state.messages[-1]["role"] != "assistant":
+        with st.chat_message("assistant"):
+            with st.spinner("Thinking..."):
+                response = chat_engine.chat(prompt)
+                st.write(response.response)
+                message = {"role": "assistant", "content": response.response}
+                st.session_state.messages.append(message)  # Add response to message history
+
+
+if __name__ == "__main__":
+    main()