Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import llama_index 
 import logging
 import sys
+import openai
 import os 
 import wikipedia
 from llama_index import VectorStoreIndex, SimpleDirectoryReader
@@ -28,7 +29,7 @@ def ui():
 
     api_key = st.text_input('Enter your OpenAI key here: ')
     if api_key is not None:
-        os.environ["OPENAI_API_KEY"] = api_key
+        openai.api_key = os.environ[api_key]
         logging.basicConfig(stream=sys.stdout, level=logging.INFO)
         logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))