add custom prompts

app.py

@@ -22,7 +22,8 @@ with gr.Blocks() as main_app:
 
         with gr.Column():
             with gr.Row():
-                options_audio = gr.Radio(["XTTS", "Elevenlabs"], label="Audio Generation")
+                options_audio = gr.Radio(["XTTS", "Elevenlabs"], value="Elevenlabs", label="Audio Generation")
+                options_prompt = gr.Radio(["Default", "Custom"], value="Default", label="Prompts")
                 output_audio = gr.Audio(interactive=False, label='Audio', autoplay=False)
 
         messages = gr.State([])
@@ -34,7 +35,44 @@ with gr.Blocks() as main_app:
             with gr.Row():
                 button_text = gr.Button(value='Submit text')
                 clear_button = gr.ClearButton([chat, messages])
-        
+    
+    with gr.Tab('Prompts'):
+        general_prompt = gr.Text(
+            placeholder='Ingrese el prompt general del bot', label='General prompt'
+        )
+        standalone_prompt = gr.Text(
+            placeholder='Ingrese el prompt usado para encontrar el contexto', label='Standalone prompt'
+        )
+        _ = gr.Markdown(
+            "```\n"
+            "Recuerde dejar estos formatos en los prompts: \n"
+            "----------------------- General --------------------------\n"
+            "=========\n"
+            "Contexto:\n"
+            "CONTEXTO\n"
+            "=========\n"
+            "\n"
+            "----------------------- Standalone -----------------------\n"
+            "You are a standalone question-maker. Given the following chat history and follow-up message, rephrase "
+            "the follow-up phrase to be a standalone question (sometimes the follow-up is not a question, so create "
+            "a standalone phrase), in spanish. In the standalone message you must include all the information at the "
+            "moment that is known about the customer, all the important nouns and what they are looking for. In cases "
+            "where you think is usefully, include what is the best recommendation for the customer. To give you "
+            "context, the conversation is about (INGRESE INFORMACIÓN DE LA MARCA, EL NOMBRE Y DE MANERA MUY GENERAL "
+            "QUE ES LO QUE VENDE).\n"
+            "There might be moments when there isn't a question in those cases return a standalone phrase: for example "
+            "if the user says 'hola' (or something similar) then the output would be 'el usuario está saludando', or "
+            "if the user says 'gracias' or 'es muy util' (or something similar) then the output would be a phrase "
+            "showing that the user is grateful and what they are grateful for, or if the user say 'si' then it would "
+            "be a phrase encapsulating the relationship to its previous question or phrase.\n"
+            "Your response cannot be more than 100 words.\n"
+            "Chat History:\n"
+            "\n"
+            "HISTORY\n"
+            "Follow-up message: QUESTION\n"
+            "Standalone message:\n", line_breaks=True
+        )
+
     with gr.Tab('Times'):
 
         columns = ["User Message", "Chatbot Response", "Standalone Question", "Create Embedding", "Query Pinecone",
@@ -49,7 +87,7 @@ with gr.Blocks() as main_app:
 
     text.submit(
         fn=get_answer,
-        inputs=[text, chat, messages, output_audio, output_video, table_times, options_audio],
+        inputs=[text, chat, messages, output_audio, output_video, table_times, options_audio, options_prompt, general_prompt, standalone_prompt],
         outputs=[chat, output_audio, output_video, table_times]
     ).then(
         lambda: None, None, [text]
@@ -61,7 +99,7 @@ with gr.Blocks() as main_app:
 
     button_text.click(
         fn=get_answer,
-        inputs=[text, chat, messages, output_audio, output_video, table_times, options_audio],
+        inputs=[text, chat, messages, output_audio, output_video, table_times, options_audio, options_prompt, general_prompt, standalone_prompt],
         outputs=[chat, output_audio, output_video, table_times]
     ).then(
         lambda: None, None, [text]

backend_functions.py

@@ -96,9 +96,12 @@ def _query_pinecone(embedding):
     return final_results
 
 
-def _general_prompt(context):
-    with open("prompt_general.txt", "r") as file:
-        file_prompt = file.read().replace("\n", "")
+def _general_prompt(context, option_prompt, general_prompt):
+    if option_prompt == "Default":
+        with open("prompt_general.txt", "r") as file:
+            file_prompt = file.read().replace("\n", "")
+    elif option_prompt == "Custom":
+        file_prompt = general_prompt
     
     context_prompt = file_prompt.replace('CONTEXT', context)
     print(context_prompt)
@@ -141,9 +144,12 @@ def _call_gpt_standalone(prompt: str):
     return response.choices[0].message.content
 
 
-def _get_standalone_question(question, history_messages):
-    with open("prompt_standalone_message.txt", "r") as file:
-        file_prompt_standalone = file.read().replace("\n", "")
+def _get_standalone_question(question, history_messages, option_prompt, standalone_prompt):
+    if option_prompt == "Default":
+        with open("prompt_standalone_message.txt", "r") as file:
+            file_prompt_standalone = file.read().replace("\n", "")
+    elif option_prompt == "Custom":
+        file_prompt_standalone = standalone_prompt
 
     history = ''
     for i, msg in enumerate(history_messages):
@@ -320,7 +326,7 @@ def _create_video(link_audio: str, unique_id: str):
     return signed_url_video
 
 
-def get_answer(question: str, chatbot: list[tuple[str, str]], history_messages, comp_audio, comp_video, df_table, option_audio):
+def get_answer(question: str, chatbot: list[tuple[str, str]], history_messages, comp_audio, comp_video, df_table, option_audio, option_prompt, general_prompt, standalone_prompt):
     """
     Gets the answer of the chatbot
     """
@@ -329,7 +335,7 @@ def get_answer(question: str, chatbot: list[tuple[str, str]], history_messages,
         message_output = 'Un placer haberte ayudado, hasta luego!'
     else:
         start_get_standalone_question = time.time()
-        standalone_msg_q = _get_standalone_question(question, history_messages) # create standalone question or message
+        standalone_msg_q = _get_standalone_question(question, history_messages, option_prompt, standalone_prompt) # create standalone question or message
         end_get_standalone_question = time.time()
         time_get_standalone_question = end_get_standalone_question - start_get_standalone_question
 
@@ -344,7 +350,7 @@ def get_answer(question: str, chatbot: list[tuple[str, str]], history_messages,
         time_query_pinecone = end_query_pinecone - start_query_pinecone
 
         start_general_prompt = time.time()
-        final_context_prompt = _general_prompt(best_results) # create context/general prompt
+        final_context_prompt = _general_prompt(best_results, option_prompt, general_prompt) # create context/general prompt
         end_general_prompt = time.time()
         time_general_prompt = end_general_prompt - start_general_prompt