Update app.py

app.py

@@ -6,6 +6,8 @@ import sys
 import os
 from IPython.display import Markdown, display
 
+# Define API key globally
+api_key = "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"
 
 def construct_index(directory_path):
     # set maximum input size
@@ -29,34 +31,20 @@ def construct_index(directory_path):
 
     return index
 
-
-
-
-'''
-def ask_ai():
-    index = GPTSimpleVectorIndex.load_from_disk('index.json')
-    while True: 
-        query = input("What do you want to ask? ")
-        response = index.query(query, response_mode="compact")
-        display(Markdown(f"Response: <b>{response.response}</b>"))
-'''
-# Define the ask_ai() function
-def ask_ai(question,api):
-    if api == "":
-      api = "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"
-    os.environ["OPENAI_API_KEY"] = api
+def ask_ai(question, api_key):
+    # Use global API key variable
+    os.environ["OPENAI_API_KEY"] = api_key
     index = GPTSimpleVectorIndex.load_from_disk('index.json')
     response = index.query(question, response_mode="compact")
     return response.response
 
-
 construct_index("data")
 
 # Create Gradio interface to prompt for API key
-api_key = gr.inputs.Textbox(label="Enter your OpenAI API key:")
+api_key_input = gr.inputs.Textbox(label="Enter your OpenAI API key:")
 
 # Define the interface
-iface = gr.Interface(fn=ask_ai, inputs=["text", api_key], outputs="text" ,title="Jim's Chatbot")
+iface = gr.Interface(fn=ask_ai, inputs=["text", api_key_input], outputs="text" ,title="Jim's Chatbot")
 
 # Start the interface
-iface.launch()
+iface.launch()