Update app.py

app.py

@@ -1,14 +1,20 @@
 import streamlit as st
 from llama_index import VectorStoreIndex, ServiceContext, Document
 from llama_index.llms import OpenAI
+from langchain.llms import OpenAI
 import openai
 from llama_index import SimpleDirectoryReader
 
 st.set_page_config(page_title="HUD Audit Guide", page_icon="🍂", layout="centered", initial_sidebar_state="auto", menu_items=None)
-openai.api_key = os.environ["OpenAI-API-Key"]
 st.title("Ask the HUD Audit Guide 💬🤖")
 st.info("Check out more info on the complete HUD Audit Guide at the official [website](https://www.hudoig.gov/library/single-audit-guidance/hud-consolidated-audit-guide)", icon="📃")
-         
+
+openai_api_key = st.sidebar.text_input('OpenAI API Key', type='password')
+
+def generate_response(input_text):
+    llm = OpenAI(temperature=0.7, openai_api_key=openai_api_key)
+    st.info(llm(input_text))
+
 if "messages" not in st.session_state.keys(): # Initialize the chat messages history
     st.session_state.messages = [
         {"role": "assistant", "content": "Ask me a question about the HUD Audit Guide - Chapter 6 - Ginnie Mae Issuers of Mortgage-Backed Securities Audit Guidance!"}
@@ -28,7 +34,10 @@ index = load_data()
 chat_engine = index.as_chat_engine(chat_mode="condense_question", verbose=True)
 
 if prompt := st.chat_input("Your question"): # Prompt for user input and save to chat history
-    st.session_state.messages.append({"role": "user", "content": prompt})
+    if not openai_api_key.startswith('sk-'):
+        st.warning('Please enter your OpenAI API key!', icon='⚠')
+    else: # Only allow the user to proceed if it appears they've entered a valid OpenAI API key
+        st.session_state.messages.append({"role": "user", "content": prompt})
 
 for message in st.session_state.messages: # Display the prior chat messages
     with st.chat_message(message["role"]):