Update app.py

app.py

@@ -173,7 +173,6 @@ if "chat_complete" not in st.session_state:
     st.session_state.chat_complete = False
 if "messages" not in st.session_state:
     st.session_state.messages = []
-st.session_state["openai_model"] = "gpt-4o"
 
 # Helper functions to update session state
 def complete_setup():
@@ -250,6 +249,10 @@ if st.session_state.setup_complete and not st.session_state.chat_complete:
     # Initialize OpenAI client
     client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
 
+    # Setting OpenAI model if not already initialized
+    if "openai_model" not in st.session_state:
+        st.session_state["openai_model"] = "gpt-4o"
+
     # Initializing the system prompt for the chatbot
     if not st.session_state.messages:
         st.session_state.messages = [{