import gradio as ui; from modules.engine_inpaint import *
feature = 'Image Inpainting'
'''
 _______  _______  _______  _____    _____    _______             __ 
|     __||_     _||    ___||     |_ |     |_ |   _   |    .---.-.|__|
|__     |  |   |  |    ___||       ||       ||       | __ |  _  ||  |
|_______|  |___|  |_______||_______||_______||___|___||__||___._||__|
_____________________________________________________________________

  Copyright © 2023-2024 Ikmal Said. All rights reserved.
   
  This program is the property of Ikmal Said. You may not reproduce
  distribute, or modify this code without the express permission of 
  the author, Ikmal Said.
_____________________________________________________________________
                                                                     
'''
with ui.Blocks(css=css, title=title(feature), theme=theme, analytics_enabled=False) as stella:
    
    with ui.Column():
        input_inpaint = ui.ImageEditor(label=ssource, brush=ui.Brush(colors=['#ffffff']), sources=['upload', 'webcam'])
        output_inpaint = ui.Gallery(label=sresults, object_fit="contain", height="60vh")
        prompt_inpaint = ui.Textbox(label="What to include:", show_copy_button=True, placeholder=spholder1)
        negative_inpaint = ui.Textbox(label="What not to include:", show_copy_button=True, placeholder=spholder2)
        number_inpaint = ui.Dropdown(label=snumber, choices=[1,2,3,4], value=4)
        prompt_example = ui.Examples(label="Suggested for you:", examples=['highly detailed face', 'detailed girl face', 'detailed man face', 'detailed hand', 'detailed leg', 'beautiful eyes'], inputs=[prompt_inpaint])
                
    with ui.Row():
        stop_inpaint = ui.Button("Cancel", variant="secondary", scale=1)
        clear_i2t = ui.ClearButton(value="Reset", components=[input_inpaint, output_inpaint, prompt_inpaint, negative_inpaint], scale=1)
        t2i_inpaint = ui.Button("Submit", variant="primary", scale=5)
                
    process_inpaint = t2i_inpaint.click(fn=quads_inpaint, inputs=[input_inpaint, prompt_inpaint, negative_inpaint, number_inpaint], outputs=[output_inpaint])
    stop_inpaint.click(fn=None, inputs=None, outputs=None, cancels=[process_inpaint])

if __name__ == "__main__":
    stella.queue(default_concurrency_limit=100, api_open=True).launch(inbrowser=True)