Creating utils

utils.py

@@ -0,0 +1,45 @@
+def display_msg(msg, author):
+    """Method to display message on the UI
+    Args:
+        msg (str): message to display
+        author (str): author of the message -user/assistant
+    """
+    st.session_state.messages.append({"role": author, "content": msg})
+    st.chat_message(author).write(msg)
+
+def configure_openai():
+    openai_api_key = st.sidebar.text_input(
+        label="OpenAI API Key",
+        type="password",
+        value=st.session_state['OPENAI_API_KEY'] if 'OPENAI_API_KEY' in st.session_state else '',
+        placeholder="sk-..."
+        )
+    if openai_api_key:
+        st.session_state['OPENAI_API_KEY'] = openai_api_key
+        os.environ['OPENAI_API_KEY'] = openai_api_key
+    else:
+        st.error("Please add your OpenAI API key to continue.")
+        st.info("Obtain your key from this link: https://platform.openai.com/account/api-keys")
+        st.stop()
+
+    model = "gpt-3.5-turbo"
+    try:
+        client = openai.OpenAI()
+        available_models = [{"id": i.id, "created":datetime.fromtimestamp(i.created)} for i in client.models.list() if str(i.id).startswith("gpt")]
+        available_models = sorted(available_models, key=lambda x: x["created"])
+        available_models = [i["id"] for i in available_models]
+
+        model = st.sidebar.selectbox(
+            label="Model",
+            options=available_models,
+            index=available_models.index(st.session_state['OPENAI_MODEL']) if 'OPENAI_MODEL' in st.session_state else 0
+        )
+        st.session_state['OPENAI_MODEL'] = model
+    except openai.AuthenticationError as e:
+        st.error(e.body["message"])
+        st.stop()
+    except Exception as e:
+        print(e)
+        st.error("Something went wrong. Please try again later.")
+        st.stop()
+    return model