Update app.py

app.py

@@ -270,7 +270,7 @@ def infer(image_input):
 
     gr.Info('Calling CLIP Interrogator ...')
 
-    yield None, None, None, gr.update()
+    yield None, None, None
     clipi_result = clipi_client.predict(
 		image=handle_file(image_input),
 		mode="best",
@@ -285,11 +285,11 @@ def infer(image_input):
     result = llama_gen_fragrance(llama_q)
 
     print(f"Llama2 result: {result}")
-    yield result, None, None, gr.update()
+    yield result, None, None
 
     parsed = parse_perfume_description(result)
     image_desc = extract_field(parsed, "Image Description")
-    return result, parsed, image_desc, gr.update(interactive=True)
+    return result, parsed, image_desc
 
 css="""
 #col-container {max-width: 910px; margin-left: auto; margin-right: auto;}
@@ -315,17 +315,24 @@ with gr.Blocks(css=css) as demo:
                 get_flacon_btn = gr.Button("Generate Flacon image", interactive=False)
                 bottle_res = gr.Image(label="Flacon")
 
-    def clean_before_infer():
+    def disable_flacon_button():         
         return gr.update(interactive=False)
+
+    def allow_flacon_button():         
+        return gr.update(interactive=True)
     
     submit_btn.click(
-        fn=clean_before_infer,
-        inputs = None,
+        fn=disable_flacon_button,
+        inputs = [],
         outputs = [get_flacon_btn]
     ).then(
         fn=infer, 
         inputs=[image_in], 
-        outputs=[fragrance, json_res, flacon_desc, get_flacon_btn]
+        outputs=[fragrance, json_res, flacon_desc]
+    ).then(
+        fn=allow_flacon_button,
+        inputs=[],
+        outputs=[get_flacon_btn]
     )
     get_flacon_btn.click(fn=infer_flux, inputs=[flacon_desc], outputs=[bottle_res])