Update app.py

app.py

@@ -10,6 +10,7 @@ client = AI71(ai71_api_key)
 # Set page config with title and favicon
 st.set_page_config(
     page_title="AI Lawyer 🧑‍⚖️",
+    page_icon="assets/lawyer_icon.png",  # Replace with your favicon path
 )
 
 # Add custom CSS for styling
@@ -66,10 +67,12 @@ st.write(
     "This is your AI Lawyer that uses the AI71 model to provide legal advice and information."
 )
 
-# Create a session state variable to store the chat messages. This ensures that the
-# messages persist across reruns.
+# Initialize session state variables if not already set
 if "messages" not in st.session_state:
     st.session_state.messages = []
+    st.session_state.max_tokens = 512
+    st.session_state.temperature = 0.7
+    st.session_state.top_p = 0.95
     instruction = "Hi! This is your AI Lawyer 🧑‍⚖️. Please describe your legal question or issue. For example: 'I need help understanding a contract clause.'"
     st.session_state.messages.append({"role": "assistant", "content": instruction})