from openai import OpenAI import streamlit as st st.set_page_config(layout="wide") st.columns(3)[1].title("HiddenLayer Chat") client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"]) col1, col2 = st.columns(2, gap="large") with col1: st.header("GPT-3.5") with col2: st.header("Currently GPT-3.5, later SafeLLM") if "openai_model" not in st.session_state: st.session_state["openai_model"] = "gpt-3.5-turbo" if "messages_col_1" not in st.session_state: st.session_state.messages_col_1 = [] if "messages_col_2" not in st.session_state: st.session_state.messages_col_2 = [] for message in st.session_state.messages_col_1: with col1: with st.chat_message(message["role"]): st.markdown(message["content"]) for message in st.session_state.messages_col_2: with col2: with st.chat_message(message["role"]): st.markdown(message["content"]) if prompt := st.chat_input("Enter a prompt"): st.session_state.messages_col_1.append({"role": "user", "content": prompt}) st.session_state.messages_col_2.append({"role": "user", "content": prompt}) with col1: with st.chat_message("user"): st.markdown(prompt) with st.chat_message("assistant"): stream = client.chat.completions.create( model=st.session_state["openai_model"], messages=[ {"role": m["role"], "content": m["content"]} for m in st.session_state.messages_col_1 ], stream=True, ) response = st.write_stream(stream) st.session_state.messages_col_1.append({"role": "assistant", "content": response}) with col2: with st.chat_message("user"): st.markdown(prompt) with st.chat_message("assistant"): stream = client.chat.completions.create( model=st.session_state["openai_model"], messages=[ {"role": m["role"], "content": m["content"]} for m in st.session_state.messages_col_2 ], stream=True, ) response = st.write_stream(stream) st.session_state.messages_col_2.append({"role": "assistant", "content": response})