|
import gradio as gr |
|
from helper_functions import * |
|
|
|
with gr.Blocks() as app: |
|
gr.Markdown('# Prompt Generator for FundedNext') |
|
session_data = gr.State([{"current_system_prompt": ""}]) |
|
|
|
def get_prompt(user_message, session_data): |
|
if(user_message == ""): |
|
return_message = "Please enter your message" |
|
return return_message |
|
pre_text = session_data[0]["current_system_prompt"] |
|
return_message = pre_text + "\n\n" + get_context(user_message) |
|
return return_message |
|
def set_pre_text(system_prompt, session_data): |
|
session_data[0]["current_system_prompt"] = system_prompt |
|
return session_data |
|
|
|
with gr.Tab("Generate Prompt"): |
|
user_message = gr.Textbox(label = "Enter your message") |
|
prompt = gr.Textbox(label="Generated Prompt", interactive=True, lines=20) |
|
with gr.Tab("Edit System Prompt"): |
|
system_prompt = gr.Textbox( |
|
label="System Prompt", interactive=True, lines=15 |
|
) |
|
gr.Markdown("## System Prompt Examples") |
|
gr.Examples( |
|
examples = [[pre_text]], |
|
inputs = [system_prompt] |
|
) |
|
|
|
user_message.submit( |
|
fn = get_prompt, inputs = [user_message, session_data], outputs = prompt |
|
).then(lambda:"", inputs=None, outputs=user_message, queue=False) |
|
system_prompt.change( |
|
fn = set_pre_text, inputs = [system_prompt, session_data], outputs = session_data, queue=False |
|
) |
|
|
|
|
|
app.launch(auth=(os.getenv("id"), os.getenv("password")), show_api=False) |
|
|
