Add kill script and port to TensorBoard

apps/article.py

@@ -1,8 +1,9 @@
 import streamlit as st
 from apps.utils import read_markdown
-from .streamlit_tensorboard import st_tensorboard
+from .streamlit_tensorboard import st_tensorboard, kill_tensorboard
 from .utils import Toc
 def app(state=None):
+    kill_tensorboard()
     toc = Toc()
     st.info("Welcome to our Multilingual-VQA demo. Please use the navigation sidebar to move to our demo, or scroll below to read all about our project. 🤗 In case the sidebar isn't properly rendered, please change to a smaller window size and back to full screen.")
     
@@ -29,7 +30,7 @@ def app(state=None):
     st.write(read_markdown("pretraining/model.md"))
     toc.subsubheader("MLM Training Logs")
     st.info("In case the TensorBoard logs are not displayed, please visit this link: https://huggingface.co/flax-community/multilingual-vqa-pt-ckpts/tensorboard")
-    st_tensorboard(logdir='./logs/pretrain_logs')
+    st_tensorboard(logdir='./logs/pretrain_logs', port=6006)
     
     
     toc.subheader("Finetuning")
@@ -39,7 +40,7 @@ def app(state=None):
     st.write(read_markdown("finetuning/model.md"))
     toc.subsubheader("VQA Training Logs")
     st.info("In case the TensorBoard logs are not displayed, please visit this link: https://huggingface.co/flax-community/multilingual-vqa-pt-60k-ft/tensorboard")
-    st_tensorboard(logdir='./logs/finetune_logs')
+    st_tensorboard(logdir='./logs/finetune_logs', port=6007)
     
     toc.header("Challenges and Technical Difficulties")
     st.write(read_markdown("challenges.md"))

apps/streamlit_tensorboard.py

@@ -6,8 +6,24 @@ import shlex
 import random
 import html
 import json
+import os, signal
 
 
+def kill_tensorboard():
+    try:
+         
+        # iterating through each instance of the process
+        for line in os.popen("ps ax | grep " + "tensorboard" + " | grep -v grep"):
+            fields = line.split()
+             
+            # extracting Process ID from the output
+            pid = fields[0]
+             
+            # terminating process
+            os.kill(int(pid), signal.SIGKILL)
+    except:
+        print("Error Encountered while running kill script")
+
 def st_tensorboard(logdir="/logs/", port=6006, width=None, height=800, scrolling=True):
     """Embed Tensorboard within a Streamlit app
     Parameters
@@ -52,8 +68,8 @@ def st_tensorboard(logdir="/logs/", port=6006, width=None, height=800, scrolling
         </script>
     """
 
-    #args_string = f"--logdir {logdir} --port {port}"
-    args_string = f"--logdir {logdir}"
+    args_string = f"--logdir {logdir} --port {port}"
+    #args_string = f"--logdir {logdir} --host localhost"
     parsed_args = shlex.split(args_string, comments=True, posix=True)
     start_result = manager.start(parsed_args)