Update app.py

app.py

@@ -1,22 +1,36 @@
 import openai
+import json
 import gradio as gr
 import os
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# retrieve secret key from HF settings
+#openai.api_key = os.getenv("OPENAI_API_KEY")
 
-messages = [
-    {"role": "system", "content": "You are a learning assistant specialized in business information systems."},
-]
+# rebuild storage context and load knowledge index
+from llama_index import StorageContext, load_index_from_storage
+storage_context = StorageContext.from_defaults(persist_dir='C:/Users/Limin/openai/index')
+index = load_index_from_storage(storage_context)
 
-def chatbot(input):
+class Chatbot:
+    def __init__(self, api_key, index):
+        self.index = index
+        openai.api_key = api_key
+ 
+    def generate_response(self, user_input):
+        query_engine = index.as_query_engine()
+        response = query_engine.query(user_input)
+
+        message = {"role": "assistant", "content": response.response}
+        return message
+
+def create_bot(input):
+    bot = Chatbot(os.getenv("OPENAI_API_KEY"), index=index)
+  
     if input:
-        messages.append({"role": "user", "content": input})
-        chat = openai.chat.completions.create(
-            model="gpt-3.5-turbo", messages=messages
-        )
-        reply = chat.choices[0].message.content
-        messages.append({"role": "assistant", "content": reply})
-        return reply
+        response = bot.generate_response(input)
+        output = response['content']
+
+        return output
 
 inputs = gr.Textbox(lines=7, label="Ask questions related to the course.")
 outputs = gr.Textbox(label="Reply")