added/updated app files

app.py

@@ -1,10 +1,6 @@
 import gradio as ui; from modules.engine_inpaint import *
 feature = 'Image Inpainting'
 '''
- _______  _______  _______  _____    _____    _______             __ 
-|     __||_     _||    ___||     |_ |     |_ |   _   |    .---.-.|__|
-|__     |  |   |  |    ___||       ||       ||       | __ |  _  ||  |
-|_______|  |___|  |_______||_______||_______||___|___||__||___._||__|
 _____________________________________________________________________
 
   Copyright © 2023-2024 Ikmal Said. All rights reserved.
@@ -16,22 +12,34 @@ _____________________________________________________________________
                                                                      
 '''
 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.Tabs() as tabs_inpaint:
+        with ui.TabItem(title(feature), id=0):
+            with ui.Group():
+                input_inpaint = ui.ImageEditor(
+                    show_label=False, transforms=[],
+                    brush=ui.Brush(default_size=24, colors=['#ffffff'], color_mode=['fixed']),
+                    eraser=ui.Eraser(default_size=24), sources=['upload', 'webcam'])
                 
-    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)
+                prompt_inpaint = ui.Textbox(label="What to include:", show_copy_button=True, placeholder=spholder1)
+                                    
+                with ui.Row():
+                    submit_inpaint = ui.Button("Submit", variant="primary")
                 
-    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])
+            prompt_example = ui.Examples(label="AI Enhance keywords:", examples=['highly detailed face', 'detailed girl face', 'detailed man face', 'detailed hand', 'detailed leg', 'beautiful eyes'], inputs=[prompt_inpaint])      
+            with ui.Accordion('Advanced', open=False):
+                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, filterable=False)
+
+        with ui.TabItem('Result', id=1):
+            with ui.Group():
+                output_inpaint = ui.Gallery(preview=True, show_label=False, object_fit="contain", height="43.4vh", show_share_button=False)  
+                
+                with ui.Row():
+                    clear_inpaint = ui.ClearButton(value="Reset", components=[input_inpaint, output_inpaint, prompt_inpaint, negative_inpaint])
+                
+    process_inpaint = submit_inpaint.click(fn=inpaint2, inputs=[input_inpaint, prompt_inpaint, negative_inpaint, number_inpaint], outputs=[tabs_inpaint, input_inpaint, output_inpaint])
+    prompt_inpaint.submit(fn=inpaint2, inputs=[input_inpaint, prompt_inpaint, negative_inpaint, number_inpaint], outputs=[tabs_inpaint, input_inpaint, output_inpaint])
+    clear_inpaint.click(fn=reset, inputs=None, outputs=tabs_inpaint)
 
 if __name__ == "__main__":
-    stella.queue(default_concurrency_limit=100, api_open=True).launch(inbrowser=True)
+    stella.queue(default_concurrency_limit=100, api_open=True).launch(inbrowser=True, favicon_path="favicon.ico")

modules/engine_inpaint.py

@@ -26,7 +26,7 @@ def inpaint(input_image, input_mask, prompt, negative):
     payload = {
         'model_version': (None, '1'),
         'prompt': (None, prompt),
-        'neg_prompt': (None, negative),
+        'neg_prompt': (None, f"({negative}), hands, face, eyes, legs"),
         'inpaint_strength': (None, '0.75'),
         'cfg': (None, '9.5'),
         'priority': (None, '1'),
@@ -51,6 +51,11 @@ def inpaint(input_image, input_mask, prompt, negative):
         print(timeout())
         return None
 
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        ui.Warning(message=single_error)
+        return None
+
 ###################################################################################################################
 # 4 image for each generation
 ###################################################################################################################
@@ -82,7 +87,11 @@ def quads_inpaint(a, b, c, d, progress=ui.Progress()):
             ui.Warning(message=single_error)
         else:
             ui.Warning(message=quads_error)
-    else:
-        ui.Info(message=success)
 
-    return successful_results
+    return successful_results
+
+def inpaint2(a, b, c, d):
+    return ui.Tabs(selected=1), a, quads_inpaint(a, b, c, d)
+
+def reset():
+    return ui.Tabs(selected=0)

modules/service_configs.py

@@ -4,25 +4,60 @@ import gradio as ui; from datetime import datetime; import logging
 logging.basicConfig(level=logging.DEBUG)
 
 # global theme
-theme = ui.themes.Base(
-    font=[ui.themes.GoogleFont('Segoe UI'), 'system-ui', 'sans-serif'],
-    text_size=ui.themes.Size(lg="15px", md="15px", sm="15px", xl="15px", xs="15px", xxl="15px", xxs="15px"),
-    primary_hue='rose', secondary_hue='rose', neutral_hue='zinc'); css = "footer {visibility: hidden};"
+theme = ui.themes.Default(
+    font=[ui.themes.GoogleFont('Roboto Condensed'), 'Arial'],
+    text_size=ui.themes.Size(lg="16px", md="16px", sm="16px", xl="16px", xs="16px", xxl="16px", xxs="16px"),
+    primary_hue='rose', secondary_hue='rose', neutral_hue='zinc', radius_size=ui.themes.sizes.radius_none)
+
+css = '''
+footer {
+    display: none !important;
+}
+
+div.svelte-vt1mxs>*, div.svelte-vt1mxs>.form > * {
+    border: none;
+}
+
+.app.svelte-182fdeq.svelte-182fdeq {
+    padding: 0px !important;
+}
+
+.grid-wrap.svelte-hpz95u.svelte-hpz95u {
+    overflow-y: auto !important;
+}
+
+.selected.svelte-1uw5tnk {
+    background: #262629 !important;
+}
+
+.closed.svelte-g7pfx4>button.svelte-g7pfx4 {
+    display: none !important;
+}
+
+gradio-app {
+    background: none !important;
+}
+
+input.svelte-1f354aw.svelte-1f354aw, textarea.svelte-1f354aw.svelte-1f354aw {
+    text-align: justify !important;
+}
+'''
 
 # global locale - english
 success         = 'That worked successfully!'
-single_error    = 'That did not work. Please check and try again.'
-quads_error     = 'Some images cannot be processed. Please check and try again.'
+single_error    = 'That did not work!'
+quads_error     = 'Some images cannot be processed!'
+empty_error     = 'Prompt field cannot be empty!'
 received        = 'Request Received'
 timed           = 'Request Timeout'
 rejected        = 'Request Error/Rejected'
 complete        = 'Request Completed'
 liability       = 'STELLA can make mistakes and inaccuracies.'
 rights          = '© 2023-2024 Ikmal Said. All rights reserved.'
-spholder        = 'Imagine your favorite person, places or anything!'
+spholder        = 'Imagine a person, place or anything!'
 spholder1       = 'Elements to add into the image!'
 spholder2       = 'Things to get rid of!'
-spholder3       = 'Sprinkle some wonders to the generated prompt!'
+spholder3       = 'Put wonders into the generated prompt!'
 sprompt         = 'Generate images of:'
 sprompt1        = 'Based on image, create:'
 smodel          = 'Using the AI model:'
@@ -31,9 +66,6 @@ sratio          = 'In the size of:'
 sstyle          = 'Inspired by the style of:'
 squality        = 'At a quality level of:'
 snumber         = 'With a quantity of:'
-ssource         = 'Source'
-sresult         = 'Result'
-sresults        = 'Results'
 
 # global function
 def timestamp():        return f"[{datetime.now().strftime('%d/%m/%y at %H:%M:%S')}]"