xtreme86's picture
Create app.py
92e92ec verified
raw
history blame contribute delete
No virus
3.28 kB
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()