Spaces:
Running
Running
File size: 3,320 Bytes
c079997 583b7ad c079997 fcb8f25 ad3aed5 fcb8f25 c079997 fcb8f25 c079997 ad3aed5 c079997 fcb8f25 c079997 fcb8f25 c079997 ad3aed5 c079997 fcb8f25 c079997 fcb8f25 c079997 fcb8f25 c079997 fcb8f25 c079997 fcb8f25 ebf25c1 fcb8f25 c079997 fcb8f25 c079997 fcb8f25 c079997 ebf25c1 fcb8f25 ebf25c1 c079997 fcb8f25 c079997 fcb8f25 ebf25c1 fcb8f25 c079997 fcb8f25 c079997 fcb8f25 c079997 fcb8f25 c079997 ebf25c1 fcb8f25 ebf25c1 24c3cc8 fcb8f25 24c3cc8 fcb8f25 24c3cc8 c079997 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
import gradio as gr
from src.agents.image_edit_agent import image_edit_agent, ImageEditDeps, EditImageResult
import os
from src.hopter.client import Hopter, Environment
from src.services.generate_mask import GenerateMaskService
from dotenv import load_dotenv
from pydantic_ai.messages import ToolReturnPart
from src.utils import upload_image
load_dotenv()
async def process_edit(image, instruction):
hopter = Hopter(os.environ.get("HOPTER_API_KEY"), environment=Environment.STAGING)
mask_service = GenerateMaskService(hopter=hopter)
image_url = upload_image(image)
messages = [
{"type": "text", "text": instruction},
]
if image:
messages.append({"type": "image_url", "image_url": {"url": image_url}})
deps = ImageEditDeps(
edit_instruction=instruction,
image_url=image_url,
hopter_client=hopter,
mask_service=mask_service,
)
result = await image_edit_agent.run(messages, deps=deps)
# Extract the edited image URL from the tool return
for message in result.new_messages():
for part in message.parts:
if isinstance(part, ToolReturnPart) and isinstance(
part.content, EditImageResult
):
return part.content.edited_image_url
return None
async def use_edited_image(edited_image):
return edited_image
def clear_instruction():
# Only clear the instruction text.
return ""
# Create the Gradio interface
with gr.Blocks() as demo:
gr.Markdown("# PicEdit")
gr.Markdown("""
Welcome to PicEdit - an AI-powered image editing tool.
Simply upload an image and describe the changes you want to make in natural language.
""")
with gr.Row():
# Input image on the left
input_image = gr.Image(label="Original Image", type="filepath")
with gr.Column():
# Output image on the right
output_image = gr.Image(
label="Edited Image", type="filepath", interactive=False, scale=3
)
use_edited_btn = gr.Button("👈 Use Edited Image 👈")
# Text input for editing instructions
instruction = gr.Textbox(
label="Editing Instructions",
placeholder="Describe the changes you want to make to the image...",
)
# Clear button
with gr.Row():
clear_btn = gr.Button("Clear")
submit_btn = gr.Button("Apply Edit", variant="primary")
# Set up the event handlers
submit_btn.click(
fn=process_edit, inputs=[input_image, instruction], outputs=output_image
)
use_edited_btn.click(
fn=use_edited_image, inputs=[output_image], outputs=[input_image]
)
# Bind the clear button's click event to only clear the instruction textbox.
clear_btn.click(fn=clear_instruction, inputs=[], outputs=[instruction])
examples = gr.Examples(
examples=[
["https://i.ibb.co/qYwhcc6j/c837c212afbf.jpg", "remove the pole"],
["https://i.ibb.co/2Mrxztw/image.png", "replace the cat with a dog"],
[
"https://i.ibb.co/9mT4cvnt/resized-78-B40-C09-1037-4-DD3-9-F48-D73637-EE4-E51.png",
"ENHANCE!",
],
],
inputs=[input_image, instruction],
)
if __name__ == "__main__":
demo.launch()
|