app.py

import gradio as gr


def generate_prompt(input_a, input_b, input_c, input_d, sequence, n):
    prompt = ""
    if sequence == 1:
        prompt = input_c
    elif sequence == 2:
        prompt = input_a + input_b + input_c
    elif sequence == 3:
        prompt = input_a + input_b
    elif sequence == 4:
        prompt = input_d + input_c
    
    return "Prompt {}:\n{}".format(n, prompt)


input_a = gr.Textbox("a battle rap")
input_b = gr.Textbox("(every line(6 syllables) rapping about:(")
input_c = gr.Textbox("riding giant Cheerios through a raging river of milk, a cereal white water rapids)")
input_d = gr.Textbox("a book cover(by Dr Seuss), that portrays a whimsical illustration of (")

sequence_options = {
    1: "Input C",
    2: "Input A + Input B + Input C",
    3: "Input A + Input B",
    4: "Input D + Input C"
}

output1_seq = gr.Dropdown(choices=sequence_options, label="Output 1 Sequence")
output2_seq = gr.Dropdown(choices=sequence_options, label="Output 2 Sequence")

submit_button = gr.Button("Generate Prompts")

output1 = gr.Textbox(label="Output 1 Prompt")
output2 = gr.Textbox(label="Output 2 Prompt")


def generate_prompts(input_a_val, input_b_val, input_c_val, input_d_val, output1_seq_val, output2_seq_val):
    
    prompt1 = generate_prompt(input_a_val, input_b_val, input_c_val, input_d_val, output1_seq_val, 1)
    prompt2 = generate_prompt(input_a_val, input_b_val, input_c_val, input_d_val, output2_seq_val, 2)
    
    return {"output1": prompt1, "output2": prompt2}
        

gr.Interface(
    generate_prompts, 
    inputs=[input_a, input_b, input_c, input_d, output1_seq, output2_seq, submit_button], 
    outputs=[output1, output2],
    title="Prompt Generator",
    description="Generate prompts based on input text and selected sequences."
).launch();