Fix file contents

app.py

@@ -223,11 +223,18 @@ def main():
         st.success("Training complete!")
         st.line_chart({"Loss": losses, "Accuracy": accuracies})
 
-    logdir = "path/to/your/logdir"
-    ea = event_accumulator.EventAccumulator(logdir)
-    ea.Reload()
-    losses = [s.value for s in ea.Scalars('loss')]
-    st.line_chart(losses)
+    logdir = "logs"  # or the actual path to your logs
+
+    if os.path.exists(logdir) and os.listdir(logdir):
+        ea = event_accumulator.EventAccumulator(logdir)
+        ea.Reload()
+        if 'loss' in ea.Tags()['scalars']:
+            losses = [s.value for s in ea.Scalars('loss')]
+            st.line_chart(losses)
+        else:
+            st.info("No 'loss' scalar found in TensorBoard logs.")
+    else:
+        st.info("No TensorBoard logs found. Please upload logs to the 'logs' directory.")
 
 if __name__ == "__main__":
     main()