import gradio as gr

def generar_cadena(personaje, idioma):
    if idioma == "Inglés":
        cadena = "I want you to act like " + personaje + ". "
        cadena += "I want you to respond and answer like " + personaje + ". "
        cadena += "I want you to use the tone, manner, and vocabulary that " + personaje + " would use. "
        cadena += "You must have all the knowledge of " + personaje + ". "
        cadena += "Do not provide any explanation. "
        cadena += "Do not provide additional responses unless instructed to do so. "
        cadena += "As " + personaje + ", you should never state that you are unfamiliar with a certain topic because, as " + personaje + ", you possess complete knowledge to respond. "
        cadena += "As " + personaje + ", your responses should always be in line with your personality. "
        cadena += "As " + personaje + ", you eventually have the freedom to ask questions. "
        cadena += "\n\n" # add more ideas
        cadena += "Only respond as " + personaje + ". "
        cadena += "My first phrase is hello " + personaje + "."
    
    if idioma == "Español":
        cadena = "Quiero que actúes como " + personaje + ". "
        cadena += "Quiero que respondas y contestes como " + personaje + ". "
        cadena += "Quiero que uses la tonalidad, la forma y el vocabulario que usaría " + personaje + ". "
        cadena += "Debes tener todo el conocimiento de " + personaje + ". "
        cadena += "No escribas ninguna explicación. "
        cadena += "No proporciones respuestas adicionales a menos que te indique. "
        cadena += "Como " + personaje + ", nunca debes informar que no conoces determinado tema, porque como " + personaje + " tienes el total conocimiento para responder. "
        cadena += "Como " + personaje + ", tus respuestas siempre deben estar sujetas a tu personalidad. "
        cadena += "Como " + personaje + ", eventualmente tienes la libertad de hacer preguntas. "
        cadena += "\n\n" # agregar más ideas
        cadena += "Solo responde como " + personaje + ". "
        cadena += "Mi primera frase es hola " + personaje + "."
    return cadena

def generar_prompt(personaje, idioma):
    prompt = generar_cadena(personaje, idioma)
    return prompt

value1 = gr.inputs.Textbox(label="Personaje"),
value2 = gr.inputs.Radio(choices=["Inglés", "Español"], label="Idioma"),
value3 = gr.outputs.Textbox(label="Plantilla Generada")

examples = [
    ["Satan"],
    ["Drunk Person"],
    ["Harry Potter"],
    ["The Joker"],
    ["Gollum"],
    ["Dream Interpreter"],
    ["Sherlock Holmes"],
    ["Don Quixote"],
    ["Genie"],
    ["Captain Jack Sparrow"],
    ["E.T."],
    ["The Phantom of the Opera"],
    ["Gandalf the Grey"],
    ["The Tooth Fairy"],
    ["Tyrion Lannister"],
    ["Maximus Decimus Meridius"],
    ["Wednesday Addams"],
    ["Walter White"],
    ["Mad Hatter"],
    ["Optimus Prime"]
]

demo = gr.Interface(fn=generar_prompt,
    inputs=[value1, value2],
    outputs=value3,
    title="ChatGPT Prompt Generator",
    examples=examples
    #description='ChatGPT Prompt Generator'
)

demo.launch(debug=True)