95%的人还在手动提取数据，用这个工具秒变结构化

#!/usr/bin/env python3# -*- coding: utf-8 -*-"""LangExtract 实战案例：医疗病历信息提取从混乱文本到结构化JSON的完整演示"""import langextract as lxfrom collections import defaultdictimport json# ============================================# 第一步：定义提取规则# ============================================prompt = """Extract patient demographics, diagnoses, and medications in order of appearance.Use exact text for extractions. Group related information using attributes."""# 提供高质量示例examples = [    lx.data.ExampleData(        text="Patient: Jane Doe, Age: 32. Diagnosis: Diabetes Mellitus. Medication: Metformin 500mg twice daily.",        extractions=[            # 患者信息            lx.data.Extraction(                extraction_class="patient_name",                extraction_text="Jane Doe",                attributes={"info_type": "demographics"}            ),            lx.data.Extraction(                extraction_class="age",                extraction_text="32",                attributes={"info_type": "demographics"}            ),            # 诊断            lx.data.Extraction(                extraction_class="diagnosis",                extraction_text="Diabetes Mellitus",                attributes={"info_type": "diagnosis"}            ),            # 用药            lx.data.Extraction(                extraction_class="medication",                extraction_text="Metformin 500mg twice daily",                attributes={                    "info_type": "medication",                    "medication_name": "Metformin"                }            )        ]    )]# ============================================# 第二步：执行提取# ============================================input_text = """Patient: John Smith, Age: 45, Gender: MaleVisit Date: 2025-03-15Chief Complaint: Persistent cough for 3 weeksHistory of Present Illness:Patient reports productive cough with yellow sputum for 3 weeks.No fever, no chest pain. History of hypertension for 5 years.Physical Exam:BP: 135/85 mmHg, HR: 78 bpm, Temp: 37.2°CLungs: Clear to auscultation bilaterallyDiagnosis:1. Acute bronchitis2. Hypertension, well-controlledTreatment Plan:Azithromycin 500mg PO once daily for 5 daysLisinopril 10mg PO daily (continue)Follow-up in 1 week if symptoms persist"""print("开始提取病历信息...")result = lx.extract(    text_or_documents=input_text,    prompt_description=prompt,    examples=examples,    model_id="gemini-2.5-flash")print(f"✓ 提取完成，共找到 {len(result.extractions)} 个实体\n")# ============================================# 第三步：处理结果# ============================================# 按信息类型分组structured_data = {    "demographics": {},    "diagnoses": [],    "medications": []}demographics = {}diagnoses = []medications = []for extraction in result.extractions:    entity_class = extraction.extraction_class    entity_text = extraction.extraction_text    if entity_class == "patient_name":        demographics["name"] = entity_text    elif entity_class == "age":        demographics["age"] = entity_text    elif entity_class == "diagnosis":        diagnoses.append(entity_text)    elif entity_class == "medication":        medications.append(extraction.text)structured_data["demographics"] = demographicsstructured_data["diagnoses"] = diagnosesstructured_data["medications"] = medications# 输出JSONprint("=" * 50)print("提取结果（JSON格式）：")print("=" * 50)print(json.dumps(structured_data, indent=2, ensure_ascii=False))# ============================================# 第四步：生成交互式可视化# ============================================print("\n" + "=" * 50)print("生成交互式可视化报告...")print("=" * 50)# 保存结果lx.io.save_annotated_documents(    [result],    output_name="medical_record_extraction.jsonl",    output_dir=".")# 生成可视化html_content = lx.visualize("medical_record_extraction.jsonl")with open("medical_viz.html", "w", encoding="utf-8") as f:    if hasattr(html_content, 'data'):        f.write(html_content.data)    else:        f.write(html_content)print("✓ 可视化报告已保存到 medical_viz.html")print("\n打开 medical_viz.html 可查看：")print("  - 每个提取字段在原文中的精确位置（高亮显示）")print("  - 点击实体查看详细信息（位置、属性、上下文）")print("  - 筛选特定类型的实体（只看诊断、只看用药）")print("  - 导出为其他格式（CSV、JSON）")