#527 including NY head model file at startup

Author: filippomc

Dockerfile

@@ -53,8 +53,6 @@ RUN jupyter labextension disable @jupyterlab/hub-extension
 RUN chown  $NB_UID .
 RUN chown -R $NB_UID workspace
 
-# Temp fixes for eeg plots
-RUN wget -P `pip show LFPykit | grep "Location:" | awk '{print $2"/lfpykit"}'` https://www.parralab.org/nyhead/sa_nyhead.mat
 
 USER $NB_UID
 

netpyne_ui/__init__.py

@@ -1,9 +1,34 @@
+import logging
+import sys
+import os
+
 from jupyter_geppetto.webapi import RouteManager
 from netpyne_ui import api
+from netpyne_ui.constants import HERE
 import sentry_sdk
 
 sentry_sdk.init(
     "https://d8bf7e40eec34cb9891f6dd8207b5e83@sentry.metacell.us/6"
 )
 
+def init_eeg():
+    import sys
+    from netpyne_ui.constants import HERE
+    # FIXES library asking for input to download
+    sys.stdin = open(os.path.join(HERE, "resources/stdin.txt"),'r') 
+    from lfpykit.eegmegcalc import NYHeadModel
+    try:
+        NYHeadModel() # Downloads the model if does not exist
+    except:
+        logging.error("Error initializing the EEG head model", exc_info=True)
+
+from multiprocessing import Process
+thread = Process(target = init_eeg)
+thread.start()
+
+sys.stdin = open(os.path.join(HERE, "stdin.txt"),'r') # FIXES library asking for input to download -- eg lfpykit models
+
+
 RouteManager.add_controller(api.NetPyNEController)
+
+

netpyne_ui/netpyne_geppetto.py

@@ -75,6 +75,16 @@ def __init__(self):
         if not simulations.local.is_running():
             [experiments.set_to_error(e) for e in running_exps]
 
+        # sys.stdin = open(os.path.join(constants.HERE, "stdin.txt"),'r') # FIXES library asking for input to download -- eg lfpykit models
+
+        # from ipykernel import kernelbase
+
+        # def raw_input(self, prompt=''):
+        #     return "y"
+        # kernelbase.Kernel.raw_input = raw_input
+
+        # from lfpykit.eegmegcalc import NYHeadModel
+        # NYHeadModel()
 
     def getData(self):
         return {

netpyne_ui/resources/stdin.txt

@@ -0,0 +1,5 @@
+y
+y
+y
+y
+y