luciano/markitdown
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Browse source
This commit is contained in:
Vijay Soni 2024-12-23 01:07:13 +05:30
parent 206b4a5ad6
commit 05084712c4
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/markitdown/__main__.py
View file

@ -6,6 +6,7 @@ import sys
from textwrap import dedent from textwrap import dedent
from .__about__ import __version__ from .__about__ import __version__
from ._markitdown import MarkItDown, DocumentConverterResult from ._markitdown import MarkItDown, DocumentConverterResult
import uvicorn
def main(): def main():
@ -57,9 +58,16 @@ def main():
"--output", "--output",
help="Output file name. If not provided, output is written to stdout.", help="Output file name. If not provided, output is written to stdout.",
) )
parser.add_argument(
"--api",
action="store_true",
help="Start the FastAPI server",
)
args = parser.parse_args() args = parser.parse_args()
if args.filename is None: if args.api:
uvicorn.run("src.markitdown.api:app", host="0.0.0.0", port=8000)
elif args.filename is None:
markitdown = MarkItDown() markitdown = MarkItDown()
result = markitdown.convert_stream(sys.stdin.buffer) result = markitdown.convert_stream(sys.stdin.buffer)
_handle_output(args, result) _handle_output(args, result)