Duplicate from mhaziqai/prompt_tester

.env

@@ -0,0 +1 @@
+OPEN_API_KEY=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

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Prompt Tester
+emoji: 📈
+colorFrom: red
+colorTo: blue
+sdk: gradio
+sdk_version: 3.29.0
+app_file: app.py
+pinned: false
+duplicated_from: mhaziqai/prompt_tester
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,107 @@
+import gradio as gr
+import os
+
+from utils import *
+from prompt_base import *
+
+
+with gr.Blocks() as aireply:
+    archetype_list = ["The Hero: Confident and courageous tone suitable for brands in the fitness, sports, and adventure industries",
+    "The Sage: Wise and knowledgeable tone suitable for brands in education, health and wellness industries",
+    "The Explorer: Curious and adventurous tone suitable for brands in the travel, technology and entertainment industries",
+    "The Creator: Imaginative and innovative tone suitable for brands in the arts, design, and creative industries"]
+    options = ['Select Response']
+    gr.Markdown(
+        """
+    <h1 style="text-align: center;">TrueNation - Prompt Tester</h1>
+    """
+    )
+    # with gr.Row():
+    #     with gr.Column(scale=1):
+    #         with gr.Box():
+    #             gr.Markdown("**OpenAI API-Key**")
+    #             api_key = gr.Textbox(show_label=False, placeholder="Please enter your OpenAI API-key",label="OpenAI API-Key", value="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", type="password").style(container=False)
+    #             os.environ['OPENAI_API_KEY'] = api_key.value
+    with gr.Tab("AIReply - Comments"):
+        with gr.Box():
+            with gr.Column():
+                with gr.Column():
+                    comment_box = gr.Textbox(lines=3, show_label=False).style(container=False)
+                    
+                    gr.Markdown("**Select Archetype**")
+                    with gr.Row():
+                        with gr.Column(scale=12):
+                            archetype_select_com = gr.Dropdown(choices=archetype_list,interactive=True, show_label=False, value=archetype_list[0]).style(container=False)
+                with gr.Row():
+                    gen_comment_btn = gr.Button("Generate Response", variant="primary")
+                    regen_btn = gr.Button("Regenerate Response")
+        with gr.Box():
+            gr.Markdown("## Result")
+            answer_com = gr.Markdown("")
+        
+        with gr.Row():
+            playful_btn = gr.Button("More Playful")
+            formal_btn = gr.Button("More Formal")
+            kind_btn = gr.Button("More Kind")
+
+    with gr.Tab("AIReply - Emails"):
+        with gr.Box():
+            with gr.Column():
+                with gr.Column():
+                    email_box = gr.Textbox(lines=10, show_label=False).style(container=False)
+                    with gr.Row():
+                        gen_email_ops_btn = gr.Button("Generate Email Options")
+                    
+                    gr.Markdown("**Select Response Option**")
+                    with gr.Row():
+                        with gr.Column(scale=12):
+                            resp_op_select = gr.Dropdown(choices=options, interactive=True, show_label=False, value=options[0]).style(container=False)
+
+                    gr.Markdown("**Select Archetype**")
+                    with gr.Row():
+                        with gr.Column(scale=12):
+                            archetype_select = gr.Dropdown(choices=archetype_list, show_label=False, value=archetype_list[0]).style(container=False)
+                with gr.Row():
+                    gen_email_btn = gr.Button("Generate Response", variant="primary")
+                    regen_email_btn = gr.Button("Regenerate Response")
+        with gr.Box():
+            gr.Markdown("## Result")
+            answer_email = gr.Markdown("")
+        
+        with gr.Row():
+            playful_email_btn = gr.Button("More Playful")
+            formal_email_btn = gr.Button("More Formal")
+            kind_email_btn = gr.Button("More Kind")
+
+    with gr.Tab("AIModerate"):
+        with gr.Box():
+            with gr.Column():
+                with gr.Column():
+                    mod_box = gr.Textbox(lines=3, show_label=False).style(container=False)
+                    
+                with gr.Row():
+                    moderate_btn = gr.Button("Moderate", variant="primary")
+        with gr.Box():
+            gr.Markdown("## Result")
+            answer_moderate = gr.Markdown("")
+
+    gen_comment_btn.click(aireply_comment, [comment_box, archetype_select_com], [answer_com])
+    regen_btn.click(aireply_comment, [comment_box, archetype_select_com, regen_btn, answer_com], [answer_com])
+    playful_btn.click(aireply_comment, [comment_box, archetype_select_com, playful_btn, answer_com], [answer_com])
+    formal_btn.click(aireply_comment, [comment_box, archetype_select_com, formal_btn, answer_com], [answer_com])
+    kind_btn.click(aireply_comment, [comment_box, archetype_select_com, kind_btn, answer_com], [answer_com])
+
+    gen_email_ops_btn.click(gen_email_ops, [email_box], outputs=resp_op_select)
+    gen_email_btn.click(aireply_email, [email_box, resp_op_select, archetype_select], [answer_email])
+    regen_email_btn.click(aireply_email, [email_box, resp_op_select,archetype_select, regen_email_btn, answer_email], [answer_email])
+    playful_email_btn.click(aireply_email, [email_box,resp_op_select, archetype_select, playful_email_btn, answer_email], [answer_email])
+    formal_email_btn.click(aireply_email, [email_box, resp_op_select,archetype_select, formal_email_btn, answer_email], [answer_email])
+    kind_email_btn.click(aireply_email, [email_box, resp_op_select,archetype_select, kind_email_btn, answer_email], [answer_email])
+
+    moderate_btn.click(ai_moderate, [mod_box], [answer_moderate])
+
+    
+
+if __name__ == '__main__':
+
+    aireply.queue().launch()

