Placeholder

app.py

@@ -1,11 +1,20 @@
 from openai import OpenAI
 import streamlit as st
 
-st.title("HiddenLayer Chat")
+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)
+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"
@@ -27,38 +36,35 @@ for message in st.session_state.messages_col_2:
             st.markdown(message["content"])
 
 
-if prompt := st.chat_input("What is up?"):
+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_1
-            ],
-            stream=True,
-        )
-        response = st.write_stream(stream)
-    st.session_state.messages_col_1.append({"role": "assistant", "content": response})
-
-    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})
+        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})