Upload folder using huggingface_hub

k12/img_generator.py

@@ -48,7 +48,7 @@ CERTIFICATIONS = [
 ]
 
 
-def _generate_teacher_data(first_name: str, last_name: str, school_name: str = "Springfield High School", email_domain: str = None) -> dict:
+def _generate_teacher_data(first_name: str, last_name: str, school_name: str = "Springfield High School", email_domain: str = None, logo_url: str = None) -> dict:
     """Generate consistent teacher data for all templates."""
     full_name = f"{first_name} {last_name}"
     initials = f"{first_name[0]}{last_name[0]}"
@@ -110,6 +110,20 @@ def _generate_teacher_data(first_name: str, last_name: str, school_name: str = "
         "CERT_DATE": cert_date.strftime("%m/%Y"),
         "SCHOOL_NAME": school_name,
         "DISTRICT_NAME": district_name,
+        # Dynamic payroll dates (recent)
+        "PAY_DATE_1": (datetime.now() - timedelta(days=5)).strftime("%m/%d/%Y"),
+        "PAY_DATE_2": (datetime.now() - timedelta(days=20)).strftime("%m/%d/%Y"),
+        "PAY_DATE_3": (datetime.now() - timedelta(days=35)).strftime("%m/%d/%Y"),
+        "PAY_PERIOD_1_START": (datetime.now() - timedelta(days=20)).strftime("%m/%d/%Y"),
+        "PAY_PERIOD_1_END": (datetime.now() - timedelta(days=6)).strftime("%m/%d/%Y"),
+        "PAY_PERIOD_2_START": (datetime.now() - timedelta(days=35)).strftime("%m/%d/%Y"),
+        "PAY_PERIOD_2_END": (datetime.now() - timedelta(days=21)).strftime("%m/%d/%Y"),
+        "PAY_PERIOD_3_START": (datetime.now() - timedelta(days=50)).strftime("%m/%d/%Y"),
+        "PAY_PERIOD_3_END": (datetime.now() - timedelta(days=36)).strftime("%m/%d/%Y"),
+        "SCHOOL_YEAR": f"{datetime.now().year-1}-{datetime.now().year}" if datetime.now().month < 7 else f"{datetime.now().year}-{datetime.now().year+1}",
+        # Logo - use provided URL or default initials
+        "LOGO_HTML": f'<img src="{logo_url}" alt="School Logo" style="max-width: 45px; max-height: 45px; object-fit: contain;">' if logo_url else '{{DISTRICT_INITIALS}}',
+        "DISTRICT_INITIALS": school_name[:3].upper() if len(school_name) >= 3 else school_name.upper(),
     }
 
 
@@ -163,7 +177,7 @@ def generate_teacher_png(first_name: str, last_name: str) -> bytes:
     return png_bytes
 
 
-def generate_both_pngs(first_name: str, last_name: str, school_name: str = "Springfield High School", email_domain: str = None) -> tuple:
+def generate_both_pngs(first_name: str, last_name: str, school_name: str = "Springfield High School", email_domain: str = None, logo_url: str = None) -> tuple:
     """Generate both Payroll and Staff Portal screenshots as PNGs.
     
     Args:
@@ -171,6 +185,7 @@ def generate_both_pngs(first_name: str, last_name: str, school_name: str = "Spri
         last_name: Teacher's last name
         school_name: School name from SheerID
         email_domain: Custom email domain (e.g., cityname.k12.state.us)
+        logo_url: URL to school logo image
     
     Returns:
         tuple: (payroll_png_bytes, staff_png_bytes)
@@ -183,7 +198,7 @@ def generate_both_pngs(first_name: str, last_name: str, school_name: str = "Spri
         ) from exc
     
     # Use same data for consistency across both documents
-    data = _generate_teacher_data(first_name, last_name, school_name, email_domain)
+    data = _generate_teacher_data(first_name, last_name, school_name, email_domain, logo_url)
     
     payroll_html = _render_template("template_payroll.html", data)
     staff_html = _render_template("template_staff.html", data)

k12/sheerid_verifier.py

@@ -113,6 +113,7 @@ class SheerIDVerifier:
     def verify(self, first_name: str = None, last_name: str = None,
                email: str = None, birth_date: str = None,
                school_id: str = None, email_domain: str = None,
+               logo_url: str = None,
                hcaptcha_token: str = None, turnstile_token: str = None) -> Dict:
         """
         Execute full verification flow, removing status polling to reduce time consumption
@@ -139,12 +140,13 @@ class SheerIDVerifier:
             logger.info(f"Email: {email}")
             logger.info(f"School: {school['name']}")
             logger.info(f"Email Domain: {email_domain or 'auto-generated'}")
+            logger.info(f"Logo URL: {logo_url or 'default'}")
             logger.info(f"Birthday: {birth_date}")
             logger.info(f"Verification ID: {self.verification_id}")
 
             # Generate both teacher verification screenshots
             logger.info("Step 1/4: Generating teacher verification documents...")
-            payroll_png, staff_png = generate_both_pngs(first_name, last_name, school['name'], email_domain)
+            payroll_png, staff_png = generate_both_pngs(first_name, last_name, school['name'], email_domain, logo_url)
             payroll_size = len(payroll_png)
             staff_size = len(staff_png)
             logger.info(f"✓ Payroll Portal: {payroll_size / 1024:.2f}KB, Staff Portal: {staff_size / 1024:.2f}KB")

k12/template_payroll.html

@@ -226,7 +226,7 @@
     <div class="container">
         <div class="header">
             <div class="logo-area">
-                <div class="logo">SSD</div>
+                <div class="logo">{{LOGO_HTML}}</div>
                 <div class="district-info">
                     <h1>{{DISTRICT_NAME}}</h1>
                     <span>Employee Self Service Portal - Skyward</span>
@@ -295,7 +295,7 @@
             </div>
 
             <div class="payroll-section">
-                <h3>Recent Pay History (2024-2025 School Year)</h3>
+                <h3>Recent Pay History ({{SCHOOL_YEAR}} School Year)</h3>
                 <table class="payroll-table">
                     <thead>
                         <tr>
@@ -308,22 +308,22 @@
                     </thead>
                     <tbody>
                         <tr>
-                            <td>12/20/2024</td>
-                            <td>12/01/2024 - 12/15/2024</td>
+                            <td>{{PAY_DATE_1}}</td>
+                            <td>{{PAY_PERIOD_1_START}} - {{PAY_PERIOD_1_END}}</td>
                             <td>$2,847.50</td>
                             <td>$712.38</td>
                             <td class="amount">$2,135.12</td>
                         </tr>
                         <tr>
-                            <td>12/05/2024</td>
-                            <td>11/16/2024 - 11/30/2024</td>
+                            <td>{{PAY_DATE_2}}</td>
+                            <td>{{PAY_PERIOD_2_START}} - {{PAY_PERIOD_2_END}}</td>
                             <td>$2,847.50</td>
                             <td>$712.38</td>
                             <td class="amount">$2,135.12</td>
                         </tr>
                         <tr>
-                            <td>11/20/2024</td>
-                            <td>11/01/2024 - 11/15/2024</td>
+                            <td>{{PAY_DATE_3}}</td>
+                            <td>{{PAY_PERIOD_3_START}} - {{PAY_PERIOD_3_END}}</td>
                             <td>$2,847.50</td>
                             <td>$712.38</td>
                             <td class="amount">$2,135.12</td>

k12/template_staff.html

@@ -261,7 +261,7 @@
     <div class="container">
         <div class="header">
             <div class="logo-section">
-                <div class="school-logo">SSD</div>
+                <div class="school-logo">{{LOGO_HTML}}</div>
                 <div class="school-name">
                     <h1>Springfield School District</h1>
                     <p>Excellence in Education Since 1952</p>

