WIP

app.py

@@ -5,8 +5,40 @@ import base64
 from io import BytesIO
 from PIL import Image
 
-def search_face(image):
-    return image, 1
+count = 0
+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):
+    global count    
+    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 [], count
+    
+    try:
+        res = r.json().get('img_array')
+        out_array = []
+        for item in res:
+            out_array.append((base64_to_image(item["image"]), item["url"] + "*********"))
+        count += 1            
+        return out_array, count
+    except:
+        raise gr.Error("Try again.")
+
 
 with gr.Blocks() as demo:
     gr.Markdown(