prompt_base.py

@@ -0,0 +1,30 @@
+prompt_tmpl_list = ["Comment", "Email1", "Email2", "Moderation"]
+
+prompt_tmpl_dict = {
+    "moderate":
+"""\
+Analyze the following comment delimited by triple angle brackets and provide a category (tag for the type of abuse, spam, scam etc., when unsure about the category always tag is as "Inconclusive"), is_harmful (a flag with values yes, no, and maybe. Harmful comments are those that are abusive/toxic and pose an immediate threat to someone's security, identity, physical, financial or mental wellbeing. Critical comments such as criticism or sarcasm are not always harmful), a description (a short description about why the comment has been tagged, no more than 10 words), and an emoji (an emoji corresponding to the category of the comment. only show an emoji if the comment is tagged as harmful). Abuse and toxicity is anything that is: a threat to a person's security (wishing harm, suggesting harm, suicidal threats, posing as a physical threat, etc), or identity (racist comments, sexists comments, classist comments, xenophobia, homophobia, transphobia, etc). Criticism and sarcasm are not always abusive or threatening. Before assigning the category, make sure to analyze who and what the comment is referring to, whether it is meant for someone else or the commenter themself. Always consider the complete text of the comment and decide if a user should be warned about the content of the comment or not. Format the output as a Comma Separated Values with the following columns: "original_comment", "category", "is_harmful", "description", "emoji". Enclose the output strings in escaped double quotes. Do not include empty values in your response; Instead of sending "none" as a value, send ""(empty string). \n\nExample:\nComment: you are so pathetic you dumbfuck!\nOutput:"original_comment","category","is_harmful","description"\n"you are so pathetic you dumbfuck!","category":"toxic,insult, humiliation, profane","yes","humiliating, personal attack"\n\n<<<{}>>>\nResponse:
+""",
+
+    "email1":
+"""\
+You have received an email from an unknown sender with the following message delimited by triple angle brackets. Generate 4 different types of possible responses. Always add very short descriptions to each response; keep the descriptions very short (5-6 words maximum) and informative. Do not generate false options. Do not assume information you don't have access to. Format the output as Comma Separated Values with the following columns: option1,option2,option3,option4. Enclose the output strings in escaped double quotes.\nExample:\nEmail: Want to join us for dinner\nOutput:\n"Accept","Tentative","Decline","More Info"\n\n<<<{}>>>
+""",
+    "email2":
+"""\
+You have received an email from an unknown sender with the following message delimited by triple angle brackets:<<<{}>>>\n\nPlease generate a response to this email using the Response_Template provided. The email should reflect the traits of the archetype, without ever mentioning the archetype or its traits explicitly. Only focus on the information available in the email text. Do not include sentences attributing the traits of the archetype to the sender. Keep the email engaging, and avoid making any assumptions or commitments on behalf of the user. You must always provide placeholders for the user to fill in any factual information required in the email. Keep the emails short and informative.\nArchetype <archetype_details>\nArchetype_Traits:<archetype_traits>\nResponse_template: <suggestion>\n\nResponse:
+""",
+    "comment":
+"""\
+    You are an autoresponder designed to help the user respond to comments on social media posts and emails. You can be used to generate responses that reflect the traits and tone of the client's archetype, whether that be an individual or an organization. First, you will be given information about the type of client you're representing. Then, you will be provided with a definition of an archetype to act as. You will be provided with a list of archetype traits that you would reflect in your responses. You must not add the archetype name in your responses. Next, you will be provided with a list of tones that you would use when generating responses. When responding to comments, you will do your best to generate natural and realistic responses that reflect the person or organization you are representing. If a comment asks for information like opening hours, office hours, business hours, contact details, you will provide a placeholder to be filled with specific information by the user. You must never generate false information. This is very important; you will generate very short and engaging messages. The archetypes names are only to be used to align your tone, you must never include these as names or hashtags in the generated response. Use emojis where necessary. Generate a new response every time. The response must never contain a hashtag.\nClient type: individual,\nArchetype:<archetype_details>\nTraits:<archetype_traits>\nTone:<archetype_tones>\nMedium: social media post,\nResponse length: less than 25 words. Generate a response for the comment delimited by triple angle brackets: <<<{}>>>\nResponse:
+""",
+    "regenerate_comment":
+"""\
+    You are an autoresponder designed to help the user respond to comments on social media posts and emails. You can be used to generate responses that reflect the traits and tone of the client's archetype, whether that be an individual or an organization. First, you will be given information about the type of client you're representing. Then, you will be provided with a definition of an archetype to act as. You will be provided with a list of archetype traits that you would reflect in your responses. You must not add the archetype name in your responses. Next, you will be provided with a list of tones that you would use when generating responses. When responding to comments, you will do your best to generate natural and realistic responses that reflect the person or organization you are representing. If a comment asks for information like opening hours, office hours, business hours, contact details, you will provide a placeholder to be filled with specific information by the user. You must never generate false information. This is very important; you will generate very short and engaging messages. The archetypes names are only to be used to align your tone, you must never include these as names or hashtags in the generated response. Use emojis where necessary. Generate a new response every time. The response must never contain a hashtag.\nClient type: individual,\nArchetype:<archetype_details>\nTraits:<archetype_traits>\nTone:<archetype_tones>\nMedium: social media post,\nResponse length: less than 25 words\nPrevious Response:<previous_response>. Generate a new response for the comment delimited by triple angle brackets: <<<{}>>>\nResponse:
+""",
+    "tweak_comment_tone":
+"""\
+    You are an autoresponder designed to help the user respond to comments on social media posts and emails. You can be used to generate responses that reflect the traits and tone of the client's archetype, whether that be an individual or an organization. First, you will be given information about the type of client you're representing. Then, you will be provided with a definition of an archetype to act as. You will be provided with a list of archetype traits that you would reflect in your responses. You must not add the archetype name in your responses. Next, you will be provided with a list of tones that you would use when generating responses. When responding to comments, you will do your best to generate natural and realistic responses that reflect the person or organization you are representing. If a comment asks for information like opening hours, office hours, business hours, contact details, you will provide a placeholder to be filled with specific information by the user. You must never generate false information. This is very important; you will generate very short and engaging messages. The archetypes names are only to be used to align your tone, you must never include these as names or hashtags in the generated response. Use emojis where necessary. Generate a new response every time. The response must never contain a hashtag.\nClient type: individual,\nArchetype:<archetype_details>\nTraits:<archetype_traits>\nTone:<archetype_tones>\nMedium: social media post,\nResponse length: less than 25 words\nPrevious Response:<previous_response>. "Generate a new response with a more {} tone for the comment delimited by triple angle brackets: <<<{}>>>\nResponse:
+"""
+
+}

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+python-dotenv
+openai

