Update app.py

app.py

@@ -30,16 +30,25 @@ def ask_ai():
         response = index.query(query, response_mode="compact")
         display(Markdown(f"Response: <b>{response.response}</b>"))
 '''
-def ask_ai(question):
+def ask_ai(query, api_key):
+    # Set default API key if user does not enter a key
+    if api_key == "":
+        api_key = "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"
+    
+    os.environ["OPENAI_API_KEY"] = api_key
+    
     index = GPTSimpleVectorIndex.load_from_disk('index.json')
-    response = index.query(question, response_mode="compact")
+    response = index.query(query, response_mode="compact")
     return response.response
 
-#input your api key
-os.environ["OPENAI_API_KEY"] = "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"
 
 train_with_data("data")
 
-inputs=gr.inputs.Textbox(label="Input your question here"), 
-iface = gr.Interface(fn=ask_ai, inputs="text", outputs="text" ,title="Jim's Chatbot")
+# Create Gradio interface to prompt for API key
+api_key = gr.inputs.Textbox(label="Enter your OpenAI API key (leave blank to use default key):")
+
+# Define the interface
+iface = gr.Interface(fn=ask_ai, inputs=["text", api_key], outputs="text" ,title="Jim's Chatbot")
+
+# Start the interface
 iface.launch()