Update app.py

app.py

@@ -269,20 +269,20 @@ def infer(image_input):
     print(clipi_result)
 
     llama_q = clipi_result
-    yield None, None
+    yield None, None, None
     
     gr.Info('Calling Llama2 ...')
     result = llama_gen_fragrance(llama_q)
 
     print(f"Llama2 result: {result}")
-    yield result, None
+    yield result, None, None
 
     parsed = parse_perfume_description(result)
-    yield result, parsed
-
-def generate_flacon(parsed, progress=gr.Progress(track_tqdm=True)):
     image_desc = extract_field(parsed, "Image Description")
+    yield result, parsed, image_desc
 
+def generate_flacon(parsed, progress=gr.Progress(track_tqdm=True)):
+    
     print(image_desc)
     gr.Info("Generating a nice marketing image with FLUX...")
     gen_bottle = infer_flux(image_desc)
@@ -305,14 +305,15 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 image_in = gr.Image(label="Image input", type="filepath", elem_id="image-in")
                 submit_btn = gr.Button('Give me a Fragrance')
-                json_res = gr.JSON(label="JSON (for further usage)", interactive=False)
+                json_res = gr.JSON(label="JSON (for further usage)")
+                flacon_desc = gr.Textbox(interactive=False, visible=False)
             with gr.Column():
                 #caption = gr.Textbox(label="Generated Caption")
                 fragrance = gr.Textbox(label="generated Fragrance", elem_id="fragrance")
                 get_flacon_btn = gr.Button("Generate Flacon image")
                 bottle_res = gr.Image(label="Flacon")
         
-    submit_btn.click(fn=infer, inputs=[image_in], outputs=[fragrance, json_res])
-    get_flacon_btn.click(fn=generate_flacon, inputs=[json_res], outputs=[bottle_res])
+    submit_btn.click(fn=infer, inputs=[image_in], outputs=[fragrance, json_res, flacon_desc])
+    get_flacon_btn.click(fn=generate_flacon, inputs=[flacon_desc], outputs=[bottle_res])
 
 demo.queue(max_size=12).launch(ssr_mode=False, mcp_server=True)