Update app.py

app.py

@@ -142,6 +142,17 @@ window.postMessageToParent = function(prompt, event, source, value) {
     };
     if (!prompt){
         prompt = window.g();
+        
+        // Find the prompt input element
+        const promptContainer = document.getElementById('prompt_input_box');
+        if (promptContainer) {
+            const promptInput = promptContainer.querySelector('input') || promptContainer.querySelector('textarea');
+            if (promptInput) {
+                promptInput.value = prompt;
+                // Trigger an input event to ensure Gradio recognizes the change
+                promptInput.dispatchEvent(new Event('input', { bubbles: true }));
+            }
+        }
     }
     // Post the message to the parent window
     window.parent.postMessage(message, '*');
@@ -213,29 +224,6 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
-
-
-with gr.Blocks(css=css,head=js,fill_height=True) as demo:
-    with gr.Row(equal_height=False):
-        with gr.Group():
-            gr.HTML(value=desc_html, elem_id='desc_html_code')
-            result = gr.Gallery(
-              label="Result",  show_label=False, columns=1, rows=1, show_share_button=True,
-              show_download_button=True,allow_preview=True,interactive=False, min_width=cfg.get("window_min_width", 340),height=360
-            )
-            with gr.Row(): 
-                prompt = gr.Text(
-                    show_label=False,
-                    max_lines=2,
-                    lines=2,
-                    placeholder="Enter what you want to see",
-                    container=False,
-                    scale=5,
-                    min_width=100,
-                )
-                random_button = gr.Button("Surprise Me", scale=1, min_width=10)
-            run_button = gr.Button( "GO!", scale=1, min_width=20, variant="primary",icon="https://huggingface.co/spaces/nsfwalex/sd_card/resolve/main/hot.svg")
-
     @spaces.GPU(duration=60)
     def generate(p, progress=gr.Progress(track_tqdm=True)):
         negative_prompt = cfg.get("negative_prompt", "")
@@ -248,8 +236,6 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
         randomize_seed = True
         guidance_scale = cfg.get("guidance_scale", 7.5)
         prompt_str = cfg.get("prompt", "{prompt}").replace("{prompt}", p)
-        if not prompt.value:
-            prompt.value = p
         seed = int(randomize_seed_fn(seed, randomize_seed))
         generator = torch.Generator(pipe.device).manual_seed(seed)
             
@@ -268,6 +254,28 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
         image_paths = [i[1] for i in images]
         print(prompt_str, image_paths)
         return [i[0] for i in images]
+
+with gr.Blocks(css=css,head=js,fill_height=True) as demo:
+    with gr.Row(equal_height=False):
+        with gr.Group():
+            gr.HTML(value=desc_html, elem_id='desc_html_code')
+            result = gr.Gallery(
+              label="Result",  show_label=False, columns=1, rows=1, show_share_button=True,
+              show_download_button=True,allow_preview=True,interactive=False, min_width=cfg.get("window_min_width", 340),height=360
+            )
+            with gr.Row(): 
+                prompt = gr.Text(
+                    show_label=False,
+                    max_lines=2,
+                    lines=2,
+                    placeholder="Enter what you want to see",
+                    container=False,
+                    scale=5,
+                    min_width=100,
+                    elem_id="prompt_input_box"
+                )
+                random_button = gr.Button("Surprise Me", scale=1, min_width=10)
+            run_button = gr.Button( "GO!", scale=1, min_width=20, variant="primary",icon="https://huggingface.co/spaces/nsfwalex/sd_card/resolve/main/hot.svg")
         
     def on_demo_load(request: gr.Request):
         current_domain = request.request.headers.get("Host", "")