fix api error

app.py

@@ -1,5 +1,6 @@
 import os
 
+import openai
 import requests
 import streamlit as st
 from streamlit_chat import message
@@ -10,17 +11,18 @@ from utils.util import *
 st.title("Welcome to RegBotBeta")
 st.header("Powered by `LlamaIndex🦙` and `OpenAI API`")
 
-if 'messages' not in st.session_state:
+if "messages" not in st.session_state:
     st.session_state.messages = []
 
 api_key = st.text_input("Enter your OpenAI API key here:", type="password")
 if api_key:
     resp = validate(api_key)
-    if ("error" in resp.json()):
+    if "error" in resp.json():
         st.info("Invalid Token! Try again.")
     else:
         st.info("Success")
         os.environ["OPENAI_API_KEY"] = api_key
+        openai.api_key = api_key
         index = create_index(bloom)
 
 st.write("---")
@@ -28,15 +30,15 @@ input_text = st.text_area("Ask your question")
 
 if input_text is not None:
     if st.button("Ask"):
-        st.session_state.messages.append(('User', input_text))
+        st.session_state.messages.append(("User", input_text))
         with st.spinner("Processing your query..."):
             bot_response = get_response(index, input_text)
         print("bot: ", bot_response)
-        st.session_state.messages.append(('Bot', bot_response))
+        st.session_state.messages.append(("Bot", bot_response))
 
 # Display previous messages
 msg_key = 0
 for sender, msg in st.session_state.messages[::-1]:
     is_user = sender == "User"
-    message(str(msg), is_user, key=str(msg_key)+f'_{sender}')
-    msg_key += 1
+    message(str(msg), is_user, key=str(msg_key) + f"_{sender}")
+    msg_key += 1

models/bloom.py

@@ -90,8 +90,8 @@ def initialize_index(index_name):
         index = load_index_from_storage(storage_context)
 
         # huggingface repo load access
-        with fs.open(file_path, "r") as file:
-            index = pickle.loads(file.readlines())
+        # with fs.open(file_path, "r") as file:
+        #     index = pickle.loads(file.readlines())
         return index
     else:
         documents = prepare_data(r"./assets/regItems.json")
@@ -102,6 +102,6 @@ def initialize_index(index_name):
         index.storage_context.persist(file_path)
 
         # huggingface repo write access
-        with fs.open(file_path, "w") as file:
-            file.write(pickle.dumps(index))
+        # with fs.open(file_path, "w") as file:
+        #     file.write(pickle.dumps(index))
         return index