#!/usr/bin/env python3
"""QC Report Generator - Creates an HTML report of all subject QC images."""
import os
from datetime import datetime
def generate_report():
base_dir = os.path.expanduser("~/Desktop/derivatives")
qc_dir = os.path.join(base_dir, "QC")
output_report = os.path.join(qc_dir, "index.html")
print(f"Generating QC Report at: {output_report}")
subjects = sorted([d for d in os.listdir(qc_dir) if d.startswith("sub-") and os.path.isdir(os.path.join(qc_dir, d))])
if not subjects:
print("No subject QC folders found.")
return
html = f"""
PET-NeuroProject QC Report
PET-NeuroProject Quality Control Report
Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
Total Subjects: {len(subjects)}
"""
for sub in subjects:
sub_qc_path = os.path.join(qc_dir, sub)
img_coreg = f"{sub}/qc_coreg.png"
img_norm = f"{sub}/qc_norm.png"
img_masks = f"{sub}/qc_masks.png"
has_coreg = os.path.exists(os.path.join(sub_qc_path, "qc_coreg.png"))
has_norm = os.path.exists(os.path.join(sub_qc_path, "qc_norm.png"))
has_masks = os.path.exists(os.path.join(sub_qc_path, "qc_masks.png"))
html += f"""
"""
html += '
'
if has_coreg:
html += f'
Coregistration
'
else:
html += '
Missing
'
html += '
'
html += '
'
if has_norm:
html += f'
Normalization
'
else:
html += '
Missing
'
html += '
'
html += '
'
if has_masks:
html += f'
Mask Alignment
'
else:
html += '
Missing
'
html += '
'
html += """
"""
html += """
"""
with open(output_report, "w") as f:
f.write(html)
print("Report generation complete.")
if __name__ == "__main__":
generate_report()