Update app.py

app.py

@@ -177,17 +177,17 @@ def load_document(path):
             images_list.append(img_np)
     else:
         image = Image.open(b)
-        images_list = [image.convert("RGB")]
+        images_list = [np.array(image.convert("RGB"))]
     return images_list
 
 def process_path(path):
     error = None
     if path:
         try:
-            img = load_document(path)
+            images_list = load_document(path)
             return (
                 path,
-                gr.update(visible=True, value=img),
+                gr.update(visible=True, value=images_list),
                 gr.update(visible=True),
                 gr.update(visible=False, value=None),
                 gr.update(visible=False, value=None),
@@ -221,9 +221,9 @@ def process_upload(file):
         )
 
 
-def img_base64(img_path):
-    with open(img_path, "rb") as f:
-        base64_str = base64.b64encode(f.read())
+def np2base64(image_np):
+    image = cv2.imencode('.jpg', image_np)[1]
+    base64_str = str(base64.b64encode(image))[2:-1]
     return base64_str
 
 
@@ -236,8 +236,11 @@ def process_prompt(prompt, document, lang="ch"):
     access_token = os.environ['token']
 
     url = f"https://aip.baidubce.com/rpc/2.0/nlp-itec/poc/docprompt?access_token={access_token}"
+    
+    image_list = load_document(document)
+    base64_str = np2base64(image_list[0])
 
-    r = requests.post(url, json={"doc": img_base64(document).decode(), "prompt": [prompt], "lang": lang})
+    r = requests.post(url, json={"doc": base64_str, "prompt": [prompt], "lang": lang})
     response = r.json()
     
     predictions = response['result']
@@ -495,4 +498,5 @@ with gr.Blocks(css=CSS) as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(enable_queue=False)
+    # demo.launch(enable_queue=False)
+    demo.launch(enable_queue=True, server_name="10.21.226.184", server_port=8072, share=False)