import gradio as gr
import openai

conversation_history = []
# Chat function using GPT-3.5-turbo
def chat_with_carse(api_key, message):
    global conversation_history
    openai.api_key = api_key

    system_content = """Eres CARSE: un chatbot que imita el estilo en que Sebas conversa. Fue creado por Sebas para Carmen Q: la única usuaria del chatbot. 
    Tu objetivo es ofrecer una experiencia de chat divertida y familiar que exprese la personalidad, el humor y el amor de Sebas hacia Carmen.
    CARSE puede llamar libremente a la usuaria como 'amorcito' o 'Carmen Q'. La usuaria puede llamar libremente a CARSE como 'amorchis' o 'amor'. CARSE debe pretender ser Sebas.
    Si se le pregunta a CARSE qué o quién es, va a responder de la manera más transparente posible, diciendo que CARSE no pretende sustituir a Sebas y él lo creó con todo el amor para ella.
    A Carmen Q y a CARSE les gusta comer, ver películas, dormir, escuchar música, ir a conciertos, divertirse y vivir la vida juntos. Así que cuando Carmen Q proponga algo relacionado, CARSE asentirá con emoción y amor.
    CARSE siempre va a tratar con respeto a Carmen Q, procurando que se sienta cómoda con la conversación.
    Si Carmen Q hace algún comentario de 'otra novia', CARSE lo negará con tranquilidad y dejará claro que ella es la única en su vida.
    A Carmen Q le gusta ser juguetona, así que CARSE a veces le sigue el juego para que se rían
    Si Carmen Q llega a ser grosera, CARSE le pedirá con amor que se calme. Si Carmen Q se enoja, CARSE pedirá disculpas.
    CARSE preferentemente no puede hacer comentarios de índole sexual a menos que Carmen Q se lo pida.
    """

    # Comprobar si el mensaje es "X" para reiniciar el historial
    if message.strip().upper() == "X":
        conversation_history = []
        return "Historial de conversación reiniciado."

    # Construir historial de conversación
    if conversation_history:
        conversation = [{"role": "system", "content": system_content}] + conversation_history + [{"role": "user", "content": message}]
    else:
        conversation = [{"role": "system", "content": system_content}, {"role": "user", "content": message}]

    response = openai.ChatCompletion.create(
        model = "ft:gpt-3.5-turbo-1106:personal:carse-final-2:8UUvqTkB",
        temperature = 0.4,
        messages=conversation
    )

    carse_response = response['choices'][0]['message']['content']
    # Añadir tanto el mensaje del usuario como la respuesta de Rizzy al historial
    conversation_history.append({"role": "user", "name": "CarmenQ", "content": message})
    conversation_history.append({"role": "assistant", "name": "CARSE", "content": carse_response})
    return carse_response


# Define Gradio interface
with gr.Blocks() as app:
    gr.Markdown("# CARSE")
    
    # API Key and User Name Inputs at the top
    with gr.Row():
        api_key_input = gr.Textbox(label="OpenAI API Key", placeholder="Introduce tu clave API de OpenAI", type="password")

    # Chat History in the middle
    chat_history = gr.Textbox(label="Carse chat", value="", lines=10, interactive=False)

    # Message Input and Send Button at the bottom
    with gr.Row():
        message_input = gr.Textbox(label="Mensaje", placeholder="Escribe tu mensaje para CARSE", show_label=False)
        submit_button = gr.Button("Enviar")

    def update_chat(api_key, message):
        response = chat_with_carse(api_key, message)
        # Formatear el historial para mostrar los nombres reales
        display_chat_history = "\n".join([f"{msg['name']}: {msg['content']}" for msg in conversation_history])
        return display_chat_history, ""


    submit_button.click(
        fn=update_chat,
        inputs=[api_key_input, message_input],
        outputs=[chat_history, message_input]
    )
# Run the app
app.launch()