Update app.py

app.py

@@ -22,14 +22,14 @@ def download_image(url):
     response = requests.get(url)
     return Image.open(BytesIO(response.content)).convert("RGB")
 
-def eraser_api_call(image_base64_file, mask_base64_file, mask_type):
+def gen_fill_api_call(image_base64_file, mask_base64_file, prompt):
 
-    url = "http://engine.prod.bria-api.com/v1/eraser"
+    url = "http://engine.int.bria-api.com/v1/gen_fill"
     
     payload = {
     "file": image_base64_file,
     "mask_file": mask_base64_file,
-    "mask_type": mask_type,
+    "prompt": prompt,
     }
     response = requests.post(url, json=payload, headers=auth_headers)
     response = response.json()
@@ -47,7 +47,7 @@ def predict(dict):
     mask_base64_file = convert_mask_image_to_base64_string(mask)
     
     mask_type = "manual"
-    gen_img = eraser_api_call(image_base64_file, mask_base64_file, mask_type)
+    gen_img = gen_fill_api_call(image_base64_file, mask_base64_file, mask_type)
     
     return gen_img
 
@@ -102,12 +102,12 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
 image_blocks = gr.Blocks(css=css, elem_id="total-container")
 with image_blocks as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("## BRIA Eraser API")
+        gr.Markdown("## BRIA Generative Fill API")
         gr.HTML('''
           <p style="margin-bottom: 10px; font-size: 94%">
-            This demo showcases the BRIA Eraser capability, which allows users to remove specific elements or objects from images.<br>
+            This demo showcases the BRIA Generative Fill capability, which allows users to remove specific elements or objects from images.<br>
             The pipeline comprises multiple components, including <a href="https://huggingface.co/briaai/BRIA-2.3" target="_blank">briaai/BRIA-2.3</a>, 
-            <a href="https://huggingface.co/briaai/BRIA-2.3-ControlNet-Inpainting" target="_blank">briaai/BRIA-2.3-ControlNet-Inpainting</a>, 
+            <a href="https://huggingface.co/briaai/BRIA-2.3-ControlNet-Generative-Fill" target="_blank">briaai/BRIA-2.3-ControlNet-Generative-Fill</a>, 
             and <a href="https://huggingface.co/briaai/BRIA-2.3-FAST-LORA" target="_blank">briaai/BRIA-2.3-FAST-LORA</a>, all trained on licensed data.<br>
             This ensures full legal liability coverage for copyright and privacy infringement.<br>
             Notes:<br>
@@ -115,21 +115,22 @@ with image_blocks as demo:
             - For multiple masks, results are better if all masks are included in inference.<br>
           </p>
         ''')
+
     with gr.Row():
         with gr.Column():
             image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
                                    brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
                                    )
+            prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
             with gr.Row(elem_id="prompt-container", equal_height=True):
-                with gr.Column():  # Wrap the button inside a Column
-                    btn = gr.Button("Erase!", elem_id="run_button")
+                with gr.Column():
+                    btn = gr.Button("Fill!", elem_id="run_button")
         
         with gr.Column():
             image_out = gr.Image(label="Output", elem_id="output-img")
 
-    # Button click will trigger the inpainting function (no prompt required)
-    btn.click(fn=predict, inputs=[image], outputs=[image_out], api_name='run')
-    
+    # Button click will trigger the inpainting function (now with prompt included)
+    btn.click(fn=predict, inputs=[image, prompt], outputs=[image_out], api_name='run')
 
     gr.HTML(
         """
@@ -140,4 +141,4 @@ with image_blocks as demo:
         """
     )
 
-image_blocks.queue(max_size=25,api_open=False).launch(show_api=False)
+image_blocks.queue(max_size=25, api_open=False).launch(show_api=False)