Test lambda to use hardcoded variables

app.py

@@ -46,7 +46,7 @@ def generate_response(user_message): #generate_response_token_by_token
         if token_buffer:
             elapsed_time = time.time() - start_time  # Calculate elapsed time
             alltokens += token_buffer
-            yield f"{alltokens} \n\n [Inference time: {elapsed_time:.2f} seconds | Average Tokens per second: { tokencount / elapsed_time }]"
+            yield f"{alltokens} \n\n [Inference time: {elapsed_time:.2f} seconds | Average Tokens per second: { round(tokencount / elapsed_time, 2) }]"
 
     finally:
         try:
@@ -65,8 +65,11 @@ def generate_response(user_message): #generate_response_token_by_token
             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_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)
 
@@ -101,7 +104,7 @@ with gr.Blocks() as iface:
         CustomButtonClassDiagram = gr.Button(CustomPrompts[0])
         CustomButtonPydotcode = gr.Button(CustomPrompts[1])
         CustomButtonHappyMovieScene = gr.Button(CustomPrompts[2])
-        CustomButtonClassDiagram.click(custom_generate_response, inputs=[CustomButtonInput], outputs=MainOutput)
+        CustomButtonClassDiagram.click(lambda x: custom_generate_response(0, x), inputs=[CustomButtonInput], outputs=MainOutput) #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)