Update app.py

app.py

@@ -1,18 +1,9 @@
 import streamlit as st
 import requests
 import asyncio
-import nest_asyncio
-import pdfplumber
-import json
-
-# Create a new event loop for the current thread
-loop = asyncio.new_event_loop()
-asyncio.set_event_loop(loop)
-
-# Apply nest_asyncio to the new event loop
-nest_asyncio.apply(loop)
-
-# Rest of your code remains the same...
+import pdfplumber  # For PDF text extraction
+import json  # For parsing JSON response
+import os  # For environment variables
 
 # Function to split text into chunks based on token limit
 def split_text_into_chunks(text, max_tokens=4000):
@@ -249,11 +240,10 @@ if uploaded_file is not None:
             # Analyze the contract
             if st.button("Start AI Review"):
                 with st.spinner("Analyzing contract..."):
-                    # Get the API key from secrets.toml
-                    if "api" in st.secrets and "key" in st.secrets["api"]:
-                        api_key = st.secrets["api"]["key"]
-                    else:
-                        st.error("API key not found in secrets.toml. Please add it.")
+                    # Get the API key from environment variables or secrets.toml
+                    api_key = os.getenv("OPENAI_API_KEY")
+                    if not api_key:
+                        st.error("OpenAI API key not found. Please set the OPENAI_API_KEY environment variable.")
                         st.stop()
                     
                     # Analyze the contract using the LLM