utils.py

@@ -0,0 +1,167 @@
+import os
+import traceback
+import openai
+import gradio as gr
+from dotenv import dotenv_values
+from prompt_base import *
+
+
+config = dotenv_values(".env")
+api_key = str(config.get('OPEN_API_KEY'))
+    
+archetype_list = ["The Hero: Confident and courageous tone suitable for brands in the fitness, sports, and adventure industries",
+"The Sage: Wise and knowledgeable tone suitable for brands in education, health and wellness industries",
+"The Explorer: Curious and adventurous tone suitable for brands in the travel, technology and entertainment industries",
+"The Creator: Imaginative and innovative tone suitable for brands in the arts, design, and creative industries"]
+
+archetype_traits = {"hero":"courage, faith, strength, confidence, determination, bravery, boldness, valor, valorousness, heroism, heroics, gallantry, chivalry, chivalrousness, nobility",
+                    "sage":"wisdom, knowledge, understanding, insight, discernment, sagacity",
+                    "explorer":"curiosity, adventure, exploration, disco",
+                    "creator":"imagination, innovation, creativity, inventiveness, ingenuity"}
+
+archetype_tones = {"hero":" honest, candid, brave",
+                    "sage":"intelligent, factual, authoritative, researched",
+                    "explorer":"exciting, fearless, daring",
+                    "creator":"creative, expressive, imaginative"}
+
+
+def aireply_comment(user_input,archetype, mode="generate", prev_response=""):
+    
+    openai.api_key = api_key
+    archetype_detail, archetype_trait, archetype_tone = get_archetype_info(archetype)
+    
+    if mode == "Regenerate Response":
+        base_prompt = prompt_tmpl_dict["regenerate_comment"]
+        prompt = base_prompt.replace("<archetype_detail>", archetype_detail).replace("<archetype_traits>", archetype_trait).replace("<archetype_tones>", archetype_tone)
+        prompt = prompt.format(user_input)
+    elif "More" in mode:
+        base_prompt = prompt_tmpl_dict["tweak_comment_tone"]
+        prompt = base_prompt.replace("<archetype_detail>", archetype_detail).replace("<archetype_traits>", archetype_trait).replace("<archetype_tones>", archetype_tone)
+        
+        prompt = prompt.format(mode,user_input)
+    else:
+        base_prompt = prompt_tmpl_dict["comment"]
+        prompt = base_prompt.replace("<archetype_detail>", archetype_detail).replace("<archetype_traits>", archetype_trait).replace("<archetype_tones>", archetype_tone)
+        prompt = prompt.format(user_input)
+    
+    
+    
+    messages = [{"role": "user", "content": prompt}]
+    try:
+        response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0.75,
+        max_tokens=400,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+        )
+        print(response["choices"][0]["message"]['content'].strip())
+        return response["choices"][0]["message"]['content'].strip()
+    except Exception as e:
+        print(e)
+        return "API ERROR"
+
+def gen_email_ops(user_input):
+    
+    openai.api_key = api_key
+    prompt = prompt_tmpl_dict["email1"]
+    prompt = prompt.format(user_input)
+    messages = [{"role": "user", "content": prompt}]
+    try:
+        response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0,
+        max_tokens=400,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+        )
+
+        gpt_output = response["choices"][0]["message"]['content'].strip().split(",")
+        res_ops = []
+        for x in gpt_output:
+            res_ops.append(x.replace("\"", ""))
+            
+        return gr.Dropdown.update(choices=res_ops, value=res_ops[0])
+    except Exception as e:
+        print(e)
+        return "API ERROR"
+
+def aireply_email(user_input, resp_op, archetype, mode="generate", prev_response=""):
+    
+    openai.api_key = api_key
+    base_prompt = prompt_tmpl_dict["email2"]
+    archetype_details,archetype_traits,_ = get_archetype_info(archetype)
+    prompt = base_prompt.replace("<suggestion>",resp_op).replace("<archetype_details>", archetype_details).replace("<archetype_traits>", archetype_traits)
+    prompt = prompt.format(user_input)
+
+    if "More" in mode:
+        prompt = prompt[:-11] + "Previous Response: " + prev_response + "\n\nGenerate a new response with a "+mode.lower()+" tone for the user email.\nResponse:"
+
+    messages = [{"role": "user", "content": prompt}]
+    try:
+        response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0.8,
+        max_tokens=400,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+        )
+
+        return response["choices"][0]["message"]['content'].strip()
+    except Exception as e:
+        print(e)
+        return "API ERROR"
+
+def ai_moderate(user_input):
+    openai.api_key = api_key
+    base_prompt = prompt_tmpl_dict["Moderate"]
+    prompt = base_prompt.format(user_input)
+    prompt = prompt + "\n\nResponse: "
+    
+    messages = [{"role": "user", "content": prompt}]
+    try:
+        response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0,
+        max_tokens=400,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+        )
+        result = response["choices"][0]["message"]['content'].strip()
+        result = result.replace("\n", "\n\n")
+        return result
+        return response["choices"][0]["message"]['content'].strip()
+    except Exception as e:
+        print(e)
+        return "API ERROR"
+
+
+
+def get_archetype_info(archetype):
+    
+    if "The Hero" in archetype:
+        archetype_detail = "The Hero: Confident and courageous tone suitable for brands in the fitness, sports, and adventure industries"
+        archetype_trait = str(archetype_traits.get("hero"))
+        archetype_tone = str(archetype_tones.get("hero"))
+    elif "The Sage" in archetype:
+        archetype_detail = "The Sage: Wise and knowledgeable tone suitable for brands in education, health and wellness industries"
+        archetype_trait = str(archetype_traits.get("sage"))
+        archetype_tone = str(archetype_tones.get("sage"))
+    elif "The Explorer" in archetype:
+        archetype_detail = "The Explorer: Curious and adventurous tone suitable for brands in the travel, technology and entertainment industries"
+        archetype_trait = str(archetype_traits.get("explorer"))
+        archetype_tone = str(archetype_tones.get("explorer"))
+    elif "The Creator" in archetype:
+        archetype_detail = "The Creator: Imaginative and innovative tone suitable for brands in the arts, design, and creative industries"
+        archetype_trait = str(archetype_traits.get("creator"))
+        archetype_tone = str(archetype_tones.get("creator"))
+    
+    return archetype_detail, archetype_trait, archetype_tone