Update app.py

app.py

@@ -535,6 +535,57 @@ def main():
         for round_num in range(NUM_ROUNDS):
             st.write(f"### Round {round_num + 1}")
             st.markdown(read_log_file2())
+            logs = read_log_file2()
+            import re
+            import plotly.graph_objects as go
+            import streamlit as st
+            import pandas as pd
+            
+            # Log data
+            log_data = logs
+            
+            # Extract relevant data
+            accuracy_pattern = re.compile(r"'accuracy': \((\d+),([\d.]+)\)\((\d+), ([\d.]+)\)")
+            loss_pattern = re.compile(r"'loss': \((\d+),([\d.]+)\)\((\d+), ([\d.]+)\)")
+            
+            accuracy_matches = accuracy_pattern.findall(log_data)
+            loss_matches = loss_pattern.findall(log_data)
+            
+            rounds = [int(match[0]) for match in accuracy_matches]
+            accuracies = [float(match[1]) for match in accuracy_matches]
+            losses = [float(match[1]) for match in loss_matches]
+            
+            # Create accuracy plot
+            accuracy_fig = go.Figure()
+            accuracy_fig.add_trace(go.Scatter(x=rounds, y=accuracies, mode='lines+markers', name='Accuracy'))
+            accuracy_fig.update_layout(title='Accuracy over Rounds', xaxis_title='Round', yaxis_title='Accuracy')
+            
+            # Create loss plot
+            loss_fig = go.Figure()
+            loss_fig.add_trace(go.Scatter(x=rounds, y=losses, mode='lines+markers', name='Loss'))
+            loss_fig.update_layout(title='Loss over Rounds', xaxis_title='Round', yaxis_title='Loss')
+            
+            # Display plots in Streamlit
+            st.plotly_chart(accuracy_fig)
+            st.plotly_chart(loss_fig)
+            
+            # Display data table
+            data = {
+                'Round': rounds,
+                'Accuracy': accuracies,
+                'Loss': losses
+            }
+            
+            df = pd.DataFrame(data)
+            st.write("## Training Metrics")
+            st.table(df)
+
+
+
+
+
+
+            
             plot_placeholders = [st.empty() for _ in range(NUM_CLIENTS)]
 
             fl.simulation.start_simulation(