| from dotenv import load_dotenv |
| import uuid |
| from graph import HealthReport, SheamiState, create_graph |
| from pdf_reader import read_pdf |
|
|
| load_dotenv(override=True) |
|
|
|
|
| def main(): |
| print("Hello from sheami!") |
| workflow = create_graph(thread_id="my_console_thread") |
| report_file_names = [ |
| "./data/jg/report_001.pdf", |
| "./data/jg/report_002.pdf", |
| "./data/jg/report_003.pdf", |
| "./data/jg/report_005.pdf", |
| "./data/jg/report_006.pdf", |
| ] |
| state = SheamiState( |
| uploaded_reports=[ |
| HealthReport( |
| report_file_name=file_name, report_contents=read_pdf(file_name) |
| ) |
| for file_name in report_file_names |
| ] |
| ) |
| config = {"configurable": {"thread_id": str(uuid.uuid4())}} |
| response = workflow.invoke(state, config=config) |
| |
| print("**All done**") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
