Test lambda to use hardcoded variables didnt work

app.py

@@ -64,11 +64,7 @@ def generate_response(user_message): #generate_response_token_by_token
         if process.returncode != 0:
             error_message = process.stderr.read()
             print(f"Error: {error_message}")
-
-def custom_generate_response(prompt_index, cust_user_message):
-    cust_user_message = CustomPrompts[prompt_index] + '\n\n' + cust_user_message
-    yield from generate_response(cust_user_message)
-
+            
 def custom_generate_response0(cust_user_message):
     cust_user_message = CustomPrompts[0] + '\n\n' + cust_user_message + '\n\nClass Diagram:'
     yield from generate_response(cust_user_message)
@@ -104,7 +100,7 @@ with gr.Blocks() as iface:
         CustomButtonClassDiagram = gr.Button(CustomPrompts[0])
         CustomButtonPydotcode = gr.Button(CustomPrompts[1])
         CustomButtonHappyMovieScene = gr.Button(CustomPrompts[2])
-        CustomButtonClassDiagram.click(lambda x: custom_generate_response(0, x), inputs=[CustomButtonInput], outputs=MainOutput) #custom_generate_response0, inputs=[CustomButtonInput], outputs=MainOutput)
+        CustomButtonClassDiagram.click(custom_generate_response0, inputs=[CustomButtonInput], outputs=MainOutput)
         CustomButtonPydotcode.click(custom_generate_response1, inputs=[CustomButtonInput], outputs=MainOutput)
         CustomButtonHappyMovieScene.click(custom_generate_response2, inputs=[CustomButtonInput], outputs=MainOutput)