From 8f208c5330ab2c8589dcde378911dbb3d1f0137a Mon Sep 17 00:00:00 2001
From: Alexey Taymanov <ataymano@gmail.com>
Date: Thu, 27 Mar 2025 20:40:40 +0000
Subject: [PATCH 1/2] util -> omniparser

---
 gradio_demo.py                                | 2 +-
 {util => omniparser}/__init__.py              | 0
 {util => omniparser}/box_annotator.py         | 0
 {util => omniparser}/omniparser.py            | 2 +-
 {util => omniparser}/utils.py                 | 2 +-
 omnitool/omniparserserver/omniparserserver.py | 2 +-
 6 files changed, 4 insertions(+), 4 deletions(-)
 rename {util => omniparser}/__init__.py (100%)
 rename {util => omniparser}/box_annotator.py (100%)
 rename {util => omniparser}/omniparser.py (93%)
 rename {util => omniparser}/utils.py (99%)

diff --git a/gradio_demo.py b/gradio_demo.py
index 15664d31..2426d4e9 100644
--- a/gradio_demo.py
+++ b/gradio_demo.py
@@ -8,7 +8,7 @@
 
 
 import base64, os
-from util.utils import check_ocr_box, get_yolo_model, get_caption_model_processor, get_som_labeled_img
+from omniparser.utils import check_ocr_box, get_yolo_model, get_caption_model_processor, get_som_labeled_img
 import torch
 from PIL import Image
 
diff --git a/util/__init__.py b/omniparser/__init__.py
similarity index 100%
rename from util/__init__.py
rename to omniparser/__init__.py
diff --git a/util/box_annotator.py b/omniparser/box_annotator.py
similarity index 100%
rename from util/box_annotator.py
rename to omniparser/box_annotator.py
diff --git a/util/omniparser.py b/omniparser/omniparser.py
similarity index 93%
rename from util/omniparser.py
rename to omniparser/omniparser.py
index 536385e6..1d0e28da 100644
--- a/util/omniparser.py
+++ b/omniparser/omniparser.py
@@ -1,4 +1,4 @@
-from util.utils import get_som_labeled_img, get_caption_model_processor, get_yolo_model, check_ocr_box
+from omniparser.utils import get_som_labeled_img, get_caption_model_processor, get_yolo_model, check_ocr_box
 import torch
 from PIL import Image
 import io
diff --git a/util/utils.py b/omniparser/utils.py
similarity index 99%
rename from util/utils.py
rename to omniparser/utils.py
index eb7c8b25..c819dc86 100644
--- a/util/utils.py
+++ b/omniparser/utils.py
@@ -41,7 +41,7 @@
 from torchvision.transforms import ToPILImage
 import supervision as sv
 import torchvision.transforms as T
-from util.box_annotator import BoxAnnotator 
+from omniparser.box_annotator import BoxAnnotator 
 
 
 def get_caption_model_processor(model_name, model_name_or_path="Salesforce/blip2-opt-2.7b", device=None):
diff --git a/omnitool/omniparserserver/omniparserserver.py b/omnitool/omniparserserver/omniparserserver.py
index 49fb306f..a67f1112 100644
--- a/omnitool/omniparserserver/omniparserserver.py
+++ b/omnitool/omniparserserver/omniparserserver.py
@@ -11,7 +11,7 @@
 import uvicorn
 root_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 sys.path.append(root_dir)
-from util.omniparser import Omniparser
+from omniparser.omniparser import Omniparser
 
 def parse_arguments():
     parser = argparse.ArgumentParser(description='Omniparser API')

From afe44cd303f5397ca3bf3b18eb286bd846041bac Mon Sep 17 00:00:00 2001
From: Alexey Taymanov <ataymano@gmail.com>
Date: Thu, 27 Mar 2025 20:43:24 +0000
Subject: [PATCH 2/2] setup.py

---
 setup.py | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 setup.py

diff --git a/setup.py b/setup.py
new file mode 100644
index 00000000..d29b3222
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,42 @@
+# This project is licensed under the Creative Commons Attribution 4.0 International License.
+# See https://creativecommons.org/licenses/by/4.0/ for details.
+
+from setuptools import setup, find_namespace_packages
+import pathlib
+
+here = pathlib.Path(__file__).parent.resolve()
+long_description = (here / "README.md").read_text(encoding="utf-8")
+
+setup(
+    name = "omniparser",
+    version = "0.1.0",
+    author = "Microsoft Research AIF Frontiers",
+    description = 'OmniParser, comprehensive method for parsing user interface screenshots into structured and easy-to-understand elements',
+    long_description = long_description,
+    long_description_content_type = "text/markdown",
+    include_package_data = True,
+    packages = find_namespace_packages(where='omniparser'),
+    package_dir = {"omniparser": "omniparser"},
+    install_requires = [
+        "torch",
+        "easyocr",
+        "torchvision",
+        "supervision==0.18.0",
+        "transformers",
+        "ultralytics==8.3.70",
+        "numpy==1.26.4",
+        "opencv-python",
+        "opencv-python-headless",
+        "gradio",
+        "dill",
+        "accelerate",
+        "timm",
+        "einops==0.8.0",
+        "paddlepaddle",
+        "paddleocr",
+    ],
+    classifiers=[
+        "License :: CC-BY-4.0"
+    ],
+    license_files = ['LICENSE']
+)
\ No newline at end of file