working base model

requirements.txt

@@ -0,0 +1,151 @@
+altair==5.0.1 ; python_version >= '3.7'
+anyio==3.7.1 ; python_version >= '3.7'
+argon2-cffi==21.3.0 ; python_version >= '3.6'
+argon2-cffi-bindings==21.2.0 ; python_version >= '3.6'
+arrow==1.2.3 ; python_version >= '3.6'
+arviz==0.12.1 ; python_version >= '3.7'
+asdm==0.0.3
+asttokens==2.2.1
+async-lru==2.0.3 ; python_version >= '3.8'
+attrs==23.1.0 ; python_version >= '3.7'
+babel==2.12.1 ; python_version >= '3.7'
+backcall==0.2.0
+beautifulsoup4==4.12.2 ; python_full_version >= '3.6.0'
+bleach==6.0.0 ; python_version >= '3.7'
+blinker==1.6.2 ; python_version >= '3.7'
+cachetools==5.3.1 ; python_version >= '3.7'
+certifi==2023.5.7 ; python_version >= '3.6'
+cffi==1.15.1
+cftime==1.6.2 ; python_version >= '3.7'
+charset-normalizer==3.2.0 ; python_full_version >= '3.7.0'
+click==8.1.5 ; python_version >= '3.7'
+colorama==0.4.6 ; platform_system == 'Windows'
+comm==0.1.3 ; python_version >= '3.6'
+contourpy==1.1.0 ; python_version >= '3.8'
+cycler==0.11.0 ; python_version >= '3.6'
+debugpy==1.6.7 ; python_version >= '3.7'
+decorator==5.1.1 ; python_version >= '3.5'
+defusedxml==0.7.1 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+deprecat==2.1.1 ; python_version >= '3.6'
+dill==0.3.6 ; python_version >= '3.7'
+et-xmlfile==1.1.0 ; python_version >= '3.6'
+exceptiongroup==1.1.2 ; python_version < '3.11'
+executing==1.2.0
+fastjsonschema==2.17.1
+fastprogress==1.0.3 ; python_version >= '3.6'
+filelock==3.12.2 ; python_version >= '3.7'
+fonttools==4.41.0 ; python_version >= '3.8'
+fqdn==1.5.1
+gitdb==4.0.10 ; python_version >= '3.7'
+gitpython==3.1.32 ; python_version >= '3.7'
+idna==3.4 ; python_version >= '3.5'
+importlib-metadata==6.8.0 ; python_version >= '3.8'
+ipykernel==6.24.0
+ipython==8.14.0 ; python_version >= '3.9'
+ipywidgets==8.0.7 ; python_version >= '3.7'
+isoduration==20.11.0
+jedi==0.18.2 ; python_version >= '3.6'
+jinja2==3.1.2 ; python_version >= '3.7'
+json5==0.9.14
+jsonpointer==2.4
+jsonschema==4.18.3 ; python_version >= '3.8'
+jsonschema-specifications==2023.6.1 ; python_version >= '3.8'
+jupyter-client==8.3.0 ; python_version >= '3.8'
+jupyter-core==5.3.1 ; python_version >= '3.8'
+jupyter-events==0.6.3 ; python_version >= '3.7'
+jupyter-lsp==2.2.0 ; python_version >= '3.8'
+jupyter-server==2.7.0 ; python_version >= '3.8'
+jupyter-server-terminals==0.4.4 ; python_version >= '3.8'
+jupyterlab==4.0.3
+jupyterlab-pygments==0.2.2 ; python_version >= '3.7'
+jupyterlab-server==2.23.0 ; python_version >= '3.7'
+jupyterlab-widgets==3.0.8 ; python_version >= '3.7'
+kiwisolver==1.4.4 ; python_version >= '3.7'
+lxml==4.9.3 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+markdown-it-py==3.0.0 ; python_version >= '3.8'
+markupsafe==2.1.3 ; python_version >= '3.7'
+matplotlib==3.7.2 ; python_version >= '3.8'
+matplotlib-inline==0.1.6 ; python_version >= '3.5'
+mdurl==0.1.2 ; python_version >= '3.7'
+mistune==3.0.1 ; python_version >= '3.7'
+nbclient==0.8.0 ; python_full_version >= '3.8.0'
+nbconvert==7.6.0 ; python_version >= '3.7'
+nbformat==5.9.1 ; python_version >= '3.8'
+nest-asyncio==1.5.6 ; python_version >= '3.5'
+netcdf4==1.6.4 ; python_version >= '3.7'
+networkx==3.1 ; python_version >= '3.8'
+notebook-shim==0.2.3 ; python_version >= '3.7'
+numpy==1.22.1 ; python_version >= '3.8'
+openpyxl==3.1.2 ; python_version >= '3.6'
+overrides==7.3.1 ; python_version >= '3.6'
+packaging==23.1 ; python_version >= '3.7'
+pandas==2.0.3 ; python_version >= '3.8'
+pandocfilters==1.5.0 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+parso==0.8.3 ; python_version >= '3.6'
+patsy==0.5.3
+pickleshare==0.7.5
+pillow==9.5.0 ; python_version >= '3.7'
+platformdirs==3.9.1 ; python_version >= '3.7'
+plotly==5.15.0
+prometheus-client==0.17.1 ; python_version >= '3.6'
+prompt-toolkit==3.0.39 ; python_full_version >= '3.7.0'
+protobuf==4.23.4 ; python_version >= '3.7'
+psutil==5.9.5 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+pure-eval==0.2.2
+pyarrow==12.0.1 ; python_version >= '3.7'
+pycparser==2.21
+pydeck==0.8.1b0 ; python_version >= '3.7'
+pygments==2.15.1 ; python_version >= '3.7'
+pymc3==3.11.5 ; python_version >= '3.7'
+pympler==1.0.1 ; python_version >= '3.6'
+pyparsing==3.0.9 ; python_full_version >= '3.6.8'
+python-dateutil==2.8.2 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+python-json-logger==2.0.7 ; python_version >= '3.6'
+pytz==2023.3
+pytz-deprecation-shim==0.1.0.post0 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4, 3.5'
+pywin32==306 ; sys_platform == 'win32' and platform_python_implementation != 'PyPy'
+pywinpty==2.0.11 ; os_name == 'nt'
+pyyaml==6.0 ; python_version >= '3.6'
+pyzmq==25.1.0 ; python_version >= '3.6'
+referencing==0.29.1 ; python_version >= '3.8'
+requests==2.31.0 ; python_version >= '3.7'
+rfc3339-validator==0.1.4 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+rfc3986-validator==0.1.1 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+rich==13.4.2 ; python_full_version >= '3.7.0'
+rpds-py==0.8.10 ; python_version >= '3.8'
+scipy==1.7.3 ; python_version < '3.11' and python_version >= '3.7'
+semver==3.0.1 ; python_version >= '3.7'
+send2trash==1.8.2 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+setuptools==68.0.0 ; python_version >= '3.7'
+six==1.16.0 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+smmap==5.0.0 ; python_version >= '3.6'
+sniffio==1.3.0 ; python_version >= '3.7'
+soupsieve==2.4.1 ; python_version >= '3.7'
+stack-data==0.6.2
+streamlit==1.24.1
+tenacity==8.2.2 ; python_version >= '3.6'
+terminado==0.17.1 ; python_version >= '3.7'
+theano-pymc==1.1.2
+tinycss2==1.2.1 ; python_version >= '3.7'
+toml==0.10.2 ; python_version >= '2.6' and python_version not in '3.0, 3.1, 3.2, 3.3'
+tomli==2.0.1 ; python_version < '3.11'
+toolz==0.12.0 ; python_version >= '3.5'
+tornado==6.3.2 ; python_version >= '3.8'
+tqdm==4.65.0 ; python_version >= '3.7'
+traitlets==5.9.0 ; python_version >= '3.7'
+typing-extensions==4.7.1 ; python_version >= '3.7'
+tzdata==2023.3 ; python_version >= '2'
+tzlocal==4.3.1 ; python_version >= '3.7'
+uri-template==1.3.0
+urllib3==2.0.3 ; python_version >= '3.7'
+validators==0.20.0 ; python_version >= '3.4'
+watchdog==3.0.0 ; platform_system != 'Darwin'
+wcwidth==0.2.6
+webcolors==1.13
+webencodings==0.5.1
+websocket-client==1.6.1 ; python_version >= '3.7'
+widgetsnbextension==4.0.8 ; python_version >= '3.7'
+wrapt==1.15.0 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+xarray==2023.6.0 ; python_version >= '3.9'
+xarray-einstats==0.6.0 ; python_version >= '3.9'
+zipp==3.16.2 ; python_version >= '3.8'

