Update photo album and database logic for better Hugging Face integration

app/database.py

@@ -820,17 +820,20 @@ class PhotoAlbum(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     title = db.Column(db.String(255), nullable=False)
     is_public = db.Column(db.Boolean, default=False, nullable=False)
+    representative_photo_id = db.Column(db.Integer, db.ForeignKey('photo.id'), nullable=True)
     created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
     updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)
 
     # 관계
-    photos = db.relationship('Photo', backref='album', lazy=True, cascade='all, delete-orphan')
+    photos = db.relationship('Photo', foreign_keys='Photo.album_id', backref='album', lazy=True, cascade='all, delete-orphan')
+    representative_photo = db.relationship('Photo', foreign_keys=[representative_photo_id], post_update=True)
 
     def to_dict(self):
         return {
             'id': self.id,
             'title': self.title,
             'is_public': self.is_public,
+            'representative_photo_id': self.representative_photo_id,
             'created_at': self.created_at.isoformat() if self.created_at else None,
             'updated_at': self.updated_at.isoformat() if self.updated_at else None,
             'photo_count': len(self.photos) if self.photos else 0

app/i18n.py

@@ -39,7 +39,10 @@ TRANSLATIONS = {
         'file_too_large': 'Vượt quá dung lượng tải lên tối đa (500MB). Vui lòng chia nhỏ tệp để tải lên.',
         'db_error': 'Kết nối máy chủ DB đang chậm lại. Chúng tôi sẽ phản hồi nhanh chóng. Vui lòng liên hệ Lee Chang-woo / rarcissus@soymedia.kr',
         'no_resource': 'Không tìm thấy tài nguyên.',
-        'upload_limit_exceeded': 'Đã vượt quá dung lượng tải lên tối đa.'
+        'upload_limit_exceeded': 'Đã vượt quá dung lượng tải lên tối đa.',
+        'representative': 'Đại diện',
+        'set_representative': 'Đặt làm đại diện',
+        'thumbnail': 'Ảnh thu nhỏ'
     },
     'ko': {
         'photo_album': '사진첩',
@@ -81,7 +84,10 @@ TRANSLATIONS = {
         'file_too_large': '업로드 가능한 최대 용량(500MB)을 초과했습니다. 파일을 나누어 업로드해주세요.',
         'db_error': 'DB 서버 연결이 느려지고 있습니다. 빠르게 대응하겠습니다. 담당자 이창우 / rarcissus@soymedia.kr 로 연락 주시길 바랍니다',
         'no_resource': '리소스를 찾을 수 없습니다.',
-        'upload_limit_exceeded': '업로드 가능한 최대 용량을 초과했습니다.'
+        'upload_limit_exceeded': '업로드 가능한 최대 용량을 초과했습니다.',
+        'representative': '대표',
+        'set_representative': '대표 설정',
+        'thumbnail': '썸네일'
     }
 }
 

app/routes.py

@@ -184,12 +184,22 @@ def check_db_schema():
     try:
         from sqlalchemy import inspect, text
         inspector = inspect(db.engine)
+        
+        # User 테이블
         columns = [c['name'] for c in inspector.get_columns('user')]
         if 'role' not in columns:
             print("Adding 'role' column to user table...")
             with db.engine.connect() as conn:
                 conn.execute(text("ALTER TABLE user ADD COLUMN role VARCHAR(20) DEFAULT 'user' NOT NULL"))
                 conn.commit()
+                
+        # PhotoAlbum 테이블
+        columns = [c['name'] for c in inspector.get_columns('photo_album')]
+        if 'representative_photo_id' not in columns:
+            print("Adding 'representative_photo_id' column to photo_album table...")
+            with db.engine.connect() as conn:
+                conn.execute(text("ALTER TABLE photo_album ADD COLUMN representative_photo_id INTEGER REFERENCES photo(id)"))
+                conn.commit()
     except Exception as e:
         print(f"Schema check error: {e}")
 
@@ -6332,6 +6342,11 @@ def admin_photo_album_photo_delete(photo_id):
     photo = Photo.query.get_or_404(photo_id)
     album_id = photo.album_id
     
+    # 만약 삭제하는 사진이 대표 사진이면 null 처리
+    album = PhotoAlbum.query.get(album_id)
+    if album and album.representative_photo_id == photo_id:
+        album.representative_photo_id = None
+    
     # 실제 파일 삭제
     base_path = os.path.dirname(os.path.dirname(__file__))
     full_file_path = os.path.join(base_path, photo.file_path)
@@ -6363,6 +6378,19 @@ def admin_photo_album_photo_delete(photo_id):
     flash('사진이 삭제되었습니다.', 'success')
     return redirect(url_for('main.admin_photo_album_photos', album_id=album_id))
 
+@main_bp.route('/admin/photo-album/photo/<int:photo_id>/representative', methods=['POST'])
+@admin_required
+def admin_photo_album_photo_representative(photo_id):
+    """대표 사진 설정"""
+    photo = Photo.query.get_or_404(photo_id)
+    album = PhotoAlbum.query.get_or_404(photo.album_id)
+    
+    album.representative_photo_id = photo.id
+    db.session.commit()
+    
+    flash('대표 사진이 설정되었습니다.', 'success')
+    return redirect(url_for('main.admin_photo_album_photos', album_id=album.id))
+
 # -----------------------------
 # 사진첩 (사용자용)
 # -----------------------------

templates/admin_photo_album.html

@@ -41,6 +41,14 @@
             background-color: #fce8e6;
             color: #d93025;
         }
