Fix Database Restore

app.py

@@ -22,12 +22,13 @@ app = Flask(__name__)
 app.config['SECRET_KEY'] = 'your-secret-key-change-this-in-production'
 
 # Database backup functions
-def backup_database():
+def backup_database(force: bool = False):
     """Backup database to Hugging Face Hub"""
     try:
-        db_manager.backup_database()
+        return db_manager.backup_database(force=force)
     except Exception as e:
         print(f"❌ Backup error: {e}")
+        return False
 
 def restore_database():
     """Restore database from Hugging Face Hub"""
@@ -186,6 +187,10 @@ def health():
 @login_required
 def my_statistics():
     """Trang thống kê feedback cá nhân của user"""
+    # Debug: Check user admin status
+    print(f"🔍 Debug - Current user: {current_user.username}")
+    print(f"🔍 Debug - Is admin: {current_user.is_admin}")
+    
     try:
         # Lấy thống kê feedback của user hiện tại
         user_feedbacks = Feedback.query.filter_by(user_id=current_user.id).all()
@@ -495,6 +500,11 @@ with app.app_context():
         total_users = User.query.count()
         print(f"📊 Total users in database: {total_users}")
         
+        # Debug: List all users and their admin status
+        all_users = User.query.all()
+        for user in all_users:
+            print(f"👤 User: {user.username}, Admin: {user.is_admin}")
+        
         admin_user = User.query.filter_by(username='admin').first()
         if not admin_user:
             # Chỉ tạo admin nếu database hoàn toàn trống
@@ -519,13 +529,10 @@ with app.app_context():
     
     print("✅ Database đã sẵn sàng!")
     
-    # Backup lại sau khi có thay đổi
+    # Backup lại sau khi có thay đổi (chỉ khi cần thiết)
     if total_users > 0:
-        print("🔄 Backing up restored database...")
-        if backup_database():
-            print("✅ Restored database backed up successfully")
-        else:
-            print("⚠️ Failed to backup restored database")
+        print("ℹ️ Database restored successfully with existing data")
+        print("💡 Backup will happen automatically when data changes")
 
 @app.route("/analyze-csv", methods=["POST"])
 @login_required

database_manager.py

@@ -177,7 +177,7 @@ class DatabaseManager:
         except Exception as e:
             print(f"❌ Error converting JSON to SQLite: {e}")
     
-    def backup_database(self) -> bool:
+    def backup_database(self, force: bool = False) -> bool:
         """Backup database to Hugging Face Hub"""
         if not self.api:
             print("⚠️ Hugging Face API not available. Skipping backup.")
@@ -203,13 +203,20 @@ class DatabaseManager:
                 json.dump(json_data, f, indent=2, ensure_ascii=False)
             
             # Upload to Hugging Face Hub
-            self.api.upload_file(
-                path_or_fileobj=temp_file,
-                path_in_repo='feedback_backup.json',
-                repo_id=self.repo_id,
-                repo_type="dataset",
-                commit_message=f"Backup database - {timestamp}"
-            )
+            try:
+                self.api.upload_file(
+                    path_or_fileobj=temp_file,
+                    path_in_repo='feedback_backup.json',
+                    repo_id=self.repo_id,
+                    repo_type="dataset",
+                    commit_message=f"Backup database - {timestamp}"
+                )
+            except Exception as upload_error:
+                if "No files have been modified" in str(upload_error):
+                    print("ℹ️ No changes to backup (same data as before)")
+                    return True  # This is actually success
+                else:
+                    raise upload_error
             
             # Clean up temporary file
             os.remove(temp_file)

templates/base.html

@@ -38,6 +38,9 @@
                     </span>
                     
                     {% block navigation_buttons %}
+                    <!-- Debug info (remove in production) -->
+                    <small class="text-muted">User: {{ current_user.username }} | Admin: {{ current_user.is_admin }}</small>
+                    
                     {% if current_user.is_admin %}
                     <a href="{{ url_for('view_database') }}" class="btn-glass">
                         <i class="fas fa-database"></i>