Update app.py

app.py

@@ -568,7 +568,7 @@ def parse_log(log_lines):
     clients = {}
     memory_usage = []
 
-    round_pattern = re.compile(r'\[ROUND (\d+)\]')
+    round_pattern = re.compile(r'ROUND(\d+)ROUND (\d+)')
     client_pattern = re.compile(r'Client (\d+) \| (INFO|DEBUG) \| (.*)')
     memory_pattern = re.compile(r'memory used=(\d+\.\d+)GB')
 
@@ -709,7 +709,7 @@ def main():
             st.write(f"### Round {round_num + 1} ✅")
 
             logs = read_log_file2()
-
+            st.markdown(logs)
             # Extract relevant data
             accuracy_pattern = re.compile(r"'accuracy': \{(\d+), ([\d.]+)\}")
             loss_pattern = re.compile(r"'loss': \{(\d+), ([\d.]+)\}")
@@ -779,7 +779,7 @@ def main():
         # Display log.txt content
         st.write("## Training Log")
         st.write(read_log_file2())
-
+        
         st.write("## Training Log Analysis")
         log_lines = read_log_file()
         rounds, clients, memory_usage = parse_log(log_lines)