Update app.py

app.py

@@ -13,6 +13,8 @@ import PIL
 from PIL import Image
 import re
 
+send_prompt(prompt):
+    return prompt
 
 def query(payload, model):
     HF_TOKEN = os.getenv("HF_TOKEN")
@@ -381,7 +383,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 with gr.Column():
                     image_input = gr.Image(type="filepath", interactive=True)
-            png_button = gr.Button("Get Info")
+            png_button = gr.Button("Get Info", variant="primary")
             with gr.Row():
                 with gr.Column():
                     exif_output = gr.HTML(label="EXIF Data")
@@ -439,7 +441,7 @@ with gr.Blocks(css=css) as demo:
             gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2')
             with open("ideas.txt", "r") as f:
                 line = f.readlines()
-            def generate(starting_text):
+            def prompt_gen(starting_text):
                 seed = random.randint(100, 1000000)
                 set_seed(seed)
                 if starting_text == "":
@@ -454,9 +456,18 @@ with gr.Blocks(css=css) as demo:
                 response_end = "\n".join(response_list)
                 response_end = re.sub('[^ ]+\.[^ ]+','', response_end)
                 response_end = response_end.replace("<", "").replace(">", "")
-
-    if response_end != "":
-        return response_end
+                if response_end != "":
+                    return response_end
+            idea = gr.Textbox(show_label=False, placeholder="Enter your idea", max_lines=3, lines=1, interactive=True)
+            prompt_button = gr.Button(value="Generate Prompt", variant="primary")
+            prompt_full = gr.Textbox(label="Full Prompt", max_lines=8, lines=3, interactive=False, show_copy_button=True)
+            send_to_txt2img_prompt = gr.Button(value="Send FULL PROMPT to txt2img")
+            send_to_img2img_prompt = gr.Button(value="Send FULL PROMPT to img2img")
+
+            send_to_txt2img_prompt.click(send_prompt, inputs=prompt_full, outputs=prompt)
+            send_to_img2img_prompt.click(send_prompt, inputs=prompt_full, outputs=i2i_prompt)
+            idea.submit(prompt_gen, inputs=idea, outputs=prompt_full)
+            prompt_button.click(prompt_gen, inputs=idea, outputs=prompt_full)
 
         with gr.Tab("BLIP"):
             with gr.Tab("Base"):