Update app.py

app.py

@@ -291,7 +291,6 @@ class BLIPVQA:
 class ConversationBot:
     def __init__(self):
         print("Initializing VisualChatGPT")
-        self.llm = OpenAI(temperature=0)
         self.edit = ImageEditing(device="cuda")
         self.i2t = ImageCaptioning(device="cuda")
         self.t2i = T2I(device="cuda")
@@ -369,6 +368,9 @@ class ConversationBot:
                  #description="useful when you want to generate a new real image from both the user desciption and a human pose image. like: generate a real image of a human from this human pose image, or generate a new real image of a human from this pose. "
                              #"The input to this tool should be a comma seperated string of two, representing the image_path and the user description")]
         ]
+
+    def init_langchain(api_key):
+        self.llm = OpenAI(temperature = 0, openai_api_key = api_key)
         self.agent = initialize_agent(
             self.tools,
             self.llm,
@@ -377,6 +379,7 @@ class ConversationBot:
             memory=self.memory,
             return_intermediate_steps=True,
             agent_kwargs={'prefix': VISUAL_CHATGPT_PREFIX, 'format_instructions': VISUAL_CHATGPT_FORMAT_INSTRUCTIONS, 'suffix': VISUAL_CHATGPT_SUFFIX}, )
+        return gr.update(visible = True)
 
     def run_text(self, text, state):
         print("===============Running run_text =============")
@@ -417,12 +420,12 @@ class ConversationBot:
 
 bot = ConversationBot()
 with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as demo:
-
+    openai_api_key_input = gr.Textbox(type = "Password", label = "Enter your OpenAI API key here")
     chatbot = gr.Chatbot(elem_id="chatbot", label="Visual ChatGPT")
     state = gr.State([])
     
     
-    with gr.Row(visible = True) as input_row:
+    with gr.Row(visible = False) as input_row:
         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.15, min_width=0):
@@ -430,6 +433,7 @@ with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as demo:
         with gr.Column(scale=0.15, min_width=0):
             btn = gr.UploadButton("Upload", file_types=["image"])
 
+    openai_api_key_input.submit(bot.init_langchain,openai_api_key_input,[input_row])
     txt.submit(bot.run_text, [txt, state], [chatbot, state])
     txt.submit(lambda: "", None, txt)
     btn.upload(bot.run_image, [btn, state, txt], [chatbot, state, txt])