Update app.py

app.py

@@ -47,9 +47,18 @@ def oai_key_isvalid(api_key):
     except:
         return False
     
-load_dotenv()
+
+def initialize_session_state():
+    """Initialize session state variables"""
+    if 'prompt' not in st.session_state:
+        st.session_state.prompt = None
+    # Add other session state initializations here as needed
+
+# Call initialization function
+initialize_session_state()
+
+# Now you can safely use session state
 ss = st.session_state
-ss.prompt = None
  
 # Set width of sidebar
 st.markdown(