Better langchain handling

app.py

@@ -1,12 +1,12 @@
 import streamlit
-
-
 import streamlit as st
-from langchain.llms import OpenAI
+from langchain_community.llms import OpenAI
 
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 
+import os
+
 LLM_GPT35 = 'GPT3.5'
 LLM_GPT4 = 'GPT4'
 
@@ -22,7 +22,9 @@ with st.sidebar:
 def get_openai_api_key():
 
     if openai_api_key_sb == None or len(openai_api_key_sb)==0:
-        openai_api_key = str(open('../oaik','r').readline())
+        openai_api_key = os.environ.get('OPENAI_API_KEY')
+        if openai_api_key==None:
+            openai_api_key = str(open('../oaik','r').readline())
     else:
         openai_api_key = openai_api_key_sb
 

requirements.txt

@@ -1,2 +1,4 @@
+langchain-community
 langchain
-openai
+openai
+langchain_openai