Fix Database Restore

app.py

@@ -414,6 +414,29 @@ def predict():
     except Exception as e:
         return jsonify({"error": f"Có lỗi xảy ra khi xử lý: {str(e)}"}), 500
 
+@app.route('/admin/backup', methods=['POST'])
+@admin_required
+def manual_backup():
+    """Manual backup endpoint for admin"""
+    try:
+        if backup_database():
+            return jsonify({"success": True, "message": "Backup completed successfully"})
+        else:
+            return jsonify({"success": False, "message": "Backup failed"}), 500
+    except Exception as e:
+        return jsonify({"success": False, "message": f"Backup error: {str(e)}"}), 500
+
+@app.route('/admin/restore', methods=['POST'])
+@admin_required
+def manual_restore():
+    """Manual restore endpoint for admin"""
+    try:
+        if restore_database():
+            return jsonify({"success": True, "message": "Database restored successfully"})
+        else:
+            return jsonify({"success": False, "message": "Restore failed"}), 500
+    except Exception as e:
+        return jsonify({"success": False, "message": f"Restore error: {str(e)}"}), 500
 
 # Khởi tạo database
 with app.app_context():
@@ -422,6 +445,13 @@ with app.app_context():
     
     db.create_all()
     
+    # Create initial backup after database setup
+    print("🔄 Creating initial backup...")
+    if db_manager.backup_database():
+        print("✅ Initial backup created successfully")
+    else:
+        print("⚠️ Initial backup failed - will retry later")
+    
     # Kiểm tra và thêm cột is_admin nếu chưa có
     try:
         # Thử query để kiểm tra xem cột is_admin có tồn tại không

database_manager.py

@@ -34,6 +34,48 @@ class DatabaseManager:
                 login(token=self.hf_token)
                 self.api = HfApi()
                 print("✅ Successfully logged in to Hugging Face Hub")
+                
+                # Test token permissions with private repo
+                try:
+                    # Test READ permission - list files
+                    files = self.api.list_repo_files(repo_id=self.repo_id, repo_type="dataset")
+                    print(f"✅ Token has READ access to private dataset: {self.repo_id}")
+                    print(f"📁 Found {len(files)} files in dataset")
+                    
+                    # Test WRITE permission - create a test file
+                    test_content = {"test": "permission_check", "timestamp": str(datetime.now())}
+                    temp_file = f"{self.backup_dir}/permission_test.json"
+                    with open(temp_file, 'w') as f:
+                        json.dump(test_content, f)
+                    
+                    try:
+                        self.api.upload_file(
+                            path_or_fileobj=temp_file,
+                            path_in_repo='permission_test.json',
+                            repo_id=self.repo_id,
+                            repo_type="dataset",
+                            commit_message="Testing write permissions"
+                        )
+                        print("✅ Token has WRITE access - can upload files")
+                        
+                        # Clean up test file
+                        self.api.delete_file(
+                            path_in_repo='permission_test.json',
+                            repo_id=self.repo_id,
+                            repo_type="dataset",
+                            commit_message="Clean up test file"
+                        )
+                        os.remove(temp_file)
+                        print("✅ Token has DELETE access - can remove files")
+                        
+                    except Exception as write_error:
+                        print(f"⚠️ WRITE permission issue: {write_error}")
+                        print("💡 Your token needs 'Write' permission for backup to work")
+                        
+                except Exception as perm_error:
+                    print(f"⚠️ Token permission issue: {perm_error}")
+                    print("💡 Make sure your token has 'Write' permission")
+                    
             except Exception as e:
                 print(f"❌ Failed to login to Hugging Face Hub: {e}")
                 self.api = None
@@ -210,7 +252,11 @@ class DatabaseManager:
             return True
             
         except Exception as e:
-            print(f"❌ Restore failed: {e}")
+            if "404" in str(e):
+                print("ℹ️ No backup found yet. This is normal for first-time setup.")
+                print("💡 Backup will be created automatically after first data entry.")
+            else:
+                print(f"❌ Restore failed: {e}")
             return False
     
     def check_database_exists(self) -> bool: