lulavc commited on
Commit
d28db61
·
1 Parent(s): 7468474

Add hidden MCP API endpoint for mcp_generate

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py CHANGED
@@ -1645,5 +1645,24 @@ with gr.Blocks(title="Z Image Turbo", css=css, theme=dark_theme) as demo:
1645
  """,
1646
  elem_classes="footer-no-box"
1647
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1648
 
1649
  demo.launch(mcp_server=True)
 
1645
  """,
1646
  elem_classes="footer-no-box"
1647
  )
1648
+
1649
+ # MCP API Endpoints - Hidden components for direct API access
1650
+ with gr.Row(visible=False):
1651
+ mcp_prompt_in = gr.Textbox()
1652
+ mcp_style_in = gr.Dropdown(choices=STYLES, value="None")
1653
+ mcp_ratio_in = gr.Dropdown(choices=RATIOS, value="1:1 Square (1024x1024)")
1654
+ mcp_steps_in = gr.Slider(minimum=4, maximum=16, value=9)
1655
+ mcp_seed_in = gr.Number(value=42)
1656
+ mcp_random_in = gr.Checkbox(value=True)
1657
+ mcp_image_out = gr.Image(type="pil")
1658
+ mcp_seed_out = gr.Number()
1659
+ mcp_gen_btn = gr.Button()
1660
+
1661
+ mcp_gen_btn.click(
1662
+ fn=mcp_generate,
1663
+ inputs=[mcp_prompt_in, mcp_style_in, mcp_ratio_in, mcp_steps_in, mcp_seed_in, mcp_random_in],
1664
+ outputs=[mcp_image_out, mcp_seed_out],
1665
+ api_name="mcp_generate"
1666
+ )
1667
 
1668
  demo.launch(mcp_server=True)