Added/Updated files

app.py

@@ -0,0 +1,33 @@
+import gradio as ui; from modules.engine_eraser import *
+feature = 'Image Eraser'
+'''
+ _______  _______  _______  _____    _____    _______             __ 
+|     __||_     _||    ___||     |_ |     |_ |   _   |    .---.-.|__|
+|__     |  |   |  |    ___||       ||       ||       | __ |  _  ||  |
+|_______|  |___|  |_______||_______||_______||___|___||__||___._||__|
+_____________________________________________________________________
+
+  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_eraser = ui.ImageEditor(label=ssource, brush=ui.Brush(colors=['#ffffff']), sources=['upload'], transforms=[])
+        output_eraser = ui.Gallery(preview=True, label=sresult, object_fit="contain", columns=1, rows=1, height="60vh")
+                
+    with ui.Row():
+        stop_eraser = ui.Button("Cancel", variant="secondary", scale=1)
+        clear_i2t = ui.ClearButton(value="Reset", components=[input_eraser, output_eraser], scale=1)
+        t2i_eraser = ui.Button("Submit", variant="primary", scale=3)
+                
+    process_eraser = t2i_eraser.click(fn=quads_eraser, inputs=[input_eraser], outputs=[output_eraser])
+    stop_eraser.click(fn=None, inputs=None, outputs=None, cancels=[process_eraser])
+
+if __name__ == "__main__":
+    stella.queue(default_concurrency_limit=100, api_open=True).launch(inbrowser=True)

modules/engine_eraser.py

@@ -0,0 +1,84 @@
+import gradio as ui; import requests; from requests.exceptions import Timeout
+from io import BytesIO; from PIL import Image; import concurrent.futures, time
+
+###################################################################################################################
+# Import from modules
+###################################################################################################################
+
+from modules.service_endpoints import *
+from modules.service_configs import *
+
+###################################################################################################################
+# Image erase process
+###################################################################################################################
+
+def eraser(input_image, input_mask):
+    image_pil = Image.fromarray(input_image)
+    mask_pil = Image.fromarray(input_mask)
+    image_bytes = BytesIO()
+    mask_bytes = BytesIO()
+    image_pil.save(image_bytes, format='PNG')
+    mask_pil.save(mask_bytes, format='PNG')
+    
+    print(receive())
+    
+    payload = {
+        'model_version': (None, '1'),
+        'cfg': (None, '9.5'),
+        'priority': (None, '1'),
+    }
+    
+    data = {
+        'image': ('input_image.png', image_bytes.getvalue(), 'image/png'),
+        'mask': ('input_mask.png', mask_bytes.getvalue(), 'image/png')
+    }
+
+    try:
+        response = requests.post(mode['eraser'], headers=head, data=payload, files=data, timeout=(None, None))
+
+        if len(response.content) < 65 * 1024:
+            print(reject())
+            return None
+        
+        print(done())
+        return Image.open(BytesIO(response.content))
+    
+    except Timeout:
+        print(timeout())
+        return None
+
+###################################################################################################################
+# 1 image for each generation
+###################################################################################################################
+
+def quads_eraser(a, progress=ui.Progress()):
+    quantities = 1
+    result_list = [None] * quantities
+    percent = 0
+
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        futures = []
+
+        for i in range(quantities):
+            future = executor.submit(lambda x: eraser(a["background"], a["layers"][0]), i)
+            futures.append(future)
+            multiplier = 0.99 / quantities
+            percent += multiplier
+            progress(percent, desc=f"Processing image {i + 1} of {quantities}")
+            time.sleep(0.25)
+
+    for i, future in enumerate(futures):
+        result = future.result()
+        result_list[i] = result
+
+    successful_results = [result for result in result_list if result is not None]
+
+    if len(successful_results) < quantities:
+        if quantities == 1:
+            ui.Warning(message=single_error)
+        else:
+            ui.Warning(message=quads_error)
+    else:
+        ui.Info(message=success)
+
+    return successful_results

modules/service_configs.py

@@ -0,0 +1,46 @@
+import gradio as ui; from datetime import datetime; import logging
+
+# enable debug
+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};"
+
+# 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.'
+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!'
+spholder1       = 'Elements to add into the image!'
+spholder2       = 'Things to get rid of!'
+spholder3       = 'Sprinkle some wonders to the generated prompt!'
+sprompt         = 'Generate images of:'
+sprompt1        = 'Based on image, create:'
+smodel          = 'Using the AI model:'
+smode           = 'Using the mode:'
+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')}]"
+def receive():          return f"{timestamp()} \U0001F680 {received}"
+def timeout():          return f"{timestamp()} \U000023F0 {timed}"
+def reject():           return f"{timestamp()} \U0000274C {rejected}"
+def done():             return f"{timestamp()} \U0001F618 {complete}"
+def header(feature):    ui.HTML(f'<center><h4 style="font-size: 1em; margin: 5px 0px 5px">{feature}</h4></center>')
+def footer():           ui.HTML(f'<center><h4 style="font-size: 1em; margin: 5px 0px 0px">{liability}<br></h4>{rights}</center>')
+def title(feature):     return f"{feature}"

modules/service_endpoints.py

@@ -0,0 +1,9 @@
+import os
+
+mode = {
+    'eraser'    : os.getenv('eraser')
+}
+
+head = {
+    'bearer'    : os.getenv('bearer')
+}