Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
| 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() | |
| st.set_page_config( | |
| page_title="Draft Demo", | |
| page_icon="✍️", | |
| layout="wide", | |
| initial_sidebar_state="expanded", | |
| menu_items={ | |
| 'Get Help': 'mailto:hello@simplexico.ai', | |
| 'Report a bug': None, | |
| 'About': "## This a demo showcasing different Legal AI Actions" | |
| } | |
| ) | |
| add_logo_to_sidebar() | |
| st.title('💬 Chat Demo') | |
| client = OpenAI(api_key=os.environ["OPENAI_API_KEY"], organization=os.environ["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"]) |