Update app.py

app.py

@@ -8,6 +8,13 @@ from langchain_core.messages import AIMessage, HumanMessage
 from langchain_core.prompts import PromptTemplate
 from langchain_community.callbacks import get_openai_callback
 
+# Set page configuration at the very beginning
+st.set_page_config(
+    page_title="PostgreSQL Query Assistant", 
+    page_icon="🤖", 
+    layout="wide"
+)
+
 # Custom CSS for enhanced styling
 def local_css():
     st.markdown("""
@@ -110,13 +117,6 @@ def main():
     # Apply custom CSS
     local_css()
 
-    # Set page configuration
-    st.set_page_config(
-        page_title="PostgreSQL Query Assistant", 
-        page_icon="🤖", 
-        layout="wide"
-    )
-
     # Main container
     with st.container():
         # Header