Upload 2 files

chart_engine.py

@@ -0,0 +1,70 @@
+import yaml, json, os
+from crewai import Agent, Task, Crew, Process
+from langchain_openai import ChatOpenAI
+
+class PatientChartCrewEngine:
+    def __init__(self, folder_path: str, measure_name: str):
+        if not measure_name.endswith(".yml"):
+            measure_name += ".yml"
+        
+        file_path = os.path.join(folder_path, measure_name)
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"Measure file not found at: {file_path}")
+
+        with open(file_path, 'r') as f:
+            self.cfg = yaml.safe_load(f)
+        
+        self.llm = ChatOpenAI(model="gpt-5.1", temperature=0.5)
+
+    def run(self):
+        scribe = Agent(
+            role="Senior Clinical Quality Scribe",
+            goal="Generate a dense, 2-page clinical encounter note with mandatory dates for all entries.",
+            backstory="""You are a senior scribe specializing in audit-ready records. 
+            Every diagnosis, procedure, and immunization MUST have a specific date (MM/DD/YYYY). 
+            You must use all YAML target_rules and expand the physical exam to hit 2 pages.""",
+            llm=self.llm,
+            allow_delegation=False
+        )
+
+        architect = Agent(
+            role="Medical Document Architect",
+            goal="Format clinical data into a professional 2-page LaTeX document.",
+            backstory="""You are a LaTeX expert. You invent unique clinic branding and 
+            use high-density formatting (no whitespace) to ensure the chart is 2 pages long.""",
+            llm=self.llm,
+            allow_delegation=False
+        )
+
+        content_task = Task(
+            description=f"""
+            MEASURE RULES: {json.dumps(self.cfg.get('target_rules', {}), indent=2)}
+            
+            DIRECTIONS FOR AUDIT VALIDITY:
+            1. MANDATORY DATES: Every diagnosis (ICD-10), procedure (CPT/HCPCS), and 
+               immunization MUST include a specific date of occurrence (MM/DD/YYYY).
+            2. NARRATIVE VOLUME: Write a 10-sentence HPI and a 12-point Review of Systems (ROS).
+            3. PHYSICAL EXAM: Include a comprehensive multi-system physical exam.
+            4. PATIENT: John Thompson (DOB 05/09/1951).
+            """,
+            agent=scribe,
+            expected_output="Extensive clinical JSON with mandatory dates."
+        )
+
+        latex_task = Task(
+            description="""
+            Convert the output into professional LaTeX. 
+            - BRANDING: Invent a UNIQUE clinic name/address (Never use ABC Medicine).
+            - PAGE FLOW: Ensure the content fills 2 full pages.
+            - CLEANLINESS: Output ONLY the LaTeX code. No summaries or conversational text.
+            - RETURN ONLY RAW LaTeX code starting with \\documentclass.
+            """,
+            agent=architect,
+            context=[content_task],
+            expected_output="Complete LaTeX source code."
+        )
+
+        crew = Crew(agents=[scribe, architect], tasks=[content_task, latex_task], process=Process.sequential)
+        
+        result = crew.kickoff()
+        return result.raw if hasattr(result, 'raw') else str(result)

pdf_renderer.py

@@ -0,0 +1,26 @@
+import os, subprocess, re, shutil
+
+def render_pdf(latex_code, output_path):
+    # Strip AI conversation and isolate LaTeX block
+    match = re.search(r"```(?:latex)?\n?(.*?)\n?```", latex_code, re.DOTALL)
+    clean_code = match.group(1).strip() if match else latex_code.strip()
+
+    tex_filename = "final_output.tex"
+    with open(tex_filename, "w") as f:
+        f.write(clean_code)
+
+    try:
+        # pdflatex requires 2 passes for stable headers and page numbering
+        for _ in range(2):
+            subprocess.run(["pdflatex", "-interaction=nonstopmode", tex_filename],
+                           check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+        # Use shutil to move from local Colab disk to Google Drive
+        if os.path.exists("final_output.pdf"):
+            shutil.move("final_output.pdf", output_path)
+            return True
+        return False
+            
+    except subprocess.CalledProcessError:
+        print("❌ LaTeX Error: The AI produced invalid syntax. Check the logs.")
+        return False