update space

.gitignore

@@ -0,0 +1 @@
+/.env

app.py

@@ -1,24 +1,34 @@
 import os
-os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
-credentials_kwargs={"aws_access_key_id": os.environ["ACCESS_KEY"],"aws_secret_access_key": os.environ["SECRET_KEY"]}
-
-# work around: https://discuss.huggingface.co/t/how-to-install-a-specific-version-of-gradio-in-spaces/13552
-os.system("pip uninstall -y gradio")
-os.system("pip install gradio==3.4.1")
-os.system(os.environ["DD_ADDONS"])
-
 import time
+import importlib.metadata
 from os import getcwd, path, environ
 import deepdoctection as dd
 from deepdoctection.dataflow.serialize import DataFromList
-
 from dd_addons.extern import PdfTextDetector, PostProcessor, get_xsl_path
 from dd_addons.pipe.conn import PostProcessorService
 
 import gradio as gr
 from botocore.config import Config
+from dotenv import load_dotenv
+
+load_dotenv()
 
 
+def check_additional_requirements():
+    if not dd.detectron2_available():
+        os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
+    if importlib.util.find_spec("gradio") is not None:
+        if importlib.metadata.version("gradio")!="3.4.1":
+            os.system("pip uninstall -y gradio")
+            os.system("pip install gradio==3.4.1")
+    else:
+        os.system("pip install gradio==3.4.1")
+    os.system(os.environ["DD_ADDONS"])
+    return
+
+
+check_additional_requirements()
+# work around: https://discuss.huggingface.co/t/how-to-install-a-specific-version-of-gradio-in-spaces/13552
 _DD_ONE = "conf_dd_one.yaml"
 _XSL_PATH = get_xsl_path()
 
@@ -207,7 +217,7 @@ def analyze_image(img, pdf, max_datapoints):
         out = dp.as_dict()
         jsonl_out.append(out)
         out.pop("_image")
-        layout_items = dp.layouts
+        layout_items = [layout for layout in dp.layouts if layout.reading_order is not None]
         layout_items.sort(key=lambda x: x.reading_order)
         layout_items_str += f"\n\n -------- PAGE NUMBER: {dp.page_number+1} ------------- \n"
         for item in layout_items:

requirements.txt

@@ -1,3 +1,4 @@
+python-dotenv
 Pillow==9.5.0
 torch==1.12.0
 torchvision==0.13.0