mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-06-02 14:45:17 +02:00
Feature: invoice ocr assistant
This commit is contained in:
parent
443c044990
commit
e3ecb88293
15 changed files with 598 additions and 4 deletions
39
examples/invoice_ocr.py
Normal file
39
examples/invoice_ocr.py
Normal file
|
|
@ -0,0 +1,39 @@
|
|||
#!/usr/bin/env python3
|
||||
# _*_ coding: utf-8 _*_
|
||||
|
||||
"""
|
||||
@Time : 2023/9/21 21:40:57
|
||||
@Author : Stitch-z
|
||||
@File : tutorial_assistant.py
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import os
|
||||
|
||||
from metagpt.roles.invoice_ocr_assistant import InvoiceOCRAssistant
|
||||
from metagpt.schema import Message
|
||||
|
||||
|
||||
async def main():
|
||||
relative_paths = [
|
||||
"../tests/data/invoices/invoice-1.pdf",
|
||||
"../tests/data/invoices/invoice-2.png",
|
||||
"../tests/data/invoices/invoice-3.jpg",
|
||||
"../tests/data/invoices/invoice-4.zip"
|
||||
]
|
||||
# Get the current working directory
|
||||
current_directory = os.getcwd()
|
||||
# The absolute path of the file
|
||||
absolute_file_paths = [os.path.abspath(os.path.join(current_directory, path)) for path in relative_paths]
|
||||
|
||||
for path in absolute_file_paths:
|
||||
role = InvoiceOCRAssistant()
|
||||
await role.run(Message(
|
||||
content="Invoicing date",
|
||||
instruct_content={"file_path": path}
|
||||
))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
asyncio.run(main())
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue