Update app.py

app.py

@@ -227,6 +227,14 @@ def np2base64(image_np):
     return base64_str
 
 
+def img2base64(path):
+    if path.endswith:
+        base64_str = np2base64(load_document(path)[0])
+    else:
+        base64_str = base64.b64encode(open(path, 'rb').read()).decode()
+    return base64_str
+
+
 def process_prompt(prompt, document, lang="ch"):
     if not prompt:
         prompt = "What is the total actual and/or obligated expenses of ECG Center?"
@@ -237,8 +245,7 @@ def process_prompt(prompt, document, lang="ch"):
 
     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])
+    base64_str = img2base64(document)
 
     r = requests.post(url, json={"doc": base64_str, "prompt": [prompt], "lang": lang})
     response = r.json()
@@ -404,7 +411,7 @@ with gr.Blocks(css=CSS) as demo:
             gr.Markdown("## 2. Make a request")
             prompt = gr.Textbox(
                 label="Prompt",
-                placeholder="What is the total actual and/or obligated expenses of ECG Center?",
+                placeholder="e.g. What is the total actual and/or obligated expenses of ECG Center?",
                 lines=1,
                 max_lines=1,
             )