Fix lfpykit head model load on osb

Author: filippomc

Dockerfile

@@ -57,6 +57,7 @@ ENV NEURON_HOME=/opt/conda
 # For lfpykit 0.4
 # RUN wget -P $(pip show LFPykit | grep "Location:" | awk '{print $2"/lfpykit"}') https://www.parralab.org/nyhead/sa_nyhead.mat
 # For lpfykit 0.5
+
 RUN wget --no-check-certificate -P ${FOLDER}/workspace https://www.parralab.org/nyhead/sa_nyhead.mat
 
 USER $NB_UID

netpyne_ui/__init__.py

@@ -11,8 +11,23 @@
     "https://d8bf7e40eec34cb9891f6dd8207b5e83@sentry.metacell.us/6"
 )
 
-
-
+HEAD_MODEL_FILE = os.getenv("NP_LFPYKIT_HEAD_FILE", "sa_nyhead.mat")
+
+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(HEAD_MODEL_FILE) # Downloads the model if does not exist
+    except:
+        logging.error("Error initializing the EEG head model", exc_info=True)
+
+if not os.path.exists(HEAD_MODEL_FILE):
+    from multiprocessing import Process
+    thread = Process(target = init_eeg)
+    thread.start()
 
 RouteManager.add_controller(api.NetPyNEController)