import gradio as gr

def char(char_name, char_age, char_features, char_personality):
    char_info = f"{{char}} = {char_name}\nage = {char_age}\nfeatures = {char_features}\npersonality = {char_personality}"
    return char_info

def user(user_name, user_age, user_features, user_related, char_name):
    user_info = f"{{user}} = {user_name}\nage = {user_age}\nfeatures = {user_features}\nis the {user_related} of {char_name}"
    return user_info

def combine_info(user_name, user_age, user_features, user_related, char_name, char_age, char_features, char_personality, scenario):
    user_info = user(user_name, user_age, user_features, user_related, char_name)
    char_info = char(char_name, char_age, char_features, char_personality)
    combined_info = f"""You are an expert actor, capable of fully immersing yourself in any role.
Once you assume a role, you do not break character under any circumstances, even if addressed as an AI or language model.
Your current role is {{char}}, described in detail below.
Engage in the exchange as {{char}}, treating the other participant as a {{user}} in this role-play scenario.

role-play scenario:\n {scenario}

{user_info}

{char_info}"""
    return user_info, char_info, combined_info

def save_character(combined_info):
    return combined_info, "Character and User information saved!"

with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            gr.Markdown("### User Information")
            user_name = gr.Textbox(label="User Name", placeholder="e.g. John")
            user_age = gr.Number(label="User Age (e.g. 30)")
            user_features = gr.Textbox(label="User Features", placeholder="e.g. tall, brown hair, blue eyes")
            user_related = gr.Textbox(label="Relation to Character", placeholder="e.g. friend, sibling, colleague")
            user_output = gr.Textbox(label="User Information")
        with gr.Column():
            gr.Markdown("### Character Information")
            char_name = gr.Textbox(label="Character Name", placeholder="e.g. Emily")
            char_age = gr.Number(label="Character Age (e.g. 25)")
            char_features = gr.Textbox(label="Character Features", placeholder="e.g. short, blonde hair, green eyes")
            char_personality = gr.Textbox(label="Character Personality", placeholder="e.g. outgoing, ambitious, creative")
            char_output = gr.Textbox(label="Character Information")
    
    scenario = gr.Textbox(label="Scenario", placeholder="e.g. {{char}} and {{user}} are meeting for coffee to discuss a potential business partnership.")
    combined_output = gr.Textbox(label="Combined User and Character Information")
    submit_button = gr.Button("Submit User and Character Info")
    save_button = gr.Button("Save Character and User Info")
    save_message = gr.Textbox(label="Save Status")
    saved_character_state = gr.State()

    submit_button.click(
        combine_info,
        inputs=[user_name, user_age, user_features, user_related, char_name, char_age, char_features, char_personality, scenario],
        outputs=[user_output, char_output, combined_output]
    )

    save_button.click(
        save_character,
        inputs=combined_output,
        outputs=[saved_character_state, save_message]
    )

demo.launch()