all buttons had same function

app.py

@@ -49,6 +49,14 @@ def custom_generate_response(cust_user_message):
     cust_user_message = CustomPrompts[0] + '\n\n' + cust_user_message
     yield from generate_response(cust_user_message)
 
+def custom_generate_response1(cust_user_message):
+    cust_user_message = CustomPrompts[1] + '\n\n' + cust_user_message
+    yield from generate_response(cust_user_message)
+
+def custom_generate_response2(cust_user_message):
+    cust_user_message = CustomPrompts[2] + '\n\n' + cust_user_message
+    yield from generate_response(cust_user_message)
+
 CustomPrompts = [
     "Write a Class Diagram based on the following text:",
     "Write a Pydot code based on the following text:",
@@ -73,7 +81,7 @@ with gr.Blocks() as iface:
         CustomButtonPydotcode = gr.Button(CustomPrompts[1])
         CustomButtonHappyMovieScene = gr.Button(CustomPrompts[2])
         CustomButtonClassDiagram.click(custom_generate_response, inputs=[CustomButtonInput], outputs=MainOutput)
-        CustomButtonPydotcode.click(custom_generate_response, inputs=[CustomButtonInput], outputs=MainOutput)
-        CustomButtonHappyMovieScene.click(custom_generate_response, inputs=[CustomButtonInput], outputs=MainOutput)
+        CustomButtonPydotcode.click(custom_generate_response1, inputs=[CustomButtonInput], outputs=MainOutput)
+        CustomButtonHappyMovieScene.click(custom_generate_response2, inputs=[CustomButtonInput], outputs=MainOutput)
 
 iface.queue().launch(server_name="0.0.0.0", share=True)