Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import tempfile | |
| import os | |
| import shutil | |
| from pathlib import Path | |
| from parse_invoices import read_msg, parse_lpo_html, init_or_load_workbook, append_row, get_excel_path | |
| def process_msg_files(files): | |
| if not files: | |
| return None, "No files uploaded." | |
| with tempfile.TemporaryDirectory() as tmpdir: | |
| tmpdir = Path(tmpdir) | |
| # Save uploaded files to temp folder | |
| for f in files: | |
| shutil.copy(f.name, tmpdir / Path(f.name).name) | |
| # Run parser | |
| excel_path = get_excel_path(tmpdir) | |
| wb, ws = init_or_load_workbook(excel_path) | |
| next_row = ws.max_row + 1 | |
| log_lines = [] | |
| for msg_path in sorted(tmpdir.glob('*.msg')): | |
| try: | |
| subject, html_bytes = read_msg(msg_path) | |
| if not html_bytes: | |
| log_lines.append(f"⚠️ {msg_path.name} — no HTML body") | |
| continue | |
| fields = parse_lpo_html(html_bytes) | |
| row_data = {'subject': subject, 'filename': msg_path.name, **fields} | |
| missing = [k for k, v in fields.items() if v is None] | |
| status = 'PARTIAL' if missing else 'OK' | |
| append_row(ws, row_data, next_row, status=status) | |
| next_row += 1 | |
| log_lines.append(f"✅ {msg_path.name} — LPO:{fields['lpo_number']} | Inv:{fields['inv_number']} | Amount:{fields['inv_amount']}") | |
| except Exception as e: | |
| log_lines.append(f"❌ {msg_path.name} — ERROR: {e}") | |
| wb.save(excel_path) | |
| wb.close() | |
| # Copy excel to a stable output path Gradio can serve | |
| out_path = Path(tempfile.mktemp(suffix='.xlsx')) | |
| shutil.copy(excel_path, out_path) | |
| return str(out_path), "\n".join(log_lines) | |
| demo = gr.Interface( | |
| fn=process_msg_files, | |
| inputs=gr.File( | |
| file_count="multiple", | |
| file_types=[".msg"], | |
| label="Upload .msg files" | |
| ), | |
| outputs=[ | |
| gr.File(label="Download Excel"), | |
| gr.Textbox(label="Processing Log", lines=15) | |
| ], | |
| title="LPO Invoice Parser", | |
| description="Upload one or more Outlook .msg LPO emails. Get back a formatted Excel with LPO #, Ref No, Inv #, and Amount." | |
| ) | |
| demo.launch() |