Update app.py

app.py

@@ -4,12 +4,13 @@ import os
 import io
 import contextlib
 
-# Paste your OpenAI API key below.
-interpreter.api_key = os.environ.get('openai_api_key')
+# Use environ variables for saving and using OpenAI API key as below or use from a textbox in gradio UI
+# interpreter.api_key = os.environ.get('openai_api_key')
 
 interpreter.auto_run = True
 
 def chat_with_interpreter(message, history=[]):
+    interpreter.api_key = openai_api_key
     if message == 'reset':
       interpreter.reset()
     # Redirect stdout to capture the streamed output
@@ -21,11 +22,16 @@ def chat_with_interpreter(message, history=[]):
     # Return this output so Gradio's ChatInterface can display it
     return output
 
+openai_api_key = gr.Textbox(label='OpenAI API Key', intercative=True)
+
 demo = gr.ChatInterface(fn=chat_with_interpreter, 
                         title="Open-Interpreter Gradio ChatInterface",
                         description="Open Interpreter lets LLMs run code (Python, Javascript, Shell, and more) locally",
                         examples=["what is 2+2?", "Can you solve for x: 10x -65=0?", "What are top 10 headlines from BBC from last week?"],
                         clear_btn=None,
                         retry_btn=None,
-                        undo_btn=None).queue()
+                        undo_btn=None,
+                        additional_inputs=[openai_api_key],
+                        additional_inputs_accordion_name = "OpenAI API Key"
+                        ).queue()
 demo.launch(debug=True)