add examples

.gitignore

@@ -1 +1,5 @@
 .idea/
+backs/
+gray/
+lowres/
+scratch/

app.py

@@ -1,5 +1,9 @@
+import zipfile
+from pathlib import Path
+
 import gradio as gr
 from gradio_client import Client
+from huggingface_hub import hf_hub_download
 
 clientGFPGAN = Client("leonelhs/GFPGAN")
 clientSuperface = Client("leonelhs/superface")
@@ -9,7 +13,14 @@ clientEnhanceLight = Client("leonelhs/Zero-DCE")
 clientZeroBackground = Client("leonelhs/ZeroBackground")
 clientFaceParser = Client("leonelhs/faceparser")
 
-context = dict()
+REPO_ID = "leonelhs/faceshine"
+
+
+def download(file):
+    file_zip = hf_hub_download(repo_id=REPO_ID, filename=file + ".zip")
+    with zipfile.ZipFile(file_zip, "r") as zip_ref:
+        zip_ref.extractall(file)
+    return list(Path(file).iterdir())
 
 
 def gfpgan_face(image):
@@ -44,6 +55,10 @@ def mirror(x):
     return x
 
 
+def active_first():
+    return gr.Tabs.update(selected=0)
+
+
 footer = r"""
 <center>
 <p>This App is running on a CPU, help us to upgrade a GPU or just give us a <a href='https://github.com/leonelhs/face-shine' target='_blank'>Github ⭐</a></p>
@@ -56,55 +71,61 @@ footer = r"""
 <center><span>leonelhs@gmail.com</span></center>
 """
 
-with gr.Blocks() as app:
-    gr.Request()
-
+with gr.Blocks(title="Face Shine") as app:
     with gr.Row():
         gr.HTML("<center><h1>Face Shine</h1></center>")
 
-    with gr.Tab("Photo restorer"):
-        with gr.Row():
-            with gr.Column(scale=1):
-                btn_hires = gr.Button(value="Enhance resolution")
-                btn_eraser = gr.Button(value="Erase scratches")
-                btn_color = gr.Button(value="Colorize photo")
-                btn_light = gr.Button(value="Enhance light")
-                with gr.Accordion("New background", open=False):
-                    img_newbgr = gr.Image(label="Left empty for a transparent background", type="filepath")
-                    btn_newbgr = gr.Button(value="Change background")
-
-            with gr.Column(scale=4):
-                with gr.Row():
-                    img_input = gr.Image(label="Input", type="filepath")
-                    img_output = gr.Image(label="Result", type="filepath", interactive=False)
-                with gr.Row():
-                    gr.ClearButton([img_input, img_output, img_newbgr], variant="stop")
-                    btn_swap = gr.Button(value="Swap images", variant="primary")
-
-    with gr.Tab("Settings"):
-        with gr.Accordion("Image restoration settings", open=False):
-            restorer = gr.Dropdown([
-                'RealESRGAN_x2plus',
-                'RealESRGAN_x4plus',
-                'RealESRNet_x4plus',
-                'AI-Forever_x2plus',
-                'AI-Forever_x4plus',
-                'RealESRGAN_x4plus_anime_6B',
-                'realesr-animevideov3',
-                'realesr-general-x4v3'],
-                type="value", value='RealESRGAN_x4plus', label='General restoration algorithm', info="version")
-            enhancer = gr.Dropdown([
-                'No additional face process',
-                'GFPGANv1.2',
-                'GFPGANv1.3',
-                'GFPGANv1.4',
-                'RestoreFormer'],
-                type="value", value='No additional face process', label='Special face restoration algorithm',
-                info="version")
-            rescale = gr.Dropdown(["1", "2", "3", "4"], type="value", value="4", label="Rescaling factor")
-        with gr.Accordion("Logs info", open=False):
-            text_logger = gr.Textbox(label="login", lines=5, show_label=False)
-        gr.Button("Save settings")
+    with gr.Tabs() as tabs:
+        with gr.TabItem("Photo restorer", id=0):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    btn_hires = gr.Button(value="Enhance resolution")
+                    btn_eraser = gr.Button(value="Erase scratches")
+                    btn_color = gr.Button(value="Colorize photo")
+                    btn_light = gr.Button(value="Enhance light")
+                    with gr.Accordion("New background", open=False):
+                        img_newbgr = gr.Image(label="Left empty for a transparent background", type="filepath")
+                        btn_newbgr = gr.Button(value="Clear background")
+
+                with gr.Column(scale=4):
+                    with gr.Row():
+                        img_input = gr.Image(label="Input", type="filepath")
+                        img_output = gr.Image(label="Result", type="filepath", interactive=False)
+                    with gr.Row():
+                        gr.ClearButton([img_input, img_output, img_newbgr], variant="stop")
+                        btn_swap = gr.Button(value="Swap images", variant="primary")
+
+        with gr.TabItem("Examples", id=1):
+            gr.Examples(examples=download("lowres"), inputs=[img_input], label="Low resolution")
+            gr.Examples(examples=download("gray"), inputs=[img_input], label="Gray scale")
+            gr.Examples(examples=download("scratch"), inputs=[img_input], label="Scratched")
+            gr.Examples(examples=download("backs"), inputs=[img_newbgr], label="Backgrounds")
+            gr.Button(value="Ok", variant="primary").click(active_first, None, tabs)
+
+        with gr.TabItem("Settings", id=2):
+            with gr.Accordion("Image restoration settings", open=False):
+                restorer = gr.Dropdown([
+                    'RealESRGAN_x2plus',
+                    'RealESRGAN_x4plus',
+                    'RealESRNet_x4plus',
+                    'AI-Forever_x2plus',
+                    'AI-Forever_x4plus',
+                    'RealESRGAN_x4plus_anime_6B',
+                    'realesr-animevideov3',
+                    'realesr-general-x4v3'],
+                    type="value", value='RealESRGAN_x4plus', label='General restoration algorithm', info="version")
+                enhancer = gr.Dropdown([
+                    'No additional face process',
+                    'GFPGANv1.2',
+                    'GFPGANv1.3',
+                    'GFPGANv1.4',
+                    'RestoreFormer'],
+                    type="value", value='No additional face process', label='Special face restoration algorithm',
+                    info="version")
+                rescale = gr.Dropdown(["1", "2", "3", "4"], type="value", value="2", label="Rescaling factor")
+            with gr.Accordion("Logs info", open=False):
+                text_logger = gr.Textbox(label="login", lines=5, show_label=False)
+            gr.Button("Save settings")
 
     btn_hires.click(enhance_face, inputs=[img_input, restorer, enhancer, rescale],
                     outputs=[img_output, text_logger])
@@ -117,4 +138,4 @@ with gr.Blocks() as app:
     with gr.Row():
         gr.HTML(footer)
 
-app.launch(share=False, debug=True, enable_queue=True)
+app.launch(share=False, debug=True, enable_queue=True, show_error=True)

requirements.txt

@@ -1 +1 @@
-gradio
+gradio~=3.35.2