Update app.py

app.py

@@ -2,22 +2,22 @@ import openai
 import streamlit as st
 from streamlit_chat import message
 import os 
+from pathlib import Path
+import base64
+
+# Initial page config
+
+st.set_page_config(
+     page_title='Streamlit cheat sheet',
+     layout="wide",
+     initial_sidebar_state="expanded",
+)
 
-# Setting page title and header
-st.set_page_config(page_title="Tutor", page_icon=":heavy_plus_sign:")
-st.markdown("""
-<div>
-<h3 style='text-align: center;'> Tutor Session with Mary B (Tutor) </h3>
-<p style='text-align: center;'> right now the bot is programmed to talk to a 1st grade student Becky for a Math class. The class only have 1 question: "If Robin has 3 cookies and Selena gives him 2 more, how many cookies would Robin have?"
-</div>"""
-            , unsafe_allow_html=True)
 
 # Set org ID and API key
 openai.api_key = os.environ['TOKEN']
 
 
-# Set context 
-
 context = """You are playing the role of a tutor named Mary. You specifically focus on math and english, though you are capable of addressing any subject.
 
 You tutor American students in the public school system. You tutor with the intent to catch students up to their respective grade level. Your goal is to improve learning outcomes through engaging teaching methods; for example, you often pull in age-appropriate real word scenarios to help explain harder concepts. You have a degree in teaching, with an emphasis in special education. You studied the works of Lev Vygotsky and believe that it is the responsibility of the teacher to meet the student at their personal level. You have worked as a teacher for 10 years, and have recently retired and tutor in your free time.
@@ -60,52 +60,23 @@ If a student asks for reward points, encourage them to engage in the session to
 
 """
 
-import openai
-import streamlit as st
-
 def main():
     cs_sidebar()
     cs_body()
 
     return None
 
-def cs_body():
-    
-    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 = []
-        st.session_state.messages.append({"role": "system", "content": context})
-    
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
-    
-    if prompt := st.chat_input("Start talking with your tutor!"):
-        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 openai.ChatCompletion.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.get("content", "")
-                message_placeholder.markdown(full_response + "▌")
-            message_placeholder.markdown(full_response)
-        st.session_state.messages.append({"role": "assistant", "content": full_response})
+# Thanks to streamlitopedia for the following code snippet
+
+def img_to_bytes(img_path):
+    img_bytes = Path(img_path).read_bytes()
+    encoded = base64.b64encode(img_bytes).decode()
+    return encoded
+
+# sidebar
 
-    return None
-    
 def cs_sidebar():
+
     st.sidebar.markdown('''[<img src='data:image/png;base64,{}' class='img-fluid' width=32 height=32>](https://streamlit.io/)'''.format(img_to_bytes("logomark_website.png")), unsafe_allow_html=True)
     st.sidebar.header('Streamlit cheat sheet')
 
@@ -146,9 +117,50 @@ pip install streamlit-nightly --upgrade
 
     st.sidebar.markdown('''<small>[st.cheat_sheet v1.8.0](https://github.com/daniellewisDL/streamlit-cheat-sheet)  | April 2022</small>''', unsafe_allow_html=True)
 
-    return None    
-        
+    return None
+
+##########################
+# Main body of cheat sheet
+##########################
+
+def cs_body():
+    # Magic commands
+    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 = []
+        st.session_state.messages.append({"role": "system", "content": context})
+    
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+    
+    if prompt := st.chat_input("Start talking with your tutor!"):
+        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 openai.ChatCompletion.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.get("content", "")
+                message_placeholder.markdown(full_response + "▌")
+            message_placeholder.markdown(full_response)
+        st.session_state.messages.append({"role": "assistant", "content": full_response})
+
+    return None
+
 # Run main()
 
 if __name__ == '__main__':
-    main()
+    main()
+