server.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "import flwr as fl\n",
    "from typing import Dict, Optional, Tuple, List\n",
    "from flwr.common import Metrics, Parameters, Scalar, FitRes, EvaluateRes\n",
    "import logging\n",
    "from datetime import datetime\n",
    "import time"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "2025-03-09 15:13:03,252 - Server - INFO - \n",
      "==================== SERVER STARTING ====================\n",
      "2025-03-09 15:13:03,253 - Server - INFO - Initializing Flower server\n",
      "Address: 127.0.0.1:8081\n",
      "Expected clients: 1\n",
      "Message size: 1073741824\n",
      "You can now start connecting clients one by one\n",
      "2025-03-09 15:13:03,254 - Server - INFO - ==========================================================\n",
      "\u001b[93mWARNING \u001b[0m:   DEPRECATED FEATURE: flwr.server.start_server() is deprecated.\n",
      "\tInstead, use the `flower-superlink` CLI command to start a SuperLink as shown below:\n",
      "\n",
      "\t\t$ flower-superlink --insecure\n",
      "\n",
      "\tTo view usage and all available options, run:\n",
      "\n",
      "\t\t$ flower-superlink --help\n",
      "\n",
      "\tUsing `start_server()` is deprecated.\n",
      "\n",
      "            This is a deprecated feature. It will be removed\n",
      "            entirely in future versions of Flower.\n",
      "        \n",
      "2025-03-09 15:13:03,255 - flwr - WARNING - DEPRECATED FEATURE: flwr.server.start_server() is deprecated.\n",
      "\tInstead, use the `flower-superlink` CLI command to start a SuperLink as shown below:\n",
      "\n",
      "\t\t$ flower-superlink --insecure\n",
      "\n",
      "\tTo view usage and all available options, run:\n",
      "\n",
      "\t\t$ flower-superlink --help\n",
      "\n",
      "\tUsing `start_server()` is deprecated.\n",
      "\n",
      "            This is a deprecated feature. It will be removed\n",
      "            entirely in future versions of Flower.\n",
      "        \n",
      "\u001b[92mINFO \u001b[0m:      Starting Flower server, config: num_rounds=3, no round_timeout\n",
      "2025-03-09 15:13:03,256 - flwr - INFO - Starting Flower server, config: num_rounds=3, no round_timeout\n",
      "\u001b[92mINFO \u001b[0m:      Flower ECE: gRPC server running (3 rounds), SSL is disabled\n",
      "2025-03-09 15:13:03,275 - flwr - INFO - Flower ECE: gRPC server running (3 rounds), SSL is disabled\n",
      "\u001b[92mINFO \u001b[0m:      [INIT]\n",
      "2025-03-09 15:13:03,278 - flwr - INFO - [INIT]\n",
      "\u001b[92mINFO \u001b[0m:      Requesting initial parameters from one random client\n",
      "2025-03-09 15:13:03,279 - flwr - INFO - Requesting initial parameters from one random client\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== SERVER STARTING ====================\n",
      "Initializing Flower server\n",
      "Address: 127.0.0.1:8081\n",
      "Expected clients: 1\n",
      "Message size: 1073741824\n",
      "You can now start connecting clients one by one\n",
      "==========================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\u001b[92mINFO \u001b[0m:      Received initial parameters from one random client\n",
      "2025-03-09 15:13:48,077 - flwr - INFO - Received initial parameters from one random client\n",
      "\u001b[92mINFO \u001b[0m:      Starting evaluation of initial global parameters\n",
      "2025-03-09 15:13:48,080 - flwr - INFO - Starting evaluation of initial global parameters\n",
      "\u001b[92mINFO \u001b[0m:      Evaluation returned no results (`None`)\n",
      "2025-03-09 15:13:48,082 - flwr - INFO - Evaluation returned no results (`None`)\n",
      "\u001b[92mINFO \u001b[0m:      \n",
      "2025-03-09 15:13:48,083 - flwr - INFO - \n",
      "\u001b[92mINFO \u001b[0m:      [ROUND 1]\n",
      "2025-03-09 15:13:48,084 - flwr - INFO - [ROUND 1]\n",
      "2025-03-09 15:13:48,085 - Server - INFO - \n",
      "==================== NEW CLIENT CONNECTED ====================\n",
      "2025-03-09 15:13:48,086 - Server - INFO - Round 1\n",
      "Total connected: 1/1\n",
      "New client(s): 1\n",
      "Waiting for 0 more\n",
      "2025-03-09 15:13:48,086 - Server - INFO - ===============================================================\n",
      "2025-03-09 15:13:48,087 - Server - INFO - \n",
      "==================== STARTING ROUND ====================\n",
      "2025-03-09 15:13:48,087 - Server - INFO - Round 1: All 1 clients connected\n",
      "Beginning training round\n",
      "2025-03-09 15:13:48,088 - Server - INFO - =========================================================\n",
      "\u001b[92mINFO \u001b[0m:      configure_fit: strategy sampled 1 clients (out of 1)\n",
      "2025-03-09 15:13:48,088 - flwr - INFO - configure_fit: strategy sampled 1 clients (out of 1)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== NEW CLIENT CONNECTED ====================\n",
      "Round 1\n",
      "Total connected: 1/1\n",
      "New client(s): 1\n",
      "Waiting for 0 more\n",
      "===============================================================\n",
      "\n",
      "==================== STARTING ROUND ====================\n",
      "Round 1: All 1 clients connected\n",
      "Beginning training round\n",
      "=========================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\u001b[92mINFO \u001b[0m:      aggregate_fit: received 1 results and 0 failures\n",
      "2025-03-09 15:21:18,601 - flwr - INFO - aggregate_fit: received 1 results and 0 failures\n",
      "2025-03-09 15:21:18,611 - Server - INFO - \n",
      "==================== ROUND COMPLETE ====================\n",
      "2025-03-09 15:21:18,612 - Server - INFO - Round 1 completed\n",
      "Successful clients: 1\n",
      "Failed clients: 0\n",
      "2025-03-09 15:21:18,613 - Server - INFO - =========================================================\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== ROUND COMPLETE ====================\n",
      "Round 1 completed\n",
      "Successful clients: 1\n",
      "Failed clients: 0\n",
      "=========================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\u001b[93mWARNING \u001b[0m:   No fit_metrics_aggregation_fn provided\n",
      "2025-03-09 15:21:19,436 - flwr - WARNING - No fit_metrics_aggregation_fn provided\n",
      "2025-03-09 15:21:19,504 - Server - INFO - \n",
      "==================== EVALUATION ====================\n",
      "2025-03-09 15:21:19,505 - Server - INFO - Round 1: Starting evaluation\n",
      "2025-03-09 15:21:19,506 - Server - INFO - =====================================================\n",
      "\u001b[92mINFO \u001b[0m:      configure_evaluate: strategy sampled 1 clients (out of 1)\n",
      "2025-03-09 15:21:19,509 - flwr - INFO - configure_evaluate: strategy sampled 1 clients (out of 1)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== EVALUATION ====================\n",
      "Round 1: Starting evaluation\n",
      "=====================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\u001b[92mINFO \u001b[0m:      aggregate_evaluate: received 0 results and 1 failures\n",
      "2025-03-09 15:21:25,747 - flwr - INFO - aggregate_evaluate: received 0 results and 1 failures\n",
      "2025-03-09 15:21:25,750 - Server - INFO - \n",
      "==================== EVALUATION COMPLETE ====================\n",
      "2025-03-09 15:21:25,750 - Server - INFO - Round 1\n",
      "Successful evaluations: 0\n",
      "Failed evaluations: 1\n",
      "2025-03-09 15:21:25,751 - Server - INFO - ==============================================================\n",
      "\u001b[92mINFO \u001b[0m:      \n",
      "2025-03-09 15:21:25,754 - flwr - INFO - \n",
      "\u001b[92mINFO \u001b[0m:      [ROUND 2]\n",
      "2025-03-09 15:21:25,755 - flwr - INFO - [ROUND 2]\n",
      "2025-03-09 15:21:25,757 - Server - INFO - \n",
      "==================== WAITING ====================\n",
      "2025-03-09 15:21:25,758 - Server - INFO - Round 2: Waiting for more clients\n",
      "Currently connected: 0/1\n",
      "You can start 1 more client(s)\n",
      "2025-03-09 15:21:25,759 - Server - INFO - ==================================================\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== EVALUATION COMPLETE ====================\n",
      "Round 1\n",
      "Successful evaluations: 0\n",
      "Failed evaluations: 1\n",
      "==============================================================\n",
      "\n",
      "==================== WAITING ====================\n",
      "Round 2: Waiting for more clients\n",
      "Currently connected: 0/1\n",
      "You can start 1 more client(s)\n",
      "==================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\u001b[92mINFO \u001b[0m:      configure_fit: strategy sampled 1 clients (out of 1)\n",
      "2025-03-09 15:21:30,968 - flwr - INFO - configure_fit: strategy sampled 1 clients (out of 1)\n",
      "\u001b[92mINFO \u001b[0m:      aggregate_fit: received 1 results and 0 failures\n",
      "2025-03-09 15:25:50,394 - flwr - INFO - aggregate_fit: received 1 results and 0 failures\n",
      "2025-03-09 15:25:50,400 - Server - INFO - \n",
      "==================== ROUND COMPLETE ====================\n",
      "2025-03-09 15:25:50,402 - Server - INFO - Round 2 completed\n",
      "Successful clients: 1\n",
      "Failed clients: 0\n",
      "2025-03-09 15:25:50,402 - Server - INFO - =========================================================\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== ROUND COMPLETE ====================\n",
      "Round 2 completed\n",
      "Successful clients: 1\n",
      "Failed clients: 0\n",
      "=========================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "2025-03-09 15:25:51,243 - Server - INFO - \n",
      "==================== EVALUATION ====================\n",
      "2025-03-09 15:25:51,244 - Server - INFO - Round 2: Starting evaluation\n",
      "2025-03-09 15:25:51,246 - Server - INFO - =====================================================\n",
      "\u001b[92mINFO \u001b[0m:      configure_evaluate: strategy sampled 1 clients (out of 1)\n",
      "2025-03-09 15:25:51,247 - flwr - INFO - configure_evaluate: strategy sampled 1 clients (out of 1)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== EVALUATION ====================\n",
      "Round 2: Starting evaluation\n",
      "=====================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\u001b[92mINFO \u001b[0m:      aggregate_evaluate: received 0 results and 1 failures\n",
      "2025-03-09 15:25:55,868 - flwr - INFO - aggregate_evaluate: received 0 results and 1 failures\n",
      "2025-03-09 15:25:55,870 - Server - INFO - \n",
      "==================== EVALUATION COMPLETE ====================\n",
      "2025-03-09 15:25:55,871 - Server - INFO - Round 2\n",
      "Successful evaluations: 0\n",
      "Failed evaluations: 1\n",
      "2025-03-09 15:25:55,871 - Server - INFO - ==============================================================\n",
      "\u001b[92mINFO \u001b[0m:      \n",
      "2025-03-09 15:25:55,954 - flwr - INFO - \n",
      "\u001b[92mINFO \u001b[0m:      [ROUND 3]\n",
      "2025-03-09 15:25:55,956 - flwr - INFO - [ROUND 3]\n",
      "2025-03-09 15:25:55,957 - Server - INFO - \n",
      "==================== WAITING ====================\n",
      "2025-03-09 15:25:55,958 - Server - INFO - Round 3: Waiting for more clients\n",
      "Currently connected: 0/1\n",
      "You can start 1 more client(s)\n",
      "2025-03-09 15:25:55,959 - Server - INFO - ==================================================\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== EVALUATION COMPLETE ====================\n",
      "Round 2\n",
      "Successful evaluations: 0\n",
      "Failed evaluations: 1\n",
      "==============================================================\n",
      "\n",
      "==================== WAITING ====================\n",
      "Round 3: Waiting for more clients\n",
      "Currently connected: 0/1\n",
      "You can start 1 more client(s)\n",
      "==================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\u001b[92mINFO \u001b[0m:      configure_fit: strategy sampled 1 clients (out of 1)\n",
      "2025-03-09 15:26:01,029 - flwr - INFO - configure_fit: strategy sampled 1 clients (out of 1)\n",
      "\u001b[92mINFO \u001b[0m:      aggregate_fit: received 1 results and 0 failures\n",
      "2025-03-09 15:30:25,808 - flwr - INFO - aggregate_fit: received 1 results and 0 failures\n",
      "2025-03-09 15:30:25,818 - Server - INFO - \n",
      "==================== ROUND COMPLETE ====================\n",
      "2025-03-09 15:30:25,818 - Server - INFO - Round 3 completed\n",
      "Successful clients: 1\n",
      "Failed clients: 0\n",
      "2025-03-09 15:30:25,820 - Server - INFO - =========================================================\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== ROUND COMPLETE ====================\n",
      "Round 3 completed\n",
      "Successful clients: 1\n",
      "Failed clients: 0\n",
      "=========================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "2025-03-09 15:30:26,634 - Server - INFO - \n",
      "==================== EVALUATION ====================\n",
      "2025-03-09 15:30:26,635 - Server - INFO - Round 3: Starting evaluation\n",
      "2025-03-09 15:30:26,636 - Server - INFO - =====================================================\n",
      "\u001b[92mINFO \u001b[0m:      configure_evaluate: strategy sampled 1 clients (out of 1)\n",
      "2025-03-09 15:30:26,637 - flwr - INFO - configure_evaluate: strategy sampled 1 clients (out of 1)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== EVALUATION ====================\n",
      "Round 3: Starting evaluation\n",
      "=====================================================\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\u001b[92mINFO \u001b[0m:      aggregate_evaluate: received 0 results and 1 failures\n",
      "2025-03-09 15:30:38,578 - flwr - INFO - aggregate_evaluate: received 0 results and 1 failures\n",
      "2025-03-09 15:30:38,580 - Server - INFO - \n",
      "==================== EVALUATION COMPLETE ====================\n",
      "2025-03-09 15:30:38,580 - Server - INFO - Round 3\n",
      "Successful evaluations: 0\n",
      "Failed evaluations: 1\n",
      "2025-03-09 15:30:38,581 - Server - INFO - ==============================================================\n",
      "\u001b[92mINFO \u001b[0m:      \n",
      "2025-03-09 15:30:38,692 - flwr - INFO - \n",
      "\u001b[92mINFO \u001b[0m:      [SUMMARY]\n",
      "2025-03-09 15:30:38,695 - flwr - INFO - [SUMMARY]\n",
      "\u001b[92mINFO \u001b[0m:      Run finished 3 round(s) in 1010.54s\n",
      "2025-03-09 15:30:38,696 - flwr - INFO - Run finished 3 round(s) in 1010.54s\n",
      "\u001b[92mINFO \u001b[0m:      \n",
      "2025-03-09 15:30:38,698 - flwr - INFO - \n",
      "2025-03-09 15:30:38,738 - Server - INFO - \n",
      "==================== SERVER COMPLETED ====================\n",
      "2025-03-09 15:30:38,744 - Server - INFO - Training completed successfully\n",
      "Total rounds: 3\n",
      "2025-03-09 15:30:38,744 - Server - INFO - ===========================================================\n",
      "2025-03-09 15:30:38,745 - Server - INFO - \n",
      "==================== SERVER SHUTDOWN ====================\n",
      "2025-03-09 15:30:38,746 - Server - INFO - Flower server has shut down\n",
      "2025-03-09 15:30:38,746 - Server - INFO - ==========================================================\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "==================== EVALUATION COMPLETE ====================\n",
      "Round 3\n",
      "Successful evaluations: 0\n",
      "Failed evaluations: 1\n",
      "==============================================================\n",
      "\n",
      "==================== SERVER COMPLETED ====================\n",
      "Training completed successfully\n",
      "Total rounds: 3\n",
      "===========================================================\n",
      "\n",
      "Training Round Metrics:\n",
      "\n",
      "Round 1:\n",
      "Timestamp: 2025-03-09T15:21:18.614139\n",
      "Active clients: 1\n",
      "Failed clients: 0\n",
      "\n",
      "Round 2:\n",
      "Timestamp: 2025-03-09T15:25:50.403659\n",
      "Active clients: 1\n",
      "Failed clients: 0\n",
      "\n",
      "Round 3:\n",
      "Timestamp: 2025-03-09T15:30:25.821534\n",
      "Active clients: 1\n",
      "Failed clients: 0\n",
      "\n",
      "==================== SERVER SHUTDOWN ====================\n",
      "Flower server has shut down\n",
      "==========================================================\n"
     ]
    }
   ],
   "source": [
    "# Configure logging\n",
    "logging.basicConfig(\n",
    "    level=logging.INFO,\n",
    "    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'\n",
    ")\n",
    "logger = logging.getLogger(\"Server\")\n",
    "\n",
    "def log_status(status: str, details: str = \"\"):\n",
    "    status_line = f\"\\n{'='*20} {status} {'='*20}\"\n",
    "    logger.info(status_line)\n",
    "    if details:\n",
    "        logger.info(details)\n",
    "    logger.info(\"=\"*len(status_line))\n",
    "    print(status_line)\n",
    "    if details:\n",
    "        print(details)\n",
    "    print(\"=\"*len(status_line))\n",
    "\n",
    "class LoggingStrategy(fl.server.strategy.FedAvg):\n",
    "    def __init__(\n",
    "        self,\n",
    "        *,\n",
    "        fraction_fit: float = 1.0,\n",
    "        fraction_evaluate: float = 1.0,\n",
    "        min_fit_clients: int = 1,\n",
    "        min_evaluate_clients: int = 1,\n",
    "        min_available_clients: int = 1,\n",
    "        timeout: Optional[float] = None,\n",
    "    ) -> None:\n",
    "        super().__init__(\n",
    "            fraction_fit=fraction_fit,\n",
    "            fraction_evaluate=fraction_evaluate,\n",
    "            min_fit_clients=min_fit_clients,\n",
    "            min_evaluate_clients=min_evaluate_clients,\n",
    "            min_available_clients=min_available_clients,\n",
    "        )\n",
    "        self.round_metrics = []\n",
    "        self.expected_clients = min_available_clients\n",
    "        self.timeout = timeout\n",
    "        self.current_round = 0\n",
    "        self.connected_clients = set()\n",
    "\n",
    "    def configure_fit(\n",
    "        self, server_round: int, parameters: Parameters, client_manager\n",
    "    ) -> List[Tuple[fl.server.client_proxy.ClientProxy, fl.common.FitIns]]:\n",
    "        self.current_round = server_round\n",
    "        connected = client_manager.num_available()\n",
    "        \n",
    "        # Get newly connected clients\n",
    "        current_clients = set(client_manager.all().keys())\n",
    "        new_clients = current_clients - self.connected_clients\n",
    "        self.connected_clients.update(new_clients)\n",
    "        \n",
    "        if new_clients:\n",
    "            log_status(\"NEW CLIENT CONNECTED\", \n",
    "                      f\"Round {server_round}\\n\"\n",
    "                      f\"Total connected: {connected}/{self.expected_clients}\\n\"\n",
    "                      f\"New client(s): {len(new_clients)}\\n\"\n",
    "                      f\"Waiting for {self.expected_clients - connected} more\")\n",
    "        \n",
    "        if connected < self.expected_clients:\n",
    "            log_status(\"WAITING\", \n",
    "                      f\"Round {server_round}: Waiting for more clients\\n\"\n",
    "                      f\"Currently connected: {connected}/{self.expected_clients}\\n\"\n",
    "                      f\"You can start {self.expected_clients - connected} more client(s)\")\n",
    "            # Sleep to prevent too frequent logging\n",
    "            time.sleep(2)\n",
    "        else:\n",
    "            log_status(\"STARTING ROUND\", \n",
    "                      f\"Round {server_round}: All {self.expected_clients} clients connected\\n\"\n",
    "                      f\"Beginning training round\")\n",
    "        \n",
    "        return super().configure_fit(server_round, parameters, client_manager)\n",
    "\n",
    "    def aggregate_fit(\n",
    "        self,\n",
    "        server_round: int,\n",
    "        results: List[Tuple[fl.server.client_proxy.ClientProxy, FitRes]],\n",
    "        failures: List[BaseException],\n",
    "    ) -> Tuple[Optional[Parameters], Dict[str, Scalar]]:\n",
    "        log_status(\"ROUND COMPLETE\", \n",
    "                  f\"Round {server_round} completed\\n\"\n",
    "                  f\"Successful clients: {len(results)}\\n\"\n",
    "                  f\"Failed clients: {len(failures)}\")\n",
    "        \n",
    "        # Store metrics\n",
    "        round_data = {\n",
    "            \"round\": server_round,\n",
    "            \"timestamp\": datetime.now().isoformat(),\n",
    "            \"num_clients\": len(results),\n",
    "            \"num_failures\": len(failures)\n",
    "        }\n",
    "        self.round_metrics.append(round_data)\n",
    "        \n",
    "        return super().aggregate_fit(server_round, results, failures)\n",
    "\n",
    "    def configure_evaluate(\n",
    "        self, server_round: int, parameters: Parameters, client_manager\n",
    "    ) -> List[Tuple[fl.server.client_proxy.ClientProxy, fl.common.EvaluateIns]]:\n",
    "        log_status(\"EVALUATION\", f\"Round {server_round}: Starting evaluation\")\n",
    "        return super().configure_evaluate(server_round, parameters, client_manager)\n",
    "\n",
    "    def aggregate_evaluate(\n",
    "        self,\n",
    "        server_round: int,\n",
    "        results: List[Tuple[fl.server.client_proxy.ClientProxy, EvaluateRes]],\n",
    "        failures: List[BaseException],\n",
    "    ) -> Tuple[Optional[float], Dict[str, Scalar]]:\n",
    "        log_status(\"EVALUATION COMPLETE\", \n",
    "                  f\"Round {server_round}\\n\"\n",
    "                  f\"Successful evaluations: {len(results)}\\n\"\n",
    "                  f\"Failed evaluations: {len(failures)}\")\n",
    "        return super().aggregate_evaluate(server_round, results, failures)\n",
    "\n",
    "# Define a larger message size (1GB)\n",
    "GRPC_MAX_MESSAGE_LENGTH = 1024 * 1024 * 1024\n",
    "\n",
    "# Initialize strategy with longer timeout\n",
    "strategy = LoggingStrategy(\n",
    "    min_fit_clients=1,\n",
    "    min_available_clients=1,\n",
    "    min_evaluate_clients=1,\n",
    "    timeout=None  # No timeout, wait indefinitely for clients\n",
    ")\n",
    "\n",
    "log_status(\"SERVER STARTING\", \n",
    "          \"Initializing Flower server\\n\"\n",
    "          f\"Address: 127.0.0.1:8081\\n\"\n",
    "          f\"Expected clients: 1\\n\"\n",
    "          f\"Message size: {GRPC_MAX_MESSAGE_LENGTH}\\n\"\n",
    "          f\"You can now start connecting clients one by one\")\n",
    "\n",
    "# Start server\n",
    "try:\n",
    "    fl.server.start_server(\n",
    "        server_address=\"127.0.0.1:8081\",\n",
    "        config=fl.server.ServerConfig(\n",
    "            num_rounds=3,\n",
    "            round_timeout=None  # No timeout for rounds\n",
    "        ),\n",
    "        grpc_max_message_length=GRPC_MAX_MESSAGE_LENGTH,\n",
    "        strategy=strategy\n",
    "    )\n",
    "    \n",
    "    log_status(\"SERVER COMPLETED\", \n",
    "              f\"Training completed successfully\\n\"\n",
    "              f\"Total rounds: {len(strategy.round_metrics)}\")\n",
    "    \n",
    "    # Print final metrics\n",
    "    print(\"\\nTraining Round Metrics:\")\n",
    "    for round_data in strategy.round_metrics:\n",
    "        print(f\"\\nRound {round_data['round']}:\")\n",
    "        print(f\"Timestamp: {round_data['timestamp']}\")\n",
    "        print(f\"Active clients: {round_data['num_clients']}\")\n",
    "        print(f\"Failed clients: {round_data['num_failures']}\")\n",
    "    \n",
    "except Exception as e:\n",
    "    log_status(\"SERVER ERROR\", str(e))\n",
    "finally:\n",
    "    log_status(\"SERVER SHUTDOWN\", \"Flower server has shut down\")"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "base",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.7"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}