Update app.py

app.py

@@ -14,9 +14,7 @@ GRADE_LIST = ["KG1", "KG2"] + [f"Grade {i}" for i in range(1, 13)]
 SECTION_LIST = ["A", "B", "C", "D"]
 current_year = 2025
 AY_LIST = [f"AY {y}-{str(y+1)[2:]}" for y in range(current_year, current_year + 6)]
-
 GOLD_COLOR = (184, 134, 11) 
-DARK_BLUE = (0, 31, 63)
 
 # --- Loading Logic ---
 if 'ai_loaded' not in st.session_state:
@@ -65,67 +63,69 @@ with tab1:
         pdf = FPDF()
         for student in data:
             pdf.add_page()
-            
-            # 1. Border
             pdf.set_draw_color(*border_rgb)
             pdf.set_line_width(5)
             pdf.rect(10, 10, 190, 277)
             
-            # 2. Logo
             if os.path.exists(LOGO_PATH):
                 pdf.image(LOGO_PATH, x=170, y=12, w=20) 
             
-            # 3. Header Info
             pdf.set_xy(15, 15)
             pdf.set_font("Arial", 'B', 11)
             pdf.set_text_color(160, 160, 160)
             pdf.cell(0, 10, f"{term} | {week} | {ay}", ln=True)
 
-            # 4. AI Background Removal
+            # AI Smart Crop & Background Removal
             raw_img = Image.open(student["image"])
-            processed_img = st.session_state['remove_fn'](raw_img)
-            processed_img = processed_img.convert("RGBA")
+            processed_img = st.session_state['remove_fn'](raw_img, alpha_matting=True)
             
+            # Tight crop to remove empty space around head
+            bbox = processed_img.getbbox()
+            if bbox:
+                processed_img = processed_img.crop(bbox)
+
             with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
                 processed_img.save(tmp.name, format='PNG')
                 temp_path = tmp.name
 
-            # 5. Golden Title (Remains at top)
+            # Title, Name, Grade
             pdf.set_y(28) 
             pdf.set_font("Arial", 'B', 32)
             pdf.set_text_color(*GOLD_COLOR) 
             pdf.cell(200, 20, "STAR OF THE WEEK", ln=True, align='C')
 
-            # 6. Student Name (NEW POSITION: Top, after title)
-            # Size reduced from 40 to 32 for better header integration
             pdf.set_y(48)
             pdf.set_font("Arial", 'B', 32)
             pdf.set_text_color(*border_rgb)
             pdf.cell(200, 12, student["name"].upper(), ln=True, align='C')
             
-            # 7. Grade-Section (NEW POSITION: Top, after name)
             pdf.set_font("Arial", '', 22)
             pdf.set_text_color(100, 100, 100)
             pdf.cell(200, 10, f"{student['grade']}-{student['section']}", ln=True, align='C')
 
-            # 8. Student Photo (MAXIMIZED AREA)
-            # Photo now starts at y=75 and has the rest of the page available
-            pdf.image(temp_path, x=32, y=75, w=145, h=0)
+            # Photo Placement
+            pdf.image(temp_path, x=32, y=75, w=145, h=190)
             os.unlink(temp_path)
             
         return pdf.output(dest='S').encode('latin-1')
 
+    # --- Generation and Auto-Visible Button ---
     if st.button("✨ Generate Star of the Week Photos"):
         if not students_data:
             st.warning("Please fill in student details.")
         else:
-            with st.status("🎨 AI is creating your photos...", expanded=True) as status:
+            # We use a standard spinner so the output isn't nested inside a box
+            with st.spinner("🎨 AI is Smart-Cropping and generating..."):
                 pdf_out = create_pdf(students_data, rgb_border, selected_term, selected_week, selected_ay)
-                status.update(label="Photos Ready!", state="complete", expanded=False)
                 
-                st.download_button(
-                    label="📥 Download Photos PDF",
-                    data=pdf_out,
-                    file_name=f"Star_of_the_Week_{selected_ay}_{selected_term}_{selected_week}.pdf",
-                    mime="application/pdf"
-                )
+            # These lines execute immediately after the spinner finishes
+            # No more drop-down arrow!
+            st.balloons() # Added a celebratory effect
+            st.success("✅ Photos Ready!")
+            st.download_button(
+                label="📥 Download Photos PDF",
+                data=pdf_out,
+                file_name=f"Star_of_the_Week_{selected_ay}_{selected_term}_{selected_week}.pdf",
+                mime="application/pdf",
+                use_container_width=True # Makes the button large and easy to click
+            )