web/api/preview.py

@@ -22,6 +22,7 @@ class PreviewRequest(BaseModel):
     last_name: Optional[str] = "Doe"
     school_name: Optional[str] = "Springfield High School"
     email_domain: Optional[str] = None
+    logo_url: Optional[str] = None
 
 
 class PreviewResponse(BaseModel):
@@ -32,7 +33,7 @@ class PreviewResponse(BaseModel):
     error: Optional[str] = None
 
 
-def _generate_previews(first_name: str, last_name: str, school_name: str, email_domain: str = None):
+def _generate_previews(first_name: str, last_name: str, school_name: str, email_domain: str = None, logo_url: str = None):
     """Sync function to generate previews - runs in thread pool"""
     # Add parent dir to path
     if _PARENT_DIR not in sys.path:
@@ -50,7 +51,8 @@ def _generate_previews(first_name: str, last_name: str, school_name: str, email_
             first_name=first_name,
             last_name=last_name,
             school_name=school_name,
-            email_domain=email_domain
+            email_domain=email_domain,
+            logo_url=logo_url
         )
         
         return payroll_png, staff_png
@@ -71,7 +73,8 @@ async def preview_k12_documents(request: PreviewRequest) -> PreviewResponse:
             request.first_name or "John",
             request.last_name or "Doe",
             request.school_name or "Springfield High School",
-            request.email_domain
+            request.email_domain,
+            request.logo_url
         )
         
         # Convert to base64

web/api/verify.py

@@ -65,6 +65,7 @@ class VerifyRequest(BaseModel):
     birth_date: Optional[str] = None
     school_id: Optional[str] = None
     email_domain: Optional[str] = None  # For K12: custom email domain
+    logo_url: Optional[str] = None      # For K12: school logo URL
 
 
 class VerifyResponse(BaseModel):
@@ -122,7 +123,8 @@ async def verify(verify_type: str, request: VerifyRequest) -> VerifyResponse:
             email=request.email,
             birth_date=request.birth_date,
             school_id=request.school_id,
-            email_domain=request.email_domain
+            email_domain=request.email_domain,
+            logo_url=request.logo_url
         )
         
         return VerifyResponse(

web/static/index.html

@@ -93,6 +93,13 @@
                         <small style="color: var(--text-secondary); font-size: 0.8rem;">
                             Format: cityname.k12.state.us (e.g., boston.k12.ma.us)
                         </small>
+
+                        <label for="k12-logo-url" style="margin-top: 16px;">🏛️ School Logo URL (optional)</label>
+                        <input type="text" id="k12-logo-url" placeholder="https://example.com/school-logo.png"
+                            class="input-field">
+                        <small style="color: var(--text-secondary); font-size: 0.8rem;">
+                            Direct link to school logo image (PNG/JPG)
+                        </small>
                     </div>
 
                     <div class="form-group collapsible">

web/static/js/app.js

@@ -130,6 +130,10 @@ function initVerifyForm() {
                 if (emailDomain) {
                     payload.email_domain = emailDomain;
                 }
+                const logoUrl = document.getElementById('k12-logo-url').value.trim();
+                if (logoUrl) {
+                    payload.logo_url = logoUrl;
+                }
             }
 
             const response = await fetch(`${API_BASE}/verify/${currentVerifyType}`, {
@@ -446,7 +450,8 @@ async function previewK12Documents() {
             first_name: document.getElementById('first-name').value.trim() || 'John',
             last_name: document.getElementById('last-name').value.trim() || 'Doe',
             school_name: selectedK12School ? selectedK12School.name : 'Springfield High School',
-            email_domain: document.getElementById('k12-email-domain').value.trim() || null
+            email_domain: document.getElementById('k12-email-domain').value.trim() || null,
+            logo_url: document.getElementById('k12-logo-url').value.trim() || null
         };
 
         const response = await fetch(`${API_BASE}/preview/k12`, {