Upload 2 files

app.py

@@ -4,25 +4,15 @@ 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)
+        prompt = gr.Textbox(label="Generated Prompt", interactive=True, lines=20).style(show_copy_button=True)
     with gr.Tab("Edit System Prompt"):
         system_prompt = gr.Textbox(
             label="System Prompt", interactive=True, lines=15
-        )
+        ).style(show_copy_button=True)
+        save_s_p = gr.Button("Save System Prompt")
         gr.Markdown("## System Prompt Examples")
         gr.Examples(
             examples = [[pre_text]],
@@ -32,7 +22,10 @@ with gr.Blocks() as app:
     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(
+    save_s_p.click(
+        fn = set_pre_text, inputs = [system_prompt, session_data], outputs = session_data, queue=False
+    )
+    system_prompt.submit(
         fn = set_pre_text, inputs = [system_prompt, session_data], outputs = session_data, queue=False
     )
     

helper_functions.py

@@ -44,3 +44,15 @@ def get_context(query):
 My question is: “{query}”
 """
 
+
+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
+