v1

app.py

@@ -9,12 +9,6 @@ import configparser
 
 
 def run(file):
-    backend_url = os.getenv('BACKEND_URL')
-    url = f'{backend_url}/raster-to-vector-base64'
-    out_json = {'json': url}
-
-
-    return out_json
     in_image = cv2.imread(file)
 
     encode_img = cv2.imencode('.jpg', in_image)[1].tostring()
@@ -23,17 +17,15 @@ def run(file):
 
     backend_url = os.getenv('BACKEND_URL')
     url = f'{backend_url}/raster-to-vector-base64'
-    out_json = {'json': url}
-    out_img = in_image
-    # payload = {'image': base64_img}
-    # image_request = requests.post(url, json=payload)
-    #
-    # out_img = image_request.json()['image']
-    # out_json = image_request.json()['graph']
-    #
-    # decode_img = base64.b64decode(out_img.split(',')[1])
-    # decode_img = np.frombuffer(decode_img, dtype=np.uint8)
-    # out_img = cv2.imdecode(decode_img, flags=cv2.IMREAD_COLOR)
+    payload = {'image': base64_img}
+    image_request = requests.post(url, json=payload)
+
+    out_img = image_request.json()['image']
+    out_json = image_request.json()['graph']
+
+    decode_img = base64.b64decode(out_img.split(',')[1])
+    decode_img = np.frombuffer(decode_img, dtype=np.uint8)
+    out_img = cv2.imdecode(decode_img, flags=cv2.IMREAD_COLOR)
 
     return out_img, out_json
 
@@ -54,7 +46,7 @@ with gr.Blocks() as demo:
                             inputs=app_input)
                 start_button = gr.Button("Run")
             with gr.Column():
-                app_output = [gr.JSON()]
+                app_output = [gr.Image(type="numpy"), gr.JSON()]
 
         start_button.click(run, inputs=app_input, outputs=app_output)