feat: iframe src 支持传入 prompt

app.py

@@ -530,6 +530,55 @@ with gr.Blocks(css=CSS, title="Krea 2 Identity Edit") as demo:
             """
         )
 
+        gr.HTML(
+            """
+            <script>
+            (() => {
+                const getPromptFromUrl = () => {
+                    const params = new URLSearchParams(window.location.search);
+                    return params.get("prompt") || "";
+                };
+
+                const applyPrompt = () => {
+                    const prompt = getPromptFromUrl();
+                    if (!prompt) return;
+
+                    const textarea = document.querySelector("textarea");
+                    if (!textarea) return false;
+
+                    if (textarea.value !== prompt) {
+                        textarea.value = prompt;
+                        textarea.dispatchEvent(new Event("input", { bubbles: true }));
+                        textarea.dispatchEvent(new Event("change", { bubbles: true }));
+                    }
+                    return true;
+                };
+
+                const start = () => {
+                    if (applyPrompt()) return;
+
+                    const observer = new MutationObserver(() => {
+                        if (applyPrompt()) observer.disconnect();
+                    });
+
+                    observer.observe(document.body, {
+                        childList: true,
+                        subtree: true
+                    });
+
+                    setTimeout(() => observer.disconnect(), 10000);
+                };
+
+                if (document.readyState === "loading") {
+                    document.addEventListener("DOMContentLoaded", start);
+                } else {
+                    start();
+                }
+            })();
+            </script>
+            """
+        )
+
         with gr.Row(equal_height=True):
             with gr.Column():
                 source_image = gr.Image(label="Source image", type="pil", height=420)