| |
| """ |
| Pydantic models for medical document processing |
| """ |
|
|
| from pydantic import BaseModel, Field |
| from typing import Dict, List, Any, Optional |
|
|
|
|
| class TemplateAnalysis(BaseModel): |
| """Model for template analysis results.""" |
| sections: List[Dict[str, Any]] = Field( |
| description="List of sections found in template") |
| formatting: Dict[str, Any] = Field(description="Formatting information") |
| document_info: Dict[str, str] = Field(description="Document metadata") |
|
|
|
|
| class MedicalTranscription(BaseModel): |
| """Model for medical transcription data.""" |
| raw_text: str = Field(description="Raw transcription text") |
| corrected_text: str = Field(description="Corrected and structured text") |
| medical_data: Dict[str, Any] = Field( |
| description="Extracted medical information") |
|
|
|
|
| class SectionContent(BaseModel): |
| """Model for section content.""" |
| technique: str = Field(description="Technique section content") |
| result: str = Field(description="Result section content") |
| conclusion: str = Field(description="Conclusion section content") |
|
|
|
|
| class InsertSectionsInput(BaseModel): |
| """Model for inserting sections into documents.""" |
| template_path: str |
| sections: Dict[str, str] |
| output_path: str |
| title: str = None |
|
|
