Upload 3 files

app.py

@@ -0,0 +1,52 @@
+import os
+import gradio as gr
+import requests
+import base64
+from io import BytesIO
+from PIL import Image
+
+def image_to_base64(image):
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    return base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+def base64_to_image(base64_str):
+    return Image.open(BytesIO(base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))))
+
+def search_face(file):
+    url = os.environ.get("SERVER_URL")
+
+    try:
+        image = Image.open(file)
+        image_base64 = image_to_base64(image)
+        r = requests.post(url=url, json={"token": os.environ.get("ACCESS_TOKEN"), "type": "novip", "image": image_base64})
+    except:
+        raise gr.Error("Please select image file!")
+    
+    if r.text == "No matches":
+        gr.Info("No images found.")
+        return []
+    
+    try:
+        res = r.json().get('img_array')
+        out_array = []
+        for item in res:
+            out_array.append((base64_to_image(item["image"]), item["url"] + "*********"))
+        return out_array
+    except:
+        raise gr.Error("Try again.")
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column(scale=1):
+            image = gr.Image(type='filepath', height=480)
+            gr.Examples(['examples/1.jpg', 'examples/2.jpg'], 
+                        inputs=image)
+            search_face_button = gr.Button("Search Face")
+        with gr.Column(scale=2):
+            output = gr.Gallery(label="Found Images", columns=[4], object_fit="contain", height="auto")
+
+    search_face_button.click(search_face, inputs=[image], outputs=[output])
+
+demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False)