Update app.py

app.py

@@ -191,6 +191,15 @@ def main():
 
         st.success(f"Training completed successfully!")
 
+        # Display final metrics
+        st.write("## Final Client Metrics")
+        for client in clients:
+            st.write(f"### Client {client.client_id}")
+            st.write(f"Final Loss: {client.losses[-1]:.4f}")
+            st.write(f"Final Accuracy: {client.accuracies[-1]:.4f}")
+            client.plot_metrics(NUM_ROUNDS)
+            st.write(" ")
+
     else:
         st.write("Click the 'Start Training' button to start the training process.")
 
@@ -201,6 +210,7 @@ if __name__ == "__main__":
 
 
 
+
 # # %%writefile app.py
 
 # import streamlit as st