Update app.py

app.py

@@ -18,12 +18,17 @@ You can create a requirements.txt with:
 Deploy this file (app.py) along with requirements.txt to Hugging Face Spaces.
 """
 
-# Dependency checks
+# Dependency checks and initial imports
 try:
     import streamlit as st
 except ImportError:
     raise ImportError("Missing dependency: streamlit. Install it with 'pip install streamlit'.")
 
+# Immediately set the page configuration
+st.set_page_config(page_title="AI Friend Therapist Assistant", layout="centered")
+
+import os
+
 try:
     from transformers import pipeline
 except ImportError:
@@ -34,15 +39,10 @@ try:
 except ImportError:
     raise ImportError("Missing dependency: torch. Install it with 'pip install torch'.")
 
-import os
-
-# Optional: display a note if running on Hugging Face Spaces (the SPACE_ID env var is set in Spaces)
+# Optional: display a note if running on Hugging Face Spaces
 if os.environ.get("SPACE_ID"):
     st.info("Running on Hugging Face Spaces.")
 
-# Configure the Streamlit page
-st.set_page_config(page_title="AI Friend Therapist Assistant", layout="centered")
-
 # Load the Hugging Face summarization model (cached to avoid reloading)
 @st.cache(allow_output_mutation=True)
 def load_summarizer():
@@ -64,7 +64,7 @@ st.write(
 # Text area for user input
 user_input = st.text_area("Your message", height=150)
 
-# When the user clicks "Send", process the input
+# Process user input when "Send" is clicked
 if st.button("Send"):
     if user_input.strip():
         # Append the message to conversation history