Changing Draft Demo to Chat Demo running on GPT 3.5

pages/{7_✍️_Draft_Demo.py → 7_💬️_Chat_Demo.py}

@@ -3,6 +3,7 @@ import os
 import streamlit as st
 import streamlit_analytics
 from utils import add_logo_to_sidebar, add_footer, add_email_signup_form
+from openai import OpenAI
 
 streamlit_analytics.start_tracking()
 
@@ -21,11 +22,39 @@ st.set_page_config(
 add_logo_to_sidebar()
 st.sidebar.success("👆 Select a demo above.")
 
-st.title('✍️ Draft Demo')
-st.markdown("🏗 This demo is currently under construction. Please visit back soon.")
-
-add_email_signup_form()
-
-add_footer()
+st.title('💬 Chat Demo')
+
+client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"], organization=st.secrets["OPENAI_ORG"])
+
+if "openai_model" not in st.session_state:
+    st.session_state["openai_model"] = "gpt-3.5-turbo"
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+if prompt := st.chat_input("What is up?"):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    with st.chat_message("assistant"):
+        message_placeholder = st.empty()
+        full_response = ""
+        for response in client.chat.completions.create(
+            model=st.session_state["openai_model"],
+            messages=[
+                {"role": m["role"], "content": m["content"]}
+                for m in st.session_state.messages
+            ],
+            stream=True,
+        ):
+            full_response += (response.choices[0].delta.content or "")
+            message_placeholder.markdown(full_response + "▌")
+        message_placeholder.markdown(full_response)
+    st.session_state.messages.append({"role": "assistant", "content": full_response})
 
 streamlit_analytics.stop_tracking(unsafe_password=os.environ["ANALYTICS_PASSWORD"])

requirements.txt

@@ -27,6 +27,7 @@ mdurl==0.1.2
 nltk==3.8.1
 numba==0.56.4
 numpy==1.23.5
+openai
 packaging==23.0
 pandas==1.5.3
 Pillow==9.4.0
@@ -51,7 +52,7 @@ six==1.16.0
 spacy==3.7.2
 https://huggingface.co/spacy/en_core_web_md/resolve/main/en_core_web_md-any-py3-none-any.whl
 smmap==5.0.0
-streamlit==1.20.0
+streamlit==1.29.0
 streamlit-analytics==0.4.1
 streamlit-toggle-switch==1.0.2
 threadpoolctl==3.1.0

🏡_Home.py

@@ -38,7 +38,7 @@ st.markdown(
     - 🔗 **Compare** - Using AI to **compare** passages of text
     - 🗂 **Organise** - Using AI to **organise** a collection of texts
     - 🔎 **Find** - Using AI to **find** relevant information from a collection of texts
-    - ✍️ **Draft** - Using AI to **draft** text
+    - 💬 **Chat** - Using AI to **chat**
     - 📝 **Summarise** - Using AI to **summarise** text
     """)