Update app.py

app.py

@@ -637,13 +637,16 @@ def process_photos_quality_selection(job_id, upload_dir, quality_mode, similarit
             stat['categories'] = cat_breakdown
 
         # Step 6: Build rejected list (photos not selected)
+        # Note: rejection_reason is already set by monthly_selector.py
         selected_filenames = {p['filename'] for p in selected_photos}
         rejected_photos = []
 
         for month, photos in photos_by_month.items():
             for photo in photos:
                 if photo['filename'] not in selected_filenames:
-                    photo['rejection_reason'] = 'Not selected for month quota'
+                    # Keep existing rejection_reason from monthly_selector, or set default
+                    if not photo.get('rejection_reason'):
+                        photo['rejection_reason'] = 'Not selected for month quota'
                     rejected_photos.append(photo)
 
         # Create thumbnails directory
@@ -756,9 +759,21 @@ def process_photos_quality_selection(job_id, upload_dir, quality_mode, similarit
             if thumbnails_created % 10 == 0 or thumbnails_created == total_thumbnails:
                 processing_jobs[job_id]['message'] = f'Creating thumbnails: {thumbnails_created}/{total_thumbnails}'
 
-            # Simple rejection reason
-            reason = "Better photos selected"
-            rejection_counts[reason] += 1
+            # Use actual rejection reason from monthly_selector
+            rejection_reason = photo.get('rejection_reason', 'Better photos selected')
+
+            # Categorize rejection reasons for breakdown chart
+            if 'Event' in rejection_reason:
+                breakdown_category = "Same event"
+            elif 'Cluster' in rejection_reason:
+                breakdown_category = "Same cluster"
+            elif 'similar' in rejection_reason.lower():
+                breakdown_category = "Too similar"
+            elif 'Target' in rejection_reason:
+                breakdown_category = "Target reached"
+            else:
+                breakdown_category = "Other"
+            rejection_counts[breakdown_category] += 1
 
             # Get embedding for this photo (convert to list for JSON serialization)
             photo_embedding = embeddings.get(filename)
@@ -785,7 +800,8 @@ def process_photos_quality_selection(job_id, upload_dir, quality_mode, similarit
                 'event_id': photo.get('event_id', -1),
                 'max_similarity': float(photo.get('max_similarity', 0)),
                 'embedding': embedding_list,
-                'reason': reason,
+                'rejection_reason': rejection_reason,
+                'reason': rejection_reason,
                 'reason_detail': f"Category: {photo.get('category', 'unknown')}"
             })
 
@@ -814,11 +830,11 @@ def process_photos_quality_selection(job_id, upload_dir, quality_mode, similarit
             print(f"  {stat['month']}: {stat['selected']}/{stat['total_photos']} ({stat['category_summary']})")
         print(f"{'='*60}\n")
 
-        # Auto-save both selected and not-selected photos organized by month
-        output_folder = save_photos_by_month(job_id, upload_dir, selected_photos, rejected_photos, month_stats)
-        if output_folder:
-            processing_jobs[job_id]['output_folder'] = output_folder
-            print(f"[Job {job_id}] Photos auto-saved to: {output_folder}")
+        # Auto-save disabled - uncomment below to re-enable
+        # output_folder = save_photos_by_month(job_id, upload_dir, selected_photos, rejected_photos, month_stats)
+        # if output_folder:
+        #     processing_jobs[job_id]['output_folder'] = output_folder
+        #     print(f"[Job {job_id}] Photos auto-saved to: {output_folder}")
 
     except Exception as e:
         print(f"[Job {job_id}] EXCEPTION: {str(e)}")
@@ -3271,6 +3287,7 @@ def finish_chunked_upload(dataset_name):
 @app.route('/process_reupload/<dataset_name>', methods=['POST'])
 def process_reupload(dataset_name):
     """Process re-uploaded photos using saved face results from Supabase."""
+    from werkzeug.exceptions import ClientDisconnected
     try:
         print(f"[Reupload] Starting reupload for dataset '{dataset_name}'")
 
@@ -3427,6 +3444,9 @@ def process_reupload(dataset_name):
             'redirect_url': f'/step3_review/{job_id}'
         })
 
+    except ClientDisconnected:
+        print(f"[Reupload] Client disconnected during upload (timeout)")
+        return jsonify({'error': 'Connection timeout - please retry with smaller batch or better connection'}), 408
     except Exception as e:
         import traceback
         traceback.print_exc()
@@ -3448,4 +3468,4 @@ if __name__ == '__main__':
     # Use port 7860 for Hugging Face Spaces, 5000 for local
     import os
     port = int(os.environ.get('PORT', 7860))
-    app.run(debug=False, host='0.0.0.0', port=port)
+    app.run(debug=False, host='0.0.0.0', port=port)