Update app.py

app.py

@@ -243,7 +243,7 @@ class canny2image:
         # need to explicitly call pipe.to("cuda").
         self.pipe.enable_model_cpu_offload()
         
-        #self.pipe.enable_xformers_memory_efficient_attention()
+        self.pipe.enable_xformers_memory_efficient_attention()
         
         # Generator seed,
         self.generator = torch.manual_seed(0)
@@ -289,9 +289,9 @@ class BLIPVQA:
         return answer
 
 class ConversationBot:
-    def __init__(self):
+    def __init__(self, api_key):
         print("Initializing VisualChatGPT")
-        self.llm = OpenAI(temperature=0)
+        self.llm = OpenAI(temperature=0, openai_api_key = api_key)
         self.edit = ImageEditing(device="cuda")
         self.i2t = ImageCaptioning(device="cuda")
         self.t2i = T2I(device="cuda")
@@ -415,11 +415,21 @@ class ConversationBot:
         return state, state, txt + ' ' + image_filename + ' '
 
 
-bot = ConversationBot()
+def init_conversation_bot(api_key):
+   bot = ConversationBot(api_key)
+   return bot,gr.update(visible = True)
+    
+
+#bot = ConversationBot()
 with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as demo:
+    openai_api_key = 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():
+    bot = gr.State([])
+    
+    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):
@@ -427,6 +437,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.submit(init_conversation_bot,[openai_api_key],[bot, 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])