streamlit_app_asdm.py

@@ -0,0 +1,57 @@
+import pandas as pd
+import streamlit as st
+import plotly.graph_objects as go
+from ASDM.ASDM import Structure
+
+def load_model(model_path):
+    try:
+        model = Structure(from_xmile=model_path)
+    except FileNotFoundError:
+        st.error(f"File {model_path} not found.")
+        return None
+    return model
+
+def run_simulation(model, simulation_time, re_investment):
+    model.sim_specs['initial_time'] = 0
+    model.sim_specs['current_time'] = 0
+    model.sim_specs['dt'] = 1
+    model.sim_specs['simulation_time'] = simulation_time
+    model.sim_specs['time_units'] = 'Months'
+
+    model.clear_last_run()
+
+    model.aux_equations['percentageOfSavingsSpentOnCessation'] = str(re_investment)
+
+    model.simulate()
+
+    results = model.export_simulation_result()
+    results_df = pd.DataFrame.from_dict(results)
+
+    columns_to_plot = ["Current_smokers", "Ex_smokers", "Ex_smokers_starting_again"]
+    return results_df['Months'], results_df[columns_to_plot]
+
+st.title('Smoking Cessation')
+
+st.markdown("""
+This simulation estimates the effects of various reinvestment levels in a smoking cessation service within a population of 900 smokers. 
+By varying the proportion of savings that are reinvested into the service, we can observe different outcomes in terms of current smokers, ex-smokers, 
+and ex-smokers who start smoking again over time.
+""")
+            
+st.subheader('Slide the Slider to Vary Re-Investment Levels')
+
+model = load_model('models/smoking cessation demo.stmx')
+
+if model is not None:
+    re_investment = st.slider("Proportion of Savings Spent on Cessation", 0, 100, 45)
+    simulation_time = st.slider("Select the number of months to simulate:", min_value=1, max_value=36, value=24)
+
+    x_values, y_values = run_simulation(model, simulation_time, re_investment)
+
+    st.subheader('Effects of Re-Investment on Smoking Levels')
+
+    fig = go.Figure()
+    for column in y_values.columns:
+        fig.add_trace(go.Scatter(x=x_values, y=y_values[column], mode='lines', name=column))
+    fig.update_layout(xaxis_title='Months', yaxis_title='Number of Smokers', autosize=False, width=800, height=500)
+    st.plotly_chart(fig)