feat: add magic-pdf-dev case

.github/workflows/cli.yml

@@ -37,7 +37,8 @@ jobs:
       run: |
         echo $GITHUB_WORKSPACE && sh tests/retry_env.sh
     - name: unit test
-      run: |        
+      run: | 
+        cd $GITHUB_WORKSPACE && python tests/clean_coverage.py
         cd $GITHUB_WORKSPACE && export PYTHONPATH=. && coverage run -m  pytest  tests/test_unit.py --cov=magic_pdf/ --cov-report term-missing --cov-report html
         cd $GITHUB_WORKSPACE && python tests/get_coverage.py
     - name: cli test

tests/clean_covrage.py

@@ -0,0 +1,24 @@
+"""
+clean coverage
+"""
+import os
+import shutil
+
+def delete_file(path):
+    """delete file."""
+    if not os.path.exists(path):
+        if os.path.isfile(path):
+            try:
+                os.remove(path)
+                print(f"File '{path}' deleted.")
+            except TypeError as e:
+                print(f"Error deleting file '{path}': {e}")
+    elif os.path.isdir(path):
+        try:
+            shutil.rmtree(path)
+            print(f"Directory '{path}' and its contents deleted.")
+        except TypeError as e:
+            print(f"Error deleting directory '{path}': {e}")
+
+if __name__ == "__main__":
+    delete_file("htmlcov")

tests/get_coverage.py

@@ -2,7 +2,7 @@
 get cov
 """
 from bs4 import BeautifulSoup
-
+import shutil
 def get_covrage():
     """get covrage"""
     # 发送请求获取网页内容

tests/test_cli/pdf_dev/test_model.json

@@ -0,0 +1,1472 @@
+[
+    {
+        "layout_dets": [
+            {
+                "category_id": 1,
+                "poly": [
+                    578.2055053710938,
+                    672.8831787109375,
+                    1579.973388671875,
+                    672.8831787109375,
+                    1579.973388671875,
+                    1034.681640625,
+                    578.2055053710938,
+                    1034.681640625
+                ],
+                "score": 0.9999963045120239
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    583.6041259765625,
+                    1067.1112060546875,
+                    1579.822265625,
+                    1067.1112060546875,
+                    1579.822265625,
+                    1537.1324462890625,
+                    583.6041259765625,
+                    1537.1324462890625
+                ],
+                "score": 0.9999961853027344
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    585.4341430664062,
+                    1568.220703125,
+                    1578.5487060546875,
+                    1568.220703125,
+                    1578.5487060546875,
+                    1931.516845703125,
+                    585.4341430664062,
+                    1931.516845703125
+                ],
+                "score": 0.9999949336051941
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    578.491455078125,
+                    532.0020141601562,
+                    1577.96337890625,
+                    532.0020141601562,
+                    1577.96337890625,
+                    641.0128784179688,
+                    578.491455078125,
+                    641.0128784179688
+                ],
+                "score": 0.999992847442627
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    66.43791961669922,
+                    1776.6951904296875,
+                    530.4810180664062,
+                    1776.6951904296875,
+                    530.4810180664062,
+                    1883.127685546875,
+                    66.43791961669922,
+                    1883.127685546875
+                ],
+                "score": 0.9999925494194031
+            },
+            {
+                "category_id": 3,
+                "poly": [
+                    70.23656463623047,
+                    818.9393920898438,
+                    517.8253784179688,
+                    818.9393920898438,
+                    517.8253784179688,
+                    1076.5823974609375,
+                    70.23656463623047,
+                    1076.5823974609375
+                ],
+                "score": 0.9999912977218628
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    64.99957275390625,
+                    651.9596557617188,
+                    436.5134582519531,
+                    651.9596557617188,
+                    436.5134582519531,
+                    723.5758056640625,
+                    64.99957275390625,
+                    723.5758056640625
+                ],
+                "score": 0.9999804496765137
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    556.2775268554688,
+                    270.2123107910156,
+                    1577.8211669921875,
+                    270.2123107910156,
+                    1577.8211669921875,
+                    408.9685974121094,
+                    556.2775268554688,
+                    408.9685974121094
+                ],
+                "score": 0.9999696016311646
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    67.8562240600586,
+                    1342.2239990234375,
+                    530.5654296875,
+                    1342.2239990234375,
+                    530.5654296875,
+                    1447.843017578125,
+                    67.8562240600586,
+                    1447.843017578125
+                ],
+                "score": 0.9999648928642273
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    65.74958801269531,
+                    1631.3671875,
+                    530.32861328125,
+                    1631.3671875,
+                    530.32861328125,
+                    1772.413818359375,
+                    65.74958801269531,
+                    1772.413818359375
+                ],
+                "score": 0.9999628067016602
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    588.5570068359375,
+                    2068.54931640625,
+                    1525.3253173828125,
+                    2068.54931640625,
+                    1525.3253173828125,
+                    2103.89013671875,
+                    588.5570068359375,
+                    2103.89013671875
+                ],
+                "score": 0.9999607801437378
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    586.5548706054688,
+                    1963.105712890625,
+                    1556.578125,
+                    1963.105712890625,
+                    1556.578125,
+                    2034.8116455078125,
+                    586.5548706054688,
+                    2034.8116455078125
+                ],
+                "score": 0.9999469518661499
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    59.96487045288086,
+                    1110.6282958984375,
+                    529.9209594726562,
+                    1110.6282958984375,
+                    529.9209594726562,
+                    1225.2921142578125,
+                    59.96487045288086,
+                    1225.2921142578125
+                ],
+                "score": 0.999945878982544
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    70.25292205810547,
+                    103.42201232910156,
+                    420.4892578125,
+                    103.42201232910156,
+                    420.4892578125,
+                    223.39370727539062,
+                    70.25292205810547,
+                    223.39370727539062
+                ],
+                "score": 0.9999405145645142
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    1081.0203857421875,
+                    2244.87890625,
+                    1554.669189453125,
+                    2244.87890625,
+                    1554.669189453125,
+                    2275.28662109375,
+                    1081.0203857421875,
+                    2275.28662109375
+                ],
+                "score": 0.9999217987060547
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    68.85404968261719,
+                    345.9093017578125,
+                    307.9080810546875,
+                    345.9093017578125,
+                    307.9080810546875,
+                    409.0098876953125,
+                    68.85404968261719,
+                    409.0098876953125
+                ],
+                "score": 0.9999183416366577
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    65.58759307861328,
+                    1295.9366455078125,
+                    180.4149932861328,
+                    1295.9366455078125,
+                    180.4149932861328,
+                    1328.867919921875,
+                    65.58759307861328,
+                    1328.867919921875
+                ],
+                "score": 0.9998926520347595
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    1245.0789794921875,
+                    108.83513641357422,
+                    1576.3131103515625,
+                    108.83513641357422,
+                    1576.3131103515625,
+                    219.29042053222656,
+                    1245.0789794921875,
+                    219.29042053222656
+                ],
+                "score": 0.9995975494384766
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    65.75041961669922,
+                    483.5210266113281,
+                    428.6028137207031,
+                    483.5210266113281,
+                    428.6028137207031,
+                    586.8894653320312,
+                    65.75041961669922,
+                    586.8894653320312
+                ],
+                "score": 0.9993270635604858
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    65.02926635742188,
+                    445.02288818359375,
+                    208.3317108154297,
+                    445.02288818359375,
+                    208.3317108154297,
+                    476.65252685546875,
+                    65.02926635742188,
+                    476.65252685546875
+                ],
+                "score": 0.9992279410362244
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    556.96630859375,
+                    453.08447265625,
+                    673.0485229492188,
+                    453.08447265625,
+                    673.0485229492188,
+                    490.60455322265625,
+                    556.96630859375,
+                    490.60455322265625
+                ],
+                "score": 0.9949817657470703
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    66.26518249511719,
+                    1524.234130859375,
+                    530.2540283203125,
+                    1524.234130859375,
+                    530.2540283203125,
+                    1627.5291748046875,
+                    66.26518249511719,
+                    1627.5291748046875
+                ],
+                "score": 0.9919581413269043
+            },
+            {
+                "category_id": 7,
+                "poly": [
+                    62.5564079284668,
+                    1227.41943359375,
+                    380.10693359375,
+                    1227.41943359375,
+                    380.10693359375,
+                    1252.8614501953125,
+                    62.5564079284668,
+                    1252.8614501953125
+                ],
+                "score": 0.9918426275253296
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    66.80464935302734,
+                    1451.4775390625,
+                    527.3795166015625,
+                    1451.4775390625,
+                    527.3795166015625,
+                    1519.5836181640625,
+                    66.80464935302734,
+                    1519.5836181640625
+                ],
+                "score": 0.9883899688720703
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    65.36080932617188,
+                    605.3754272460938,
+                    181.24375915527344,
+                    605.3754272460938,
+                    181.24375915527344,
+                    637.0076904296875,
+                    65.36080932617188,
+                    637.0076904296875
+                ],
+                "score": 0.9870840311050415
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    178.82904052734375,
+                    264.6627197265625,
+                    396.52825927734375,
+                    264.6627197265625,
+                    396.52825927734375,
+                    315.41900634765625,
+                    178.82904052734375,
+                    315.41900634765625
+                ],
+                "score": 0.9779323935508728
+            },
+            {
+                "category_id": 4,
+                "poly": [
+                    66.15127563476562,
+                    767.24658203125,
+                    181.25694274902344,
+                    767.24658203125,
+                    181.25694274902344,
+                    799.7832641601562,
+                    66.15127563476562,
+                    799.7832641601562
+                ],
+                "score": 0.8932801485061646
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    590,
+                    747,
+                    688,
+                    747,
+                    688,
+                    778,
+                    590,
+                    778
+                ],
+                "score": 0.91,
+                "latex": "+24.4\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1433,
+                    855,
+                    1492,
+                    855,
+                    1492,
+                    886,
+                    1433,
+                    886
+                ],
+                "score": 0.86,
+                "latex": "30\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    238,
+                    689,
+                    264,
+                    689,
+                    264,
+                    717,
+                    238,
+                    717
+                ],
+                "score": 0.34,
+                "latex": "@"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    702,
+                    1002,
+                    722,
+                    1002,
+                    722,
+                    1026,
+                    702,
+                    1026
+                ],
+                "score": 0.33,
+                "latex": "^+"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    177,
+                    1154,
+                    223,
+                    1154,
+                    223,
+                    1185,
+                    177,
+                    1185
+                ],
+                "score": 0.28,
+                "latex": "(\\%)"
+            }
+        ],
+        "page_info": {
+            "page_no": 0,
+            "height": 2339,
+            "width": 1654
+        }
+    },
+    {
+        "layout_dets": [
+            {
+                "category_id": 2,
+                "poly": [
+                    88.00849151611328,
+                    31.891826629638672,
+                    300.7432861328125,
+                    31.891826629638672,
+                    300.7432861328125,
+                    113.5999755859375,
+                    88.00849151611328,
+                    113.5999755859375
+                ],
+                "score": 0.9999986886978149
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    771.0192260742188,
+                    2213.479248046875,
+                    827.4273681640625,
+                    2213.479248046875,
+                    827.4273681640625,
+                    2239.40185546875,
+                    771.0192260742188,
+                    2239.40185546875
+                ],
+                "score": 0.9999963641166687
+            },
+            {
+                "category_id": 7,
+                "poly": [
+                    544.2962646484375,
+                    488.5493469238281,
+                    988.3958129882812,
+                    488.5493469238281,
+                    988.3958129882812,
+                    541.0634155273438,
+                    544.2962646484375,
+                    541.0634155273438
+                ],
+                "score": 0.9999918341636658
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    1082.88232421875,
+                    82.37471771240234,
+                    1519.4150390625,
+                    82.37471771240234,
+                    1519.4150390625,
+                    114.9271011352539,
+                    1082.88232421875,
+                    114.9271011352539
+                ],
+                "score": 0.9999632835388184
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    1009.1597900390625,
+                    2210.9462890625,
+                    1535.9239501953125,
+                    2210.9462890625,
+                    1535.9239501953125,
+                    2241.830322265625,
+                    1009.1597900390625,
+                    2241.830322265625
+                ],
+                "score": 0.9999324679374695
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    537.349365234375,
+                    156.8784637451172,
+                    1584.9866943359375,
+                    156.8784637451172,
+                    1584.9866943359375,
+                    485.3042907714844,
+                    537.349365234375,
+                    485.3042907714844
+                ],
+                "score": 0.9985955953598022
+            },
+            {
+                "category_id": 7,
+                "poly": [
+                    62.69784927368164,
+                    443.4034118652344,
+                    249.9097137451172,
+                    443.4034118652344,
+                    249.9097137451172,
+                    467.4612731933594,
+                    62.69784927368164,
+                    467.4612731933594
+                ],
+                "score": 0.9873980283737183
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    61.374210357666016,
+                    138.51153564453125,
+                    528.30517578125,
+                    138.51153564453125,
+                    528.30517578125,
+                    443.5376281738281,
+                    61.374210357666016,
+                    443.5376281738281
+                ],
+                "score": 0.9232220649719238
+            },
+            {
+                "category_id": 6,
+                "poly": [
+                    548.1119384765625,
+                    148.7312774658203,
+                    797.3070678710938,
+                    148.7312774658203,
+                    797.3070678710938,
+                    180.74609375,
+                    548.1119384765625,
+                    180.74609375
+                ],
+                "score": 0.6074804663658142
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    864,
+                    455,
+                    922,
+                    455,
+                    922,
+                    482,
+                    864,
+                    482
+                ],
+                "score": 0.74,
+                "latex": "6.0\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    850,
+                    418,
+                    922,
+                    418,
+                    922,
+                    445,
+                    850,
+                    445
+                ],
+                "score": 0.64,
+                "latex": "35.3\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1501,
+                    270,
+                    1571,
+                    270,
+                    1571,
+                    298,
+                    1501,
+                    298
+                ],
+                "score": 0.54,
+                "latex": "13.8\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1013,
+                    454,
+                    1083,
+                    454,
+                    1083,
+                    482,
+                    1013,
+                    482
+                ],
+                "score": 0.52,
+                "latex": "15.0\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1012,
+                    417,
+                    1083,
+                    417,
+                    1083,
+                    444,
+                    1012,
+                    444
+                ],
+                "score": 0.52,
+                "latex": "33.7\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    689,
+                    456,
+                    725,
+                    456,
+                    725,
+                    482,
+                    689,
+                    482
+                ],
+                "score": 0.48,
+                "latex": "(\\%)"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    850,
+                    344,
+                    922,
+                    344,
+                    922,
+                    372,
+                    850,
+                    372
+                ],
+                "score": 0.4,
+                "latex": "83.8\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    863,
+                    270,
+                    922,
+                    270,
+                    922,
+                    298,
+                    863,
+                    298
+                ],
+                "score": 0.4,
+                "latex": "4.5\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1334,
+                    270,
+                    1406,
+                    270,
+                    1406,
+                    298,
+                    1334,
+                    298
+                ],
+                "score": 0.35,
+                "latex": "37.2\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    618,
+                    419,
+                    656,
+                    419,
+                    656,
+                    446,
+                    618,
+                    446
+                ],
+                "score": 0.35,
+                "latex": "(\\%)"
+            }
+        ],
+        "page_info": {
+            "page_no": 1,
+            "height": 2339,
+            "width": 1654
+        }
+    },
+    {
+        "layout_dets": [
+            {
+                "category_id": 2,
+                "poly": [
+                    87.9037094116211,
+                    31.59800148010254,
+                    300.9930419921875,
+                    31.59800148010254,
+                    300.9930419921875,
+                    113.4053955078125,
+                    87.9037094116211,
+                    113.4053955078125
+                ],
+                "score": 0.9999939799308777
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    1008.992919921875,
+                    2209.248779296875,
+                    1534.9334716796875,
+                    2209.248779296875,
+                    1534.9334716796875,
+                    2242.77294921875,
+                    1008.992919921875,
+                    2242.77294921875
+                ],
+                "score": 0.9999377131462097
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    770.6600341796875,
+                    2212.857666015625,
+                    827.4126586914062,
+                    2212.857666015625,
+                    827.4126586914062,
+                    2239.77197265625,
+                    770.6600341796875,
+                    2239.77197265625
+                ],
+                "score": 0.9998395442962646
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    1082.096923828125,
+                    82.25012969970703,
+                    1518.9267578125,
+                    82.25012969970703,
+                    1518.9267578125,
+                    114.52576446533203,
+                    1082.096923828125,
+                    114.52576446533203
+                ],
+                "score": 0.9996457099914551
+            },
+            {
+                "category_id": 7,
+                "poly": [
+                    95.39900970458984,
+                    1846.6380615234375,
+                    564.4166870117188,
+                    1846.6380615234375,
+                    564.4166870117188,
+                    1899.209716796875,
+                    95.39900970458984,
+                    1899.209716796875
+                ],
+                "score": 0.9908766746520996
+            },
+            {
+                "category_id": 6,
+                "poly": [
+                    95.4662094116211,
+                    173.42832946777344,
+                    470.21905517578125,
+                    173.42832946777344,
+                    470.21905517578125,
+                    217.74632263183594,
+                    95.4662094116211,
+                    217.74632263183594
+                ],
+                "score": 0.9437939524650574
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    854.1142578125,
+                    1043.93603515625,
+                    1592.0174560546875,
+                    1043.93603515625,
+                    1592.0174560546875,
+                    1846.16552734375,
+                    854.1142578125,
+                    1846.16552734375
+                ],
+                "score": 0.8844046592712402
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    92.02946472167969,
+                    1331.8909912109375,
+                    814.2915649414062,
+                    1331.8909912109375,
+                    814.2915649414062,
+                    1842.6195068359375,
+                    92.02946472167969,
+                    1842.6195068359375
+                ],
+                "score": 0.8743430972099304
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    851.83984375,
+                    224.99559020996094,
+                    1592.4068603515625,
+                    224.99559020996094,
+                    1592.4068603515625,
+                    1018.7105712890625,
+                    851.83984375,
+                    1018.7105712890625
+                ],
+                "score": 0.8650150299072266
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    91.79800415039062,
+                    224.10838317871094,
+                    816.58154296875,
+                    224.10838317871094,
+                    816.58154296875,
+                    1248.422607421875,
+                    91.79800415039062,
+                    1248.422607421875
+                ],
+                "score": 0.8604844808578491
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    85.19661712646484,
+                    220.71524047851562,
+                    1602.3074951171875,
+                    220.71524047851562,
+                    1602.3074951171875,
+                    1844.488525390625,
+                    85.19661712646484,
+                    1844.488525390625
+                ],
+                "score": 0.6638449430465698
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    737,
+                    704,
+                    804,
+                    704,
+                    804,
+                    730,
+                    737,
+                    730
+                ],
+                "score": 0.56,
+                "latex": "\\pmb{26.5\\%}"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    738,
+                    673,
+                    804,
+                    673,
+                    804,
+                    699,
+                    738,
+                    699
+                ],
+                "score": 0.48,
+                "latex": "\\pmb{16.2\\%}"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    736,
+                    767,
+                    805,
+                    767,
+                    805,
+                    795,
+                    736,
+                    795
+                ],
+                "score": 0.48,
+                "latex": "\\mathbf{\\lambda_{23.7\\%}}"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    231,
+                    611,
+                    268,
+                    611,
+                    268,
+                    638,
+                    231,
+                    638
+                ],
+                "score": 0.47,
+                "latex": "(\\%)"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    749,
+                    736,
+                    804,
+                    736,
+                    804,
+                    763,
+                    749,
+                    763
+                ],
+                "score": 0.41,
+                "latex": "\\pmb{9.2\\%}"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    737,
+                    641,
+                    804,
+                    641,
+                    804,
+                    668,
+                    737,
+                    668
+                ],
+                "score": 0.41,
+                "latex": "{\\bf38.0\\%}"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    748,
+                    577,
+                    805,
+                    577,
+                    805,
+                    606,
+                    748,
+                    606
+                ],
+                "score": 0.35,
+                "latex": "0.1\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    187,
+                    800,
+                    222,
+                    800,
+                    222,
+                    827,
+                    187,
+                    827
+                ],
+                "score": 0.32,
+                "latex": "(\\%)"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    738,
+                    830,
+                    805,
+                    830,
+                    805,
+                    857,
+                    738,
+                    857
+                ],
+                "score": 0.28,
+                "latex": "\\mathbf{13.8\\%}"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    737,
+                    862,
+                    805,
+                    862,
+                    805,
+                    889,
+                    737,
+                    889
+                ],
+                "score": 0.27,
+                "latex": "\\mathbf{31.9\\%}"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    736,
+                    955,
+                    804,
+                    955,
+                    804,
+                    983,
+                    736,
+                    983
+                ],
+                "score": 0.26,
+                "latex": "\\pmb{65.3\\%}"
+            }
+        ],
+        "page_info": {
+            "page_no": 2,
+            "height": 2339,
+            "width": 1654
+        }
+    },
+    {
+        "layout_dets": [
+            {
+                "category_id": 2,
+                "poly": [
+                    86.3010025024414,
+                    32.05937194824219,
+                    303.65325927734375,
+                    32.05937194824219,
+                    303.65325927734375,
+                    114.77494049072266,
+                    86.3010025024414,
+                    114.77494049072266
+                ],
+                "score": 0.9999954700469971
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    108.4952392578125,
+                    590.2026977539062,
+                    1536.75146484375,
+                    590.2026977539062,
+                    1536.75146484375,
+                    688.4915771484375,
+                    108.4952392578125,
+                    688.4915771484375
+                ],
+                "score": 0.9999932646751404
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    95.94864654541016,
+                    1205.4134521484375,
+                    252.92477416992188,
+                    1205.4134521484375,
+                    252.92477416992188,
+                    1246.0015869140625,
+                    95.94864654541016,
+                    1246.0015869140625
+                ],
+                "score": 0.999992847442627
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    106.48407745361328,
+                    338.27471923828125,
+                    1568.86328125,
+                    338.27471923828125,
+                    1568.86328125,
+                    437.84783935546875,
+                    106.48407745361328,
+                    437.84783935546875
+                ],
+                "score": 0.9999897480010986
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    767.6918334960938,
+                    2212.269287109375,
+                    830.787353515625,
+                    2212.269287109375,
+                    830.787353515625,
+                    2239.28515625,
+                    767.6918334960938,
+                    2239.28515625
+                ],
+                "score": 0.9999850988388062
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    96.18482208251953,
+                    508.36334228515625,
+                    291.4427490234375,
+                    508.36334228515625,
+                    291.4427490234375,
+                    549.4661865234375,
+                    96.18482208251953,
+                    549.4661865234375
+                ],
+                "score": 0.9999837875366211
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    1082.2672119140625,
+                    81.18732452392578,
+                    1520.2149658203125,
+                    81.18732452392578,
+                    1520.2149658203125,
+                    116.55751037597656,
+                    1082.2672119140625,
+                    116.55751037597656
+                ],
+                "score": 0.9999496340751648
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    96.45167541503906,
+                    157.92835998535156,
+                    319.21392822265625,
+                    157.92835998535156,
+                    319.21392822265625,
+                    213.8436279296875,
+                    96.45167541503906,
+                    213.8436279296875
+                ],
+                "score": 0.9999274015426636
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    96.99238586425781,
+                    257.6522216796875,
+                    483.6472473144531,
+                    257.6522216796875,
+                    483.6472473144531,
+                    301.53717041015625,
+                    96.99238586425781,
+                    301.53717041015625
+                ],
+                "score": 0.9999104738235474
+            },
+            {
+                "category_id": 2,
+                "poly": [
+                    1008.8760986328125,
+                    2208.609375,
+                    1536.0474853515625,
+                    2208.609375,
+                    1536.0474853515625,
+                    2243.414306640625,
+                    1008.8760986328125,
+                    2243.414306640625
+                ],
+                "score": 0.9998928308486938
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    108.46533203125,
+                    1288.0927734375,
+                    1546.7518310546875,
+                    1288.0927734375,
+                    1546.7518310546875,
+                    1383.8438720703125,
+                    108.46533203125,
+                    1383.8438720703125
+                ],
+                "score": 0.9997898936271667
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    107.81462860107422,
+                    1678.24609375,
+                    1227.880615234375,
+                    1678.24609375,
+                    1227.880615234375,
+                    1711.37255859375,
+                    107.81462860107422,
+                    1711.37255859375
+                ],
+                "score": 0.99957275390625
+            },
+            {
+                "category_id": 5,
+                "poly": [
+                    109.75360107421875,
+                    810.0169677734375,
+                    1579.9549560546875,
+                    810.0169677734375,
+                    1579.9549560546875,
+                    1171.6383056640625,
+                    109.75360107421875,
+                    1171.6383056640625
+                ],
+                "score": 0.9994542598724365
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    106.46218872070312,
+                    1548.299072265625,
+                    1540.3388671875,
+                    1548.299072265625,
+                    1540.3388671875,
+                    1676.67919921875,
+                    106.46218872070312,
+                    1676.67919921875
+                ],
+                "score": 0.9886452555656433
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    107.52558898925781,
+                    1386.4000244140625,
+                    1540.886962890625,
+                    1386.4000244140625,
+                    1540.886962890625,
+                    1447.8128662109375,
+                    107.52558898925781,
+                    1447.8128662109375
+                ],
+                "score": 0.9709398150444031
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    107.66414642333984,
+                    1451.8369140625,
+                    1537.99169921875,
+                    1451.8369140625,
+                    1537.99169921875,
+                    1546.690185546875,
+                    107.66414642333984,
+                    1546.690185546875
+                ],
+                "score": 0.9590120315551758
+            },
+            {
+                "category_id": 6,
+                "poly": [
+                    95.90371704101562,
+                    728.2855224609375,
+                    328.1967468261719,
+                    728.2855224609375,
+                    328.1967468261719,
+                    768.121826171875,
+                    95.90371704101562,
+                    768.121826171875
+                ],
+                "score": 0.6999977827072144
+            },
+            {
+                "category_id": 1,
+                "poly": [
+                    106.67481994628906,
+                    1371.857421875,
+                    1544.84814453125,
+                    1371.857421875,
+                    1544.84814453125,
+                    1678.67236328125,
+                    106.67481994628906,
+                    1678.67236328125
+                ],
+                "score": 0.5645973086357117
+            },
+            {
+                "category_id": 0,
+                "poly": [
+                    95.94171142578125,
+                    728.264404296875,
+                    328.1947937011719,
+                    728.264404296875,
+                    328.1947937011719,
+                    768.1663818359375,
+                    95.94171142578125,
+                    768.1663818359375
+                ],
+                "score": 0.30702608823776245
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1247,
+                    887,
+                    1353,
+                    887,
+                    1353,
+                    914,
+                    1247,
+                    914
+                ],
+                "score": 0.91,
+                "latex": "5\\%{\\sim}20\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1181,
+                    923,
+                    1290,
+                    923,
+                    1290,
+                    950,
+                    1181,
+                    950
+                ],
+                "score": 0.9,
+                "latex": "-5\\%{+}5\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1416,
+                    1047,
+                    1469,
+                    1047,
+                    1469,
+                    1077,
+                    1416,
+                    1077
+                ],
+                "score": 0.87,
+                "latex": "10\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1254,
+                    963,
+                    1296,
+                    963,
+                    1296,
+                    991,
+                    1254,
+                    991
+                ],
+                "score": 0.86,
+                "latex": "5\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1373,
+                    1003,
+                    1428,
+                    1003,
+                    1428,
+                    1032,
+                    1373,
+                    1032
+                ],
+                "score": 0.86,
+                "latex": "10\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1332,
+                    1047,
+                    1388,
+                    1047,
+                    1388,
+                    1076,
+                    1332,
+                    1076
+                ],
+                "score": 0.86,
+                "latex": "\\cdot10\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1373,
+                    1112,
+                    1428,
+                    1112,
+                    1428,
+                    1141,
+                    1373,
+                    1141
+                ],
+                "score": 0.85,
+                "latex": "10\\%"
+            },
+            {
+                "category_id": 13,
+                "poly": [
+                    1248,
+                    854,
+                    1302,
+                    854,
+                    1302,
+                    880,
+                    1248,
+                    880
+                ],
+                "score": 0.85,
+                "latex": "z0\\%"
+            }
+        ],
+        "page_info": {
+            "page_no": 3,
+            "height": 2339,
+            "width": 1654
+        }
+    }
+]

tests/test_cli/test_cli_sdk.py

@@ -182,7 +182,7 @@ class TestCli:
     def test_pdf_dev_cli_local_jsonl_txt(self):
         """magic_pdf_dev cli local txt."""
         jsonl_path = os.path.join(pdf_dev_path, 'line1.jsonl')
-        cmd = 'magic-pdf-dev --jsonl %s -m %s' % (jsonl_path, "txt")
+        cmd = 'magic-pdf-dev --jsonl %s --method %s' % (jsonl_path, "txt")
         logging.info(cmd)
         os.system(cmd)
 
@@ -191,7 +191,7 @@ class TestCli:
     def test_pdf_dev_cli_local_jsonl_ocr(self):
         """magic_pdf_dev cli local ocr."""
         jsonl_path = os.path.join(pdf_dev_path, 'line1.jsonl')
-        cmd = 'magic-pdf-dev --jsonl %s -m %s' % (jsonl_path, 'ocr')
+        cmd = 'magic-pdf-dev --jsonl %s --method %s' % (jsonl_path, 'ocr')
         logging.info(cmd)
         os.system(cmd)
 
@@ -199,7 +199,7 @@ class TestCli:
     def test_pdf_dev_cli_local_jsonl_auto(self):
         """magic_pdf_dev cli local auto."""
         jsonl_path = os.path.join(pdf_dev_path, 'line1.jsonl')
-        cmd = 'magic-pdf-dev --jsonl %s -m %s' % (jsonl_path, 'auto')
+        cmd = 'magic-pdf-dev --jsonl %s --method %s' % (jsonl_path, 'auto')
         logging.info(cmd)
         os.system(cmd)
 
@@ -207,7 +207,7 @@ class TestCli:
     def test_pdf_dev_cli_s3_jsonl_txt(self):
         """magic_pdf_dev cli s3 txt."""
         jsonl_path = os.path.join(pdf_dev_path, 'line1.jsonl')
-        cmd = 'magic-pdf-dev --jsonl %s -m %s' % (jsonl_path, "txt")
+        cmd = 'magic-pdf-dev --jsonl %s --method %s' % (jsonl_path, "txt")
         logging.info(cmd)
         os.system(cmd)
 
@@ -216,7 +216,7 @@ class TestCli:
     def test_pdf_dev_cli_s3_jsonl_ocr(self):
         """magic_pdf_dev cli s3 ocr."""
         jsonl_path = os.path.join(pdf_dev_path, 'line1.jsonl')
-        cmd = 'magic-pdf-dev --jsonl %s -m %s' % (jsonl_path, 'ocr')
+        cmd = 'magic-pdf-dev --jsonl %s --method %s' % (jsonl_path, 'ocr')
         logging.info(cmd)
         os.system(cmd)
 
@@ -229,5 +229,26 @@ class TestCli:
         os.system(cmd)
 
 
+    @pytest.mark.P1
+    def test_pdf_dev_cli_pdf_json_auto(self):
+        """magic_pdf_dev cli pdf+json auto."""
+        json_path = os.path.join(pdf_dev_path, 'test_model.json')
+        pdf_path = os.path.join(pdf_dev_path, 'pdf', 'research_report_1f978cd81fb7260c8f7644039ec2c054.pdf')
+        cmd = 'magic-pdf-dev --pdf %s --json %s --method %s' % (pdf_path, json_path, 'auto')
+        logging.info(cmd)
+        os.system(cmd)
+    
+    @pytest.mark.P1
+    def test_pdf_dev_cli_pdf_json_ocr(self):
+        """magic_pdf_dev cli pdf+json ocr."""
+        json_path = os.path.join(pdf_dev_path, 'test_model.json')
+        pdf_path = os.path.join(pdf_dev_path, 'pdf', 'research_report_1f978cd81fb7260c8f7644039ec2c054.pdf')
+        cmd = 'magic-pdf-dev --pdf %s --json %s --method %s' % (pdf_path, json_path, 'auto')
+        logging.info(cmd)
+        os.system(cmd)
+
+
+
+
 if __name__ == '__main__':
     pytest.main()