x

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+from utils import PingApi
+
+
+def inference(img):
+    papi = PingApi()
+    rsp = papi.ping_api(img, 'vin-mining')
+
+    gathered_data = rsp['pages'][0]['gathered_data']
+    parsed_rsp = {
+        gathered_data['fields']['VIN']['texts'][0]['text']: gathered_data['fields']['VIN']['texts'][0]['confidence']
+    }
+
+    return parsed_rsp
+
+
+title = "Car VIN mining"
+description = "Extract VIN code of vehicle, if present on the image"
+examples = [['./examples/ex1.jpg'], ['./examples/ex2.jpg'], ['./examples/ex3.jpg']]
+gr.Interface(
+    inference,
+    inputs=gr.Image(type="filepath"),
+    outputs=[gr.Label(num_top_classes=1, label='Output')],
+    title=title,
+    description=description,
+    examples=examples
+).launch()

requirements.txt

@@ -0,0 +1,2 @@
+requests
+gradio

utils.py

@@ -0,0 +1,37 @@
+import os
+import requests
+
+
+class PingApi():
+    def __init__(self):
+        self.url = os.getenv('API_URL')
+        self.auth_url = os.getenv('AUTH_URL')
+        client_id = os.getenv('CLIENT_ID')
+        client_secret = os.getenv('CLIENT_SECRET')
+        self.auth_payload = f'grant_type=client_credentials&client_id={client_id}&client_secret={client_secret}&scope=categorization'
+        self.access_token = self.authorize()
+
+    def authorize(self):
+        headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+        response = requests.request("POST", self.auth_url, headers=headers, data=self.auth_payload)
+        try:
+            access_token = response.json()['access_token']
+        except Exception as e:
+            print(response.text)
+            raise e
+        return access_token
+
+    def ping_api(self, f_path, endpoint):
+        payload = {}
+        headers = {
+            'Authorization': f"Bearer {self.access_token}"
+        }
+        files = [('file', (os.path.basename(f_path), open(f_path, 'rb'), 'image/jpeg'))]
+        response = requests.request("POST", self.url + '/' + endpoint, headers=headers, data=payload, files=files)
+        try:
+            rsp_json = response.json()
+        except Exception as e:
+            print(response.text)
+            raise e
+
+        return rsp_json