Fix Database Restore

app.py

@@ -144,6 +144,10 @@ def register():
         user.set_password(form.password.data)
         db.session.add(user)
         db.session.commit()
+        
+        # Backup database after user registration
+        backup_database()
+        
         flash('Đăng ký thành công! Vui lòng đăng nhập.', 'success')
         return redirect(url_for('login'))
     
@@ -414,6 +418,10 @@ def predict():
             )
             db.session.add(feedback)
             db.session.commit()
+            
+            # Backup database after adding feedback
+            backup_database()
+            
         except Exception as db_error:
             print(f"Database error: {db_error}")
             # Không dừng quá trình nếu lưu DB thất bại
@@ -481,25 +489,43 @@ with app.app_context():
         except Exception as e:
             print(f"⚠️ Lỗi khi thêm cột: {e}")
     
-    # Tạo tài khoản admin mặc định nếu chưa có
+    # Tạo tài khoản admin mặc định nếu chưa có (chỉ khi database trống)
     try:
+        # Kiểm tra xem có user nào trong database không
+        total_users = User.query.count()
+        print(f"📊 Total users in database: {total_users}")
+        
         admin_user = User.query.filter_by(username='admin').first()
         if not admin_user:
-            admin_user = User(username='admin', is_admin=True)
-            admin_user.set_password('123456')
-            db.session.add(admin_user)
-            db.session.commit()
-            print("✅ Đã tạo tài khoản admin mặc định (username: admin, password: 123456)")
+            # Chỉ tạo admin nếu database hoàn toàn trống
+            if total_users == 0:
+                admin_user = User(username='admin', is_admin=True)
+                admin_user.set_password('123456')
+                db.session.add(admin_user)
+                db.session.commit()
+                print("✅ Đã tạo tài khoản admin mặc định (username: admin, password: 123456)")
+            else:
+                print("ℹ️ Database có dữ liệu từ backup, không tạo admin mặc định")
         else:
             # Cập nhật user admin hiện có thành admin nếu chưa
             if not admin_user.is_admin:
                 admin_user.is_admin = True
                 db.session.commit()
                 print("✅ Đã cập nhật tài khoản admin hiện có")
+            else:
+                print("✅ Admin user đã tồn tại")
     except Exception as e:
-        print(f"⚠️ Lỗi khi tạo/cập nhật admin: {e}")
+        print(f"⚠️ Lỗi khi xử lý tài khoản admin: {e}")
     
     print("✅ Database đã sẵn sàng!")
+    
+    # Backup lại sau khi có thay đổi
+    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")
 
 @app.route("/analyze-csv", methods=["POST"])
 @login_required
@@ -667,6 +693,10 @@ def analyze_csv():
         try:
             db.session.commit()
             print(f"✅ Đã lưu {processed_count} feedback vào database")
+            
+            # Backup database after CSV processing
+            backup_database()
+            
         except Exception as commit_error:
             db.session.rollback()
             print(f"❌ Lỗi khi commit database: {commit_error}")