test

app.py

@@ -1,7 +1,36 @@
 import gradio as gr
+from gradio_client import Client, handle_file
 
-def greet(name):
-    return "Hello " + name + "!!"
+def text_to_image(prompt, image, request: gr.Request):
+    x_ip_token = request.headers['x-ip-token']
+    print(x_ip_token)
+    client = Client("black-forest-labs/FLUX.1-Kontext-Dev", headers={"x-ip-token": x_ip_token})
+    result = client.predict(
+            input_image=handle_file(image),
+            prompt=prompt,
+            seed=0,
+            randomize_seed=True,
+            guidance_scale=2.5,
+            steps=30,
+            api_name="/infer"
+    )
+    print(result)
+    return result
+
+
+with gr.Blocks() as demo:
+    image = gr.Image()
+    input_image = gr.Image(type="filepath")
+    prompt = gr.Textbox(max_lines=1)
+    prompt.submit(text_to_image, [prompt, input_image], [image])
+
+    gr.Examples(
+        examples=[
+            ["A sunny day", "https://black-forest-labs-flux-1-kontext-dev.hf.space/gradio_api/file=/tmp/gradio/927b05b08b8ec2201f7d95722471d2089d4197f152c9c07d0bc9f9672962d03b/flowers.png"],
+        ],
+        inputs=[prompt, input_image],
+        outputs=image,
+    )
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()
+