修正已知错误

util/model_util.py

@@ -35,9 +35,10 @@ def ie_settlement(img_path, layout):
     url = 'http://ie_settlement:5002'
     response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)})
     if response.status_code == 200:
-        return response.json()
-    else:
-        return None
+        ie_result = response.json()
+        if ie_result.get(0):
+            return ie_result[0]
+    return None
 
 
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
@@ -51,9 +52,10 @@ def ie_settlement_text(text):
     url = 'http://ie_settlement:5002/text'
     response = requests.post(url, {'text': text})
     if response.status_code == 200:
-        return response.json()
-    else:
-        return None
+        ie_result = response.json()
+        if ie_result.get(0):
+            return ie_result[0]
+    return None
 
 
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
@@ -68,9 +70,10 @@ def ie_discharge(img_path, layout):
     url = 'http://ie_discharge:5003'
     response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)})
     if response.status_code == 200:
-        return response.json()
-    else:
-        return None
+        ie_result = response.json()
+        if ie_result.get(0):
+            return ie_result[0]
+    return None
 
 
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
@@ -84,9 +87,10 @@ def ie_discharge_text(text):
     url = 'http://ie_discharge:5003/text'
     response = requests.post(url, {'text': text})
     if response.status_code == 200:
-        return response.json()
-    else:
-        return None
+        ie_result = response.json()
+        if ie_result.get(0):
+            return ie_result[0]
+    return None
 
 
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
@@ -101,9 +105,10 @@ def ie_cost(img_path, layout):
     url = 'http://ie_cost:5004'
     response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)})
     if response.status_code == 200:
-        return response.json()
-    else:
-        return None
+        ie_result = response.json()
+        if ie_result.get(0):
+            return ie_result[0]
+    return None
 
 
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
@@ -117,9 +122,10 @@ def ie_cost_text(text):
     url = 'http://ie_cost:5004/text'
     response = requests.post(url, {'text': text})
     if response.status_code == 200:
-        return response.json()
-    else:
-        return None
+        ie_result = response.json()
+        if ie_result.get(0):
+            return ie_result[0]
+    return None
 
 
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,