Update app.py

app.py

@@ -1,10 +1,12 @@
 import streamlit as st
 from ai71 import AI71
+import os
 
 # Initialize AI71 client
-AI71_API_KEY = "ai71-api-d95427d8-ce80-42cb-8117-cd12ecfe7907"  # Your API key
+AI71_API_KEY = os.getenv("AI71_API_KEY")  # Access the secret from environment variables
 client = AI71(AI71_API_KEY)
 
+# Define your functions and Streamlit UI
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -14,7 +16,6 @@ def respond(
     top_p,
 ):
     messages = [{"role": "system", "content": system_message}]
-
     for val in history:
         if val[0]:
             messages.append({"role": "user", "content": val[0]})
@@ -95,4 +96,4 @@ if 'history' not in st.session_state:
 # Display chat history above the message input
 st.text_area("Chat History", value="\n\n".join([f"User: {h[0]}\n\nAI Lawyer: {h[1]}" for h in st.session_state.history]), height=400, key="chat_history")
 
-message = st.text_input("Your message", key="new_message", on_change=send_message)
+message = st.text_input("Your message", key="new_message", on_change=send_message)