app.py

app.py

@@ -11,7 +11,7 @@ from pydub import AudioSegment
 from io import BytesIO
 
 # Streamlit Page Config
-st.set_page_config(page_title="Schlager ContractAi")
+st.set_page_config(page_title="Schlager ContractAi", layout="wide")
 
 # Authentication
 AUTHORIZED_USERS = {
@@ -43,9 +43,11 @@ if not st.session_state["authenticated"]:
 st.title("Schlager ContractAi")
 st.caption("Chat with your contract or manage meeting minutes")
 
-# Sidebar for API Key input
-with st.sidebar:
-    OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies Access Key", type="password")
+# Load API Key from Environment Variable
+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.")
+    st.stop()
 
 # Tabs
 tab1, tab2, tab3, tab4 = st.tabs(["Contract", "Technical", "Minutes", "Document Preparation"])
@@ -111,12 +113,8 @@ with tab4:
 def contract_chat_section(tab, assistant_id, session_key, input_key):
     with tab:
         st.subheader("Chat")
-        
-        if OPENAI_API_KEY:
-            client = OpenAI(api_key=OPENAI_API_KEY)
-        else:
-            st.error("Please enter your C2 Group of Technologies Access Key to continue.")
-            st.stop()
+
+        client = OpenAI(api_key=OPENAI_API_KEY)
 
         if session_key not in st.session_state:
             st.session_state[session_key] = []
@@ -164,4 +162,4 @@ 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")
+contract_chat_section(tab2, ASSISTANT_TECHNICAL_ID, "technical_messages", "technical_input")