Update app.py

app.py

@@ -80,9 +80,12 @@ models =[key for key in model_links.keys()]
 # Create the sidebar with the dropdown for model selection
 selected_model = st.sidebar.selectbox("Select Model", models)
 
-#Create a temperature slider
+# Create a temperature slider
 temp_values = st.sidebar.slider('Select a temperature value', 0.0, 1.0, (0.5))
 
+# Add a system prompt input
+system_prompt = st.sidebar.text_input("System Prompt")
+
 
 #Add reset button to clear conversation
 st.sidebar.button('Reset Chat', on_click=reset_conversation) #Reset button
@@ -132,17 +135,18 @@ for message in st.session_state.messages:
 
 # Accept user input
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
-
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
 
+    # Add system prompt to chat history if it exists
+    if system_prompt:
+        st.session_state.messages.append({"role": "system", "content": system_prompt})
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
-
         try:
             stream = client.chat.completions.create(
                 model=model_links[selected_model],
@@ -171,7 +175,4 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
             st.write("This was the error message:")
             st.write(e)
 
-
-
-        
     st.session_state.messages.append({"role": "assistant", "content": response})