Update app.py

app.py

@@ -1018,11 +1018,18 @@ def read_log_file2():
 def main():
     st.write("## Federated Learning with Dynamic Models and Datasets for Mobile Devices")
     logs = read_log_file2()
-    st.markdown(logs)
+    # cleanLogs = # Define a pattern to match relevant log entries
+    pattern = re.compile(r"memory|loss|accuracy|round|client", re.IGNORECASE)
+    
+
+    # Filter the log data
+    filtered_logs = [line for line in logs.splitlines() if pattern.search(line)]
+    st.markdown(filtered_logs)
+
     # Provide a download button for the logs
     st.download_button(
         label="Download Logs",
-        data=logs,
+        data=filtered_logs,
         file_name="./log.txt",
         mime="text/plain"
     )
@@ -1087,7 +1094,10 @@ def main():
             st.write(f"### Round {round_num + 1} ✅")
 
             logs = read_log_file2()
-            st.markdown(logs)
+            filtered_logs = [line for line in logs.splitlines() if pattern.search(line)]
+
+            
+            st.markdown(filtered_logs)
             # Provide a download button for the logs
             # st.download_button(
             #     label="Download Logs",
@@ -1099,8 +1109,8 @@ def main():
             accuracy_pattern = re.compile(r"'accuracy': \{(\d+), ([\d.]+)\}")
             loss_pattern = re.compile(r"'loss': \{(\d+), ([\d.]+)\}")
 
-            accuracy_matches = accuracy_pattern.findall(logs)
-            loss_matches = loss_pattern.findall(logs)
+            accuracy_matches = accuracy_pattern.findall(filtered_logs)
+            loss_matches = loss_pattern.findall(filtered_logs)
 
             rounds = [int(match[0]) for match in accuracy_matches]
             accuracies = [float(match[1]) for match in accuracy_matches]