Flexible function for different prompts to ensure group categories

app.py

@@ -70,16 +70,30 @@ 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):
+# def custom_generate_response(cust_user_message, prompt_index):
+#     """
+#     Generates a custom response based on the user message and the selected prompt,
+#     including a custom ending specific to the prompt.
+
+#     Parameters:
+#     - cust_user_message: The message input from the user.
+#     - prompt_index: The index of the custom prompt to use.
+#     """
+#     prompt, ending = CustomPrompts[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)
+            
+def custom_generate_response(cust_user_message, prompt_index, prompts_list):
     """
-    Generates a custom response based on the user message and the selected prompt,
-    including a custom ending specific to the prompt.
+    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.
 
     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.
     """
-    prompt, ending = CustomPrompts[prompt_index]  # Unpack the prompt and its ending
+    prompt, ending = prompts_list[prompt_index]  # Unpack the prompt and its ending from the provided list
     cust_user_message = f"{prompt}\n\n{cust_user_message}\n\n{ending}"
     yield from generate_response(cust_user_message)
 
@@ -123,18 +137,32 @@ with gr.Blocks() as iface:
     MainOutput = gr.TextArea(placeholder='Output will show here')
     CustomButtonInput = gr.TextArea(lines=1, placeholder='Prompt goes here')
 
+    # with gr.Accordion("Random Ideas"):
+    #     with gr.Group():        
+    #         # Dynamically create buttons and assign actions
+    #         for index, (prompt, _) in enumerate(CustomPrompts):
+    #             button = gr.Button(prompt)
+    #             button.click(custom_generate_response, inputs=[CustomButtonInput, gr.State(index)], outputs=MainOutput)
+
+    # with gr.Accordion("General Product based", open=False):
+    #     with gr.Group():
+    #         # Dynamically create buttons and assign actions
+    #         for index, (prompt, _) in enumerate(BusinessPrompts):
+    #             button = gr.Button(prompt)
+    #             button.click(custom_generate_response, inputs=[CustomButtonInput, gr.State(index)], outputs=MainOutput)
+
     with gr.Accordion("Random Ideas"):
         with gr.Group():        
-            # Dynamically create buttons and assign actions
             for index, (prompt, _) in enumerate(CustomPrompts):
                 button = gr.Button(prompt)
-                button.click(custom_generate_response, inputs=[CustomButtonInput, gr.State(index)], outputs=MainOutput)
+                # Pass CustomPrompts list as an argument
+                button.click(custom_generate_response, inputs=[CustomButtonInput, gr.State(index), gr.State(CustomPrompts)], outputs=MainOutput)
 
     with gr.Accordion("General Product based", open=False):
         with gr.Group():
-            # Dynamically create buttons and assign actions
             for index, (prompt, _) in enumerate(BusinessPrompts):
                 button = gr.Button(prompt)
-                button.click(custom_generate_response, inputs=[CustomButtonInput, gr.State(index)], outputs=MainOutput)
+                # Pass BusinessPrompts list as an argument
+                button.click(custom_generate_response, inputs=[CustomButtonInput, gr.State(index), gr.State(BusinessPrompts)], outputs=MainOutput)
 
 iface.queue().launch(server_name="0.0.0.0", share=True)