custom generate unpack issue

app.py

@@ -80,7 +80,7 @@ def generate_response(user_message): #generate_response_token_by_token
             error_message = process.stderr.read()
             print(f"Error: {error_message}")
             
-def custom_generate_response(cust_user_message, prompt_index, prompts_list):
+def custom_generate_response(cust_user_message, prompt_index, category):
     """
     Generates a custom response based on the user message, the selected prompt,
     and the provided list of prompts, including a custom ending specific to the prompt.
@@ -88,9 +88,10 @@ def custom_generate_response(cust_user_message, prompt_index, prompts_list):
     Parameters:
     - cust_user_message: The message input from the user.
     - prompt_index: The index of the custom prompt to use.
-    - prompts_list: The list of prompts to use for generating the response.
+    - category: The key of list of prompts to use for generating the response.
     """
-    prompt, ending = prompts_list[prompt_index]  # Unpack the prompt and its ending from the provided list
+    prompts_list = Allprompts[category]  # Retrieve the correct list of prompts based on the category
+    prompt, ending = prompts_list[prompt_index]  # Unpack the prompt and its ending
     cust_user_message = f"{prompt}\n\n{cust_user_message}\n\n{ending}"
     yield from generate_response(cust_user_message)