| import api |
| import gradio as gr |
| import random |
|
|
| with gr.Blocks() as demo: |
| with gr.Column() as setup_col: |
| gr.HTML("<h1 style='text-align: center; margin-top: 1em; font-size: 2em;'>Rizz++</h1>") |
| gr.Markdown("Rizz++ simulates conversations you might have when approaching someone you're interested in, helping you develop your skills. \nRandomly generate a scenario involving...") |
| with gr.Row(): |
| approach_girl = gr.Button("Approaching a Girl") |
| approach_boy = gr.Button("Approaching a Boy") |
| gr.Markdown("...or enter your own scenario:") |
| with gr.Group(): |
| custom_scenario = gr.Textbox(lines=3, label="Scenario", placeholder="There's a cute girl in my boxing class I want to talk to.") |
| scenario_btn = gr.Button("Submit Scenario") |
|
|
| with gr.Column(visible=False) as convo_col: |
| chat = gr.Chatbot() |
| recording = gr.Audio(label="Your Line", sources="microphone", type="filepath") |
| with gr.Group(): |
| with gr.Row(): |
| help_btn = gr.Button("Help Me", scale=0) |
| suggestion = gr.Textbox(container=False, placeholder="Stuck? Let Rizz++ suggest your next line!") |
|
|
| @gr.on([approach_boy.click, approach_girl.click, scenario_btn.click], outputs=[setup_col, convo_col]) |
| def start_convo(): |
| return { |
| setup_col: gr.Column(visible=False), |
| convo_col: gr.Column(visible=True) |
| } |
| |
| scenario = gr.State() |
| personality = gr.State(lambda: random.choice(["shy", "confident", "awkward", "charming", "rude"])) |
| interested = gr.State(lambda: random.choice([True, False])) |
| SCENARIO_LINE = "The scenario is: **{}** \n\n Record your opening line below!" |
|
|
| def start_chat(scenario): |
| return [[SCENARIO_LINE.format(scenario), None]] |
|
|
| approach_boy.click(lambda: api.generate_scenario("boy"), outputs=scenario).then(start_chat, scenario, chat) |
| approach_girl.click(lambda: api.generate_scenario("girl"), outputs=scenario).then(start_chat, scenario, chat) |
| scenario_btn.click(lambda x: x, custom_scenario, scenario).then(start_chat, scenario, chat) |
|
|
|
|
| def transcribe(recording, chat): |
| user_message = api.transcribe_audio(recording) |
| chat.append([user_message, None]) |
| return chat |
| |
| def respond(scenario, chat, personality, interested): |
| response = api.generate_response(scenario, chat[1:], personality, interested) |
| chat[-1][1] = response |
| return chat |
|
|
|
|
| recording.stop_recording(transcribe, inputs=[recording, chat], outputs=[chat] |
| ).then(respond, inputs=[scenario, chat, personality, interested], outputs=[chat] |
| ).then(lambda: None, outputs=[recording]) |
|
|
| help_btn.click(lambda scenario, chat: api.suggest_next_line(scenario, chat[1:]), inputs=[scenario, chat], outputs=[suggestion]) |
| |
| demo.launch() |
