feat: allow sending the raw json input

.gitignore

@@ -0,0 +1 @@
+__pycache__

app.py

@@ -191,11 +191,20 @@ def generate(
     height=1024,
     seed=0,
     randomize_seed=False,
+    raw_json_prompt="",
     progress=gr.Progress(track_tqdm=True),
 ):
     if randomize_seed or seed < 0:
         seed = random.randint(0, MAX_SEED)
 
+    raw_json_prompt = (raw_json_prompt or "").strip()
+    if raw_json_prompt:
+        try:
+            final_prompt = json.dumps(json.loads(raw_json_prompt), ensure_ascii=False, separators=(",", ":"))
+        except Exception as e:
+            raise gr.Error(f"Raw JSON prompt is not valid JSON: {e}")
+        return _gpu_generate(final_prompt, mode, width, height, seed, False)
+
     # Remote upsample is a network call -> run it here, OFF the GPU. Fall back to local (on-GPU) on failure.
     final_prompt, do_local = prompt, True
     if upsampler == UPSAMPLERS[0] and IDEOGRAM_API_KEY:
@@ -251,6 +260,13 @@ with gr.Blocks(theme=gr.themes.Citrus(), title="Ideogram 4", css=CSS) as demo:
                     label="Prompt upsampler",
                     info="Rewrite into Ideogram's native JSON caption. Remote (Ideogram) preferred; falls back to local.",
                 )
+                raw_json_prompt = gr.Textbox(
+                    label="Raw JSON prompt (optional)",
+                    info="Paste a generated JSON caption here to send it directly to the model, skipping prompt upsampling.",
+                    lines=8,
+                    placeholder='{"prompt":"a ginger cat wearing a tiny wizard hat reading a spellbook"}',
+                    buttons=["copy"],
+                )
                 with gr.Row():
                     width = gr.Slider(512, 2048, value=1024, step=64, label="Width")
                     height = gr.Slider(512, 2048, value=1024, step=64, label="Height")
@@ -259,7 +275,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), title="Ideogram 4", css=CSS) as demo:
                     randomize = gr.Checkbox(label="Randomize seed", value=True)
         with gr.Column():
             out_image = gr.Image(label="Output", type="pil")
-            out_caption = gr.JSON(label="Caption fed to the model (upsampled when enabled)")
+            out_caption = gr.JSON(label="Caption fed to the model (upsampled when enabled)", buttons=["copy"])
 
     gr.Examples(
         examples=[
@@ -276,7 +292,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), title="Ideogram 4", css=CSS) as demo:
 
     run.click(
         generate,
-        inputs=[prompt, mode, upsampler, width, height, seed, randomize],
+        inputs=[prompt, mode, upsampler, width, height, seed, randomize, raw_json_prompt],
         outputs=[out_image, seed, out_caption],
     )