feat: working gradio_demo.py

README.md

@@ -1,8 +1,10 @@
+---
 title: chat-image-editor
 emoji: 💬
 colorFrom: yellow
 colorTo: purple
 sdk: gradio
 sdk_version: 5.16.1
-app_file: gradio_app.py
+app_file: gradio_demo.py
 pinned: false
+---

gradio_demo.py

@@ -0,0 +1,92 @@
+import gradio as gr
+from src.agents.mask_generation_agent import mask_generation_agent, ImageEditDeps
+import os
+from src.hopter.client import Hopter, Environment
+from src.services.generate_mask import GenerateMaskService
+from dotenv import load_dotenv
+from src.utils import image_path_to_uri
+from pydantic_ai.messages import (
+    ToolCallPart,
+    ToolReturnPart
+)
+from src.agents.mask_generation_agent import EditImageResult
+from pydantic_ai.agent import Agent 
+from pydantic_ai.models.openai import OpenAIModel
+
+load_dotenv()
+
+async def process_edit(image, instruction):
+    hopter = Hopter(os.environ.get("HOPTER_API_KEY"), environment=Environment.STAGING)
+    mask_service = GenerateMaskService(hopter=hopter)
+    messages = [
+        {
+            "type": "text",
+            "text": instruction
+        },
+    ]
+    if image:
+        messages.append(
+            {"type": "image_url", "image_url": {"url": image_path_to_uri(image)}}
+        )
+    deps = ImageEditDeps(
+        edit_instruction=instruction,
+        image_url=image_path_to_uri(image),
+        hopter_client=hopter,
+        mask_service=mask_service
+    )
+    result = await mask_generation_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
+
+# 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")
+        # Output image on the right
+        with gr.Column():
+            output_image = gr.Image(label="Edited Image", type="filepath")
+            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..."
+    )
+    
+    # Submit button
+    submit_btn = gr.Button("Apply Edit")
+    
+    # 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]
+    )
+
+if __name__ == "__main__":
+    demo.launch()
+
+