# Based on: https://github.com/allenai/s2-folks/tree/main/examples/python
# 
# 

import os
import re

from neuromllite.utils import load_simulation_json
from neuromllite.utils import load_network_json
from neuromllite.NetworkGenerator import generate_and_run

MODELS = {'Muscle model': ["models/Sim_IClamp_GenericMuscleCell.json","models/IClamp_GenericMuscleCell.json"],
          'Neuron model': ["models/Sim_IClamp_GenericNeuronCell.json","models/IClamp_GenericNeuronCell.json"]}

def run_model(text, model, verbose = False):

    info = '''Running [%s] with parameter: [%s]...

'''%(model,text)

    print(info)

    sim_file = MODELS[model][0]
    net_file = MODELS[model][1]

    sim = load_simulation_json(sim_file)
    net = load_network_json(net_file)
    

    net.parameters["stim_amp"] = text.strip()


    if verbose:
        info+="""```%s```

"""%sim.to_json()
        info+="""```%s```

"""%net.to_json()

    traces, events = generate_and_run(sim, 
                                      network=net, 
                                      simulator="jNeuroML",
                                      base_dir='./models',
                                      target_dir='./models',
                                      return_results=True)

    #info+="""Results returned: %s"""%(list(traces.keys()))
    info+="""Finished simulation!"""

    

    print(info)

    return info, traces, events