Get it to load the API key from environment variables

.gitignore

@@ -1,2 +1,4 @@
 /venv
-/__pycache__
+/__pycache__
+
+start.sh

app.py

@@ -50,6 +50,9 @@ from langchain.llms.openai import OpenAI
 import re
 import gradio as gr
 
+import os
+
+OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
 
 def cut_dialogue_history(history_memory, keep_last_n_words=400):
     if history_memory is None or len(history_memory) == 0:
@@ -129,6 +132,8 @@ class ConversationBot:
             return_intermediate_steps=True,
             agent_kwargs={'prefix': VISUAL_CHATGPT_PREFIX, 'format_instructions': VISUAL_CHATGPT_FORMAT_INSTRUCTIONS, 'suffix': VISUAL_CHATGPT_SUFFIX}, )
 
+        print("Agent initialized.")
+
         return gr.update(visible = True)
 
 bot = ConversationBot({'Text2Image': 'cuda:0',
@@ -142,6 +147,10 @@ bot = ConversationBot({'Text2Image': 'cuda:0',
                     #    'DepthText2Image': 'cuda:0',
                        })
 
+if OPENAI_API_KEY:
+    print("OPENAI_API_KEY found in environment variables. Starting agent.")
+    bot.init_agent(OPENAI_API_KEY)
+
 with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
     gr.Markdown("<h3><center>Visual ChatGPT</center></h3>")
     gr.Markdown(
@@ -151,18 +160,19 @@ with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
         """
     )
 
-    with gr.Row():
-        openai_api_key_textbox = gr.Textbox(
-            placeholder="Paste your OpenAI API key here to start Visual ChatGPT(sk-...) and press Enter ↵️",
-            show_label=False,
-            lines=1,
-            type="password",
-        )
+    if not OPENAI_API_KEY:
+        with gr.Row():
+            openai_api_key_textbox = gr.Textbox(
+                placeholder="Paste your OpenAI API key here to start Visual ChatGPT(sk-...) and press Enter ↵️",
+                show_label=False,
+                lines=1,
+                type="password",
+            )
 
     chatbot = gr.Chatbot(elem_id="chatbot", label="Visual ChatGPT")
     state = gr.State([])
 
-    with gr.Row(visible=False) as input_raws:
+    with gr.Row(visible=bot.agent) as input_raws:
         with gr.Column(scale=0.7):
             txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter, or upload an image").style(container=False)
         with gr.Column(scale=0.10, min_width=0):
@@ -193,7 +203,9 @@ with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
 
 
 
-    openai_api_key_textbox.submit(bot.init_agent, [openai_api_key_textbox], [input_raws])
+    if not OPENAI_API_KEY:
+        openai_api_key_textbox.submit(bot.init_agent, [openai_api_key_textbox], [input_raws])
+    
     txt.submit(bot.run_text, [txt, state], [chatbot, state])
     txt.submit(lambda: "", None, txt)
     run.click(bot.run_text, [txt, state], [chatbot, state])