+        .admin-thumbnail {
+            width: 60px;
+            height: 60px;
+            object-fit: cover;
+            border-radius: 6px;
+            background-color: #f1f3f4;
+            border: 1px solid #dee2e6;
+        }
     </style>
 </head>
 <body>
@@ -90,6 +98,7 @@
                     <thead>
                         <tr>
                             <th>ID</th>
+                            <th>{{ gettext('thumbnail') }}</th>
                             <th>{{ gettext('album_title') }}</th>
                             <th>{{ gettext('photos_count', count='')|replace(' ', '')|replace('{}', '') }}</th>
                             <th>{{ gettext('public_status') }}</th>
@@ -101,6 +110,16 @@
                         {% for album in albums %}
                         <tr>
                             <td>{{ album.id }}</td>
+                            <td>
+                                {% set display_photo = album.representative_photo if album.representative_photo else (album.photos[0] if album.photos else None) %}
+                                {% if display_photo %}
+                                    <img src="{{ url_for('static', filename=display_photo.thumbnail_path.replace('static/', '')) if display_photo.thumbnail_path.startswith('static/') else '/' + display_photo.thumbnail_path }}" class="admin-thumbnail">
+                                {% else %}
+                                    <div class="admin-thumbnail d-flex align-items-center justify-content-center">
+                                        <span class="text-muted" style="font-size: 0.6rem;">N/A</span>
+                                    </div>
+                                {% endif %}
+                            </td>
                             <td>
                                 <a href="{{ url_for('main.admin_photo_album_photos', album_id=album.id) }}" class="text-decoration-none fw-medium text-dark">
                                     {{ album.title }}

templates/admin_photo_album_photos.html

@@ -31,6 +31,15 @@
             overflow: hidden;
             box-shadow: 0 1px 3px rgba(0,0,0,0.1);
             position: relative;
+            transition: transform 0.2s;
+        }
+        .photo-card:hover {
+            transform: translateY(-2px);
+            box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+        }
+        .photo-card.representative {
+            outline: 3px solid #1a73e8;
+            outline-offset: -3px;
         }
         .photo-card img {
             width: 100%;
@@ -39,6 +48,11 @@
         }
         .photo-actions {
             padding: 10px;
+            display: flex;
+            flex-direction: column;
+            gap: 8px;
+        }
+        .photo-info {
             display: flex;
             justify-content: space-between;
             align-items: center;
@@ -49,10 +63,23 @@
             background: none;
             border: none;
             font-size: 0.8rem;
+            padding: 0;
         }
         .delete-btn:hover {
             text-decoration: underline;
         }
+        .representative-badge {
+            position: absolute;
+            top: 10px;
+            right: 10px;
+            background: #1a73e8;
+            color: white;
+            padding: 2px 8px;
+            border-radius: 4px;
+            font-size: 0.75rem;
+            font-weight: 500;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.2);
+        }
         .drop-zone {
             border: 2px dashed #dadce0;
             border-radius: 8px;
@@ -143,17 +170,34 @@
 
         <div class="photo-grid">
             {% for photo in album.photos %}
-            <div class="photo-card">
+            <div class="photo-card {{ 'representative' if album.representative_photo_id == photo.id else '' }}">
+                {% if album.representative_photo_id == photo.id %}
+                    <div class="representative-badge">{{ gettext('representative') }}</div>
+                {% endif %}
+                
                 {% if photo.thumbnail_path %}
                     <img src="{{ url_for('static', filename=photo.thumbnail_path.replace('static/', '')) if photo.thumbnail_path.startswith('static/') else '/' + photo.thumbnail_path }}" alt="{{ photo.original_filename }}">
                 {% else %}
                     <img src="{{ url_for('static', filename=photo.file_path.replace('static/', '')) if photo.file_path.startswith('static/') else '/' + photo.file_path }}" alt="{{ photo.original_filename }}">
                 {% endif %}
+                
                 <div class="photo-actions">
-                    <span class="text-truncate small text-muted" style="max-width: 120px;" title="{{ photo.original_filename }}">{{ photo.original_filename }}</span>
-                    <form action="{{ url_for('main.admin_photo_album_photo_delete', photo_id=photo.id) }}" method="POST" onsubmit="return confirm('{{ gettext('delete') }}?')">
-                        <button type="submit" class="delete-btn">{{ gettext('delete') }}</button>
-                    </form>
+                    <div class="photo-info">
+                        <span class="text-truncate small text-muted" style="max-width: 150px;" title="{{ photo.original_filename }}">{{ photo.original_filename }}</span>
+                    </div>
+                    <div class="d-flex justify-content-between align-items-center mt-1">
+                        {% if album.representative_photo_id != photo.id %}
+                        <form action="{{ url_for('main.admin_photo_album_photo_representative', photo_id=photo.id) }}" method="POST">
+                            <button type="submit" class="btn btn-sm btn-outline-primary py-0 px-2" style="font-size: 0.75rem;">{{ gettext('set_representative') }}</button>
+                        </form>
+                        {% else %}
+                        <span></span>
+                        {% endif %}
+                        
+                        <form action="{{ url_for('main.admin_photo_album_photo_delete', photo_id=photo.id) }}" method="POST" onsubmit="return confirm('{{ gettext('delete') }}?')">
+                            <button type="submit" class="delete-btn">{{ gettext('delete') }}</button>
+                        </form>
+                    </div>
                 </div>
             </div>
             {% endfor %}

templates/photo_album_list.html

@@ -116,11 +116,11 @@
                 <a href="{{ url_for('main.photo_album_view', album_id=album.id) }}" class="album-card">
                     <div class="album-thumbnail">
                         {% if album.photos %}
-                            {% set first_photo = album.photos[0] %}
-                            {% if first_photo.thumbnail_path %}
-                                <img src="/{{ first_photo.thumbnail_path }}" alt="{{ album.title }}">
+                            {% set display_photo = album.representative_photo if album.representative_photo else album.photos[0] %}
+                            {% if display_photo.thumbnail_path %}
+                                <img src="{{ url_for('static', filename=display_photo.thumbnail_path.replace('static/', '')) if display_photo.thumbnail_path.startswith('static/') else '/' + display_photo.thumbnail_path }}" alt="{{ album.title }}">
                             {% else %}
-                                <img src="/{{ first_photo.file_path }}" alt="{{ album.title }}">
+                                <img src="{{ url_for('static', filename=display_photo.file_path.replace('static/', '')) if display_photo.file_path.startswith('static/') else '/' + display_photo.file_path }}" alt="{{ album.title }}">
                             {% endif %}
                         {% else %}
                             <div class="text-muted">{{ gettext('no_photos') }}</div>