Yield from exists ?

app.py

@@ -89,7 +89,7 @@ with gr.Blocks() as iface:
     CustomButtonInput = gr.TextArea()
     CustomButtonClassDiagram = gr.Button(CustomPrompts[0])
     CustomButtonPydotcode = gr.Button(CustomPrompts[1])
-    CustomButtonClassDiagram .click(custom_generate_response, inputs=[CustomButtonInput, CustomPrompts[0]], outputs=MainOutput)
-    CustomButtonPydotcode.click(custom_generate_response, inputs=[CustomButtonInput, CustomPrompts[1]], outputs=MainOutput)
+    CustomButtonClassDiagram .click(custom_generate_response, inputs=[CustomButtonInput], outputs=MainOutput)
+    CustomButtonPydotcode.click(custom_generate_response, inputs=[CustomButtonInput], outputs=MainOutput)
 
 iface.launch(server_name="0.0.0.0") #share=True)