fix: output not shown for all input docs

main.py

@@ -23,16 +23,17 @@ def t5(input):
 
 @app.post("/classify")
 async def classify_doc(files: List[UploadFile] = File(...)):
-    for file in files:
-        try:
-            contents = file.file.read()
-            classify_res = classify_acct_dtype_str(BytesIO(contents))
-        except Exception as err:
-            print(Exception, err)
-            return {"message": "There was an error in uploading file(s)"}
-        finally:
-            file.file.close()
-    return {"message": f"{[file.filename for file in files]} : {[classify_res for file in files]}"}
+#    for file in files:
+#        try:
+#            contents = file.file.read()
+#            filename = file.filename
+#            classify_res = classify_acct_dtype_str(contents, filename)
+#        except Exception as err:
+#            print(Exception, err)
+#            return {"message": "There was an error in uploading file(s)"}
+#        finally:
+#            file.file.close()
+    return {"message": f"{[file.filename for file in files]} : {[getDocClassType(file) for file in files]}"}
 
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")
@@ -42,6 +43,19 @@ def index() -> FileResponse:
     return FileResponse(path="/app/static/index.html", media_type="text/html")
 
 
+def getDocClassType(file):
+    classify_res = ""
+    try:
+            contents = file.file.read()
+            filename = file.filename
+            classify_res = classify_acct_dtype_str(contents, filename)
+    except Exception as err:
+        print(Exception, err)
+        return {"Error in reading file."}
+    finally:
+        file.file.close()
+
+    return classify_res
 
 
 import re
@@ -146,10 +160,9 @@ import os
 import requests
 from io import BytesIO
 
-def classify_acct_dtype_str(input_path):
-    response = requests.get("https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/city-streets.jpg")
-    ipt = Image.open(BytesIO(response.content))
-    dtype_inf, dtype_conf = doctype_classify(ipt, "city-streets.jpg")
+def classify_acct_dtype_str(content, filename):
+    ipt = Image.open(BytesIO(content))
+    dtype_inf, dtype_conf = doctype_classify(ipt, filename)
 
     return dtype_inf
 

requirements.txt

@@ -2,6 +2,7 @@ fastapi
 python-multipart
 typing
 uvicorn[standard]
+pdf2image==1.16.3
 aiohttp==3.9.1
 aiosignal==1.3.1
 altgraph==0.17.4