Update app.py

app.py

@@ -93,7 +93,6 @@ class CustomClient(fl.client.NumPyClient):
         loss, accuracy = test(self.net, self.testloader)
         self.losses.append(loss)
         self.accuracies.append(accuracy)
-        self.plot_metrics()
         return self.get_parameters(config={}), len(self.trainloader.dataset), {}
 
     def evaluate(self, parameters, config):
@@ -101,7 +100,11 @@ class CustomClient(fl.client.NumPyClient):
         loss, accuracy = test(self.net, self.testloader)
         return float(loss), len(self.testloader.dataset), {"accuracy": float(accuracy)}
 
-    def plot_metrics(self):
+    def plot_metrics(self, round_num):
+        st.write(f"#### Client {self.client_id} Metrics for Round {round_num}")
+        st.write(f"Loss: {self.losses[-1]:.4f}")
+        st.write(f"Accuracy: {self.accuracies[-1]:.4f}")
+
         fig, ax1 = plt.subplots()
 
         ax2 = ax1.twinx()
@@ -182,8 +185,7 @@ def main():
             )
 
             for client in clients:
-                st.write(f"### Client {client.client_id} Metrics for Round {round_num + 1}")
-                client.plot_metrics()
+                client.plot_metrics(round_num + 1)
 
         st.success(f"Training completed successfully!")
 
@@ -194,6 +196,7 @@ if __name__ == "__main__":
     main()
 
 
+
 # # %%writefile app.py
 
 # import streamlit as st