app.py

app.py

@@ -38,10 +38,13 @@ if not st.session_state["authenticated"]:
 st.title("Schlager ContractAi")
 st.caption("Chat with your contract or manage meeting minutes")
 
-# Load API Key from Environment Variable
+# Load API Key and Assistant IDs from Environment Variables
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-if not OPENAI_API_KEY:
-    st.error("Missing OpenAI API key. Please set it as an environment variable.")
+ASSISTANT_CONTRACT_ID = os.getenv("ASSISTANT_CONTRACT_ID")
+ASSISTANT_TECHNICAL_ID = os.getenv("ASSISTANT_TECHNICAL_ID")
+
+if not OPENAI_API_KEY or not ASSISTANT_CONTRACT_ID or not ASSISTANT_TECHNICAL_ID:
+    st.error("Missing required environment variables. Please set OPENAI_API_KEY, ASSISTANT_CONTRACT_ID, and ASSISTANT_TECHNICAL_ID.")
     st.stop()
 
 # Tabs
@@ -135,10 +138,6 @@ def contract_chat_section(tab, assistant_id, session_key, input_key):
             except Exception as e:
                 st.error(f"Error: {str(e)}")
 
-# Assign Assistant IDs
-ASSISTANT_CONTRACT_ID = "asst_rd9h8PfYuOmHbkvOF3RTmVfn"
-ASSISTANT_TECHNICAL_ID = "asst_xizNZBCJuy4TqdjqjwkxbAki"
-
 contract_chat_section(tab1, ASSISTANT_CONTRACT_ID, "contract_messages", "contract_input")
 contract_chat_section(tab2, ASSISTANT_TECHNICAL_ID, "technical_messages", "technical_input")
 
@@ -151,3 +150,5 @@ with tab3:
         st.download_button("Download Flagged Responses", data=flagged_responses, file_name="flagged_responses.json", mime="application/json")
     else:
         st.info("No flagged responses available.")
+
+