annotate COBRAxy/ras_to_bounds_beta.py @ 414:5086145cfb96 draft

Uploaded
author francesco_lapi
date Mon, 08 Sep 2025 21:54:14 +0000
parents 6b015d3184ab
children 5f8f4a2d1370
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
1 import argparse
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
2 import utils.general_utils as utils
407
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
3 from typing import Optional, Dict, Set, List, Tuple
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
4 import os
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
5 import numpy as np
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
6 import pandas as pd
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
7 import cobra
407
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
8 from cobra import Model, Reaction, Metabolite
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
9 import re
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
10 import sys
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
11 import csv
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
12 from joblib import Parallel, delayed, cpu_count
414
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
13 import utils.rule_parsing as rulesUtils
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
14
407
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
15 # , medium
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
16
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
17 ################################# process args ###############################
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
18 def process_args(args :List[str] = None) -> argparse.Namespace:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
19 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
20 Processes command-line arguments.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
21
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
22 Args:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
23 args (list): List of command-line arguments.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
24
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
25 Returns:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
26 Namespace: An object containing parsed arguments.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
27 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
28 parser = argparse.ArgumentParser(usage = '%(prog)s [options]',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
29 description = 'process some value\'s')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
30
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
31
407
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
32 parser.add_argument("-mo", "--model_upload", type = str,
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
33 help = "path to input file with custom rules, if provided")
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
34
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
35 parser.add_argument('-ol', '--out_log',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
36 help = "Output log")
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
37
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
38 parser.add_argument('-td', '--tool_dir',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
39 type = str,
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
40 required = True,
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
41 help = 'your tool directory')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
42
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
43 parser.add_argument('-ir', '--input_ras',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
44 type=str,
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
45 required = False,
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
46 help = 'input ras')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
47
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
48 parser.add_argument('-rn', '--name',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
49 type=str,
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
50 help = 'ras class names')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
51
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
52 parser.add_argument('-rs', '--ras_selector',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
53 required = True,
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
54 type=utils.Bool("using_RAS"),
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
55 help = 'ras selector')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
56
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
57 parser.add_argument('-cc', '--cell_class',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
58 type = str,
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
59 help = 'output of cell class')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
60 parser.add_argument(
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
61 '-idop', '--output_path',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
62 type = str,
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
63 default='ras_to_bounds/',
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
64 help = 'output path for maps')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
65
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
66 parser.add_argument('-sm', '--save_models',
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
67 type=utils.Bool("save_models"),
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
68 default=False,
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
69 help = 'whether to save models with applied bounds')
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
70
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
71 parser.add_argument('-smp', '--save_models_path',
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
72 type = str,
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
73 default='saved_models/',
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
74 help = 'output path for saved models')
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
75
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
76 parser.add_argument('-smf', '--save_models_format',
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
77 type = str,
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
78 default='csv',
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
79 help = 'format for saved models (csv, xml, json, mat, yaml, tabular)')
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
80
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
81
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
82 ARGS = parser.parse_args(args)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
83 return ARGS
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
84
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
85 ########################### warning ###########################################
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
86 def warning(s :str) -> None:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
87 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
88 Log a warning message to an output log file and print it to the console.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
89
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
90 Args:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
91 s (str): The warning message to be logged and printed.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
92
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
93 Returns:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
94 None
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
95 """
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
96 if ARGS.out_log:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
97 with open(ARGS.out_log, 'a') as log:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
98 log.write(s + "\n\n")
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
99 print(s)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
100
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
101 ############################ dataset input ####################################
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
102 def read_dataset(data :str, name :str) -> pd.DataFrame:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
103 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
104 Read a dataset from a CSV file and return it as a pandas DataFrame.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
105
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
106 Args:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
107 data (str): Path to the CSV file containing the dataset.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
108 name (str): Name of the dataset, used in error messages.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
109
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
110 Returns:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
111 pandas.DataFrame: DataFrame containing the dataset.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
112
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
113 Raises:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
114 pd.errors.EmptyDataError: If the CSV file is empty.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
115 sys.exit: If the CSV file has the wrong format, the execution is aborted.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
116 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
117 try:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
118 dataset = pd.read_csv(data, sep = '\t', header = 0, engine='python')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
119 except pd.errors.EmptyDataError:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
120 sys.exit('Execution aborted: wrong format of ' + name + '\n')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
121 if len(dataset.columns) < 2:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
122 sys.exit('Execution aborted: wrong format of ' + name + '\n')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
123 return dataset
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
124
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
125
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
126 def apply_ras_bounds(bounds, ras_row):
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
127 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
128 Adjust the bounds of reactions in the model based on RAS values.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
129
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
130 Args:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
131 bounds (pd.DataFrame): Model bounds.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
132 ras_row (pd.Series): A row from a RAS DataFrame containing scaling factors for reaction bounds.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
133 Returns:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
134 new_bounds (pd.DataFrame): integrated bounds.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
135 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
136 new_bounds = bounds.copy()
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
137 for reaction in ras_row.index:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
138 scaling_factor = ras_row[reaction]
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
139 if not np.isnan(scaling_factor):
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
140 lower_bound=bounds.loc[reaction, "lower_bound"]
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
141 upper_bound=bounds.loc[reaction, "upper_bound"]
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
142 valMax=float((upper_bound)*scaling_factor)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
143 valMin=float((lower_bound)*scaling_factor)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
144 if upper_bound!=0 and lower_bound==0:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
145 new_bounds.loc[reaction, "upper_bound"] = valMax
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
146 if upper_bound==0 and lower_bound!=0:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
147 new_bounds.loc[reaction, "lower_bound"] = valMin
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
148 if upper_bound!=0 and lower_bound!=0:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
149 new_bounds.loc[reaction, "lower_bound"] = valMin
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
150 new_bounds.loc[reaction, "upper_bound"] = valMax
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
151 return new_bounds
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
152
414
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
153 ################################- DATA GENERATION -################################
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
154 ReactionId = str
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
155 def generate_rules(model: cobra.Model, *, asParsed = True) -> Union[Dict[ReactionId, rulesUtils.OpList], Dict[ReactionId, str]]:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
156 """
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
157 Generates a dictionary mapping reaction ids to rules from the model.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
158
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
159 Args:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
160 model : the model to derive data from.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
161 asParsed : if True parses the rules to an optimized runtime format, otherwise leaves them as strings.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
162
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
163 Returns:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
164 Dict[ReactionId, rulesUtils.OpList] : the generated dictionary of parsed rules.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
165 Dict[ReactionId, str] : the generated dictionary of raw rules.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
166 """
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
167 # Is the below approach convoluted? yes
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
168 # Ok but is it inefficient? probably
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
169 # Ok but at least I don't have to repeat the check at every rule (I'm clinically insane)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
170 _ruleGetter = lambda reaction : reaction.gene_reaction_rule
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
171 ruleExtractor = (lambda reaction :
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
172 rulesUtils.parseRuleToNestedList(_ruleGetter(reaction))) if asParsed else _ruleGetter
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
173
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
174 return {
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
175 reaction.id : ruleExtractor(reaction)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
176 for reaction in model.reactions
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
177 if reaction.gene_reaction_rule }
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
178
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
179 def generate_reactions(model :cobra.Model, *, asParsed = True) -> Dict[ReactionId, str]:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
180 """
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
181 Generates a dictionary mapping reaction ids to reaction formulas from the model.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
182
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
183 Args:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
184 model : the model to derive data from.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
185 asParsed : if True parses the reactions to an optimized runtime format, otherwise leaves them as they are.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
186
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
187 Returns:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
188 Dict[ReactionId, str] : the generated dictionary.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
189 """
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
190
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
191 unparsedReactions = {
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
192 reaction.id : reaction.reaction
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
193 for reaction in model.reactions
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
194 if reaction.reaction
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
195 }
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
196
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
197 if not asParsed: return unparsedReactions
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
198
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
199 return reactionUtils.create_reaction_dict(unparsedReactions)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
200
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
201 def get_medium(model:cobra.Model) -> pd.DataFrame:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
202 trueMedium=[]
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
203 for r in model.reactions:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
204 positiveCoeff=0
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
205 for m in r.metabolites:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
206 if r.get_coefficient(m.id)>0:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
207 positiveCoeff=1;
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
208 if (positiveCoeff==0 and r.lower_bound<0):
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
209 trueMedium.append(r.id)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
210
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
211 df_medium = pd.DataFrame()
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
212 df_medium["reaction"] = trueMedium
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
213 return df_medium
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
214
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
215 def generate_bounds(model:cobra.Model) -> pd.DataFrame:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
216
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
217 rxns = []
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
218 for reaction in model.reactions:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
219 rxns.append(reaction.id)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
220
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
221 bounds = pd.DataFrame(columns = ["lower_bound", "upper_bound"], index=rxns)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
222
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
223 for reaction in model.reactions:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
224 bounds.loc[reaction.id] = [reaction.lower_bound, reaction.upper_bound]
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
225 return bounds
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
226
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
227
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
228
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
229 def generate_compartments(model: cobra.Model) -> pd.DataFrame:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
230 """
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
231 Generates a DataFrame containing compartment information for each reaction.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
232 Creates columns for each compartment position (Compartment_1, Compartment_2, etc.)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
233
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
234 Args:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
235 model: the COBRA model to extract compartment data from.
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
236
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
237 Returns:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
238 pd.DataFrame: DataFrame with ReactionID and compartment columns
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
239 """
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
240 pathway_data = []
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
241
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
242 # First pass: determine the maximum number of pathways any reaction has
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
243 max_pathways = 0
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
244 reaction_pathways = {}
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
245
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
246 for reaction in model.reactions:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
247 # Get unique pathways from all metabolites in the reaction
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
248 if type(reaction.annotation['pathways']) == list:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
249 reaction_pathways[reaction.id] = reaction.annotation['pathways']
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
250 max_pathways = max(max_pathways, len(reaction.annotation['pathways']))
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
251 else:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
252 reaction_pathways[reaction.id] = [reaction.annotation['pathways']]
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
253
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
254 # Create column names for pathways
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
255 pathway_columns = [f"Pathway_{i+1}" for i in range(max_pathways)]
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
256
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
257 # Second pass: create the data
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
258 for reaction_id, pathways in reaction_pathways.items():
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
259 row = {"ReactionID": reaction_id}
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
260
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
261 # Fill pathway columns
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
262 for i in range(max_pathways):
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
263 col_name = pathway_columns[i]
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
264 if i < len(pathways):
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
265 row[col_name] = pathways[i]
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
266 else:
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
267 row[col_name] = None # or "" if you prefer empty strings
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
268
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
269 pathway_data.append(row)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
270
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
271 return pd.DataFrame(pathway_data)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
272
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
273 def save_model(model, filename, output_folder, file_format='csv'):
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
274 """
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
275 Save a COBRA model to file in the specified format.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
276
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
277 Args:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
278 model (cobra.Model): The model to save.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
279 filename (str): Base filename (without extension).
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
280 output_folder (str): Output directory.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
281 file_format (str): File format ('xml', 'json', 'mat', 'yaml', 'tabular', 'csv').
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
282
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
283 Returns:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
284 None
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
285 """
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
286 if not os.path.exists(output_folder):
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
287 os.makedirs(output_folder)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
288
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
289 try:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
290 if file_format == 'tabular' or file_format == 'csv':
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
291 # Special handling for tabular format using utils functions
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
292 filepath = os.path.join(output_folder, f"{filename}.csv")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
293
414
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
294 rules = generate_rules(model, asParsed = False)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
295 reactions = generate_reactions(model, asParsed = False)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
296 bounds = generate_bounds(model)
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
297 medium = get_medium(model)
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
298
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
299 try:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
300 compartments = utils.generate_compartments(model)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
301 except:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
302 compartments = None
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
303
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
304 df_rules = pd.DataFrame(list(rules.items()), columns = ["ReactionID", "Rule"])
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
305 df_reactions = pd.DataFrame(list(reactions.items()), columns = ["ReactionID", "Reaction"])
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
306 df_bounds = bounds.reset_index().rename(columns = {"index": "ReactionID"})
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
307 df_medium = medium.rename(columns = {"reaction": "ReactionID"})
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
308 df_medium["InMedium"] = True # flag per indicare la presenza nel medium
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
309
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
310 merged = df_reactions.merge(df_rules, on = "ReactionID", how = "outer")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
311 merged = merged.merge(df_bounds, on = "ReactionID", how = "outer")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
312
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
313 # Add compartments only if they exist and model name is ENGRO2
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
314 if compartments is not None and hasattr(ARGS, 'name') and ARGS.name == "ENGRO2":
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
315 merged = merged.merge(compartments, on = "ReactionID", how = "outer")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
316
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
317 merged = merged.merge(df_medium, on = "ReactionID", how = "left")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
318 merged["InMedium"] = merged["InMedium"].fillna(False)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
319 merged = merged.sort_values(by = "InMedium", ascending = False)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
320
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
321 merged.to_csv(filepath, sep="\t", index=False)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
322
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
323 else:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
324 # Standard COBRA formats
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
325 filepath = os.path.join(output_folder, f"{filename}.{file_format}")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
326
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
327 if file_format == 'xml':
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
328 cobra.io.write_sbml_model(model, filepath)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
329 elif file_format == 'json':
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
330 cobra.io.save_json_model(model, filepath)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
331 elif file_format == 'mat':
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
332 cobra.io.save_matlab_model(model, filepath)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
333 elif file_format == 'yaml':
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
334 cobra.io.save_yaml_model(model, filepath)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
335 else:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
336 raise ValueError(f"Unsupported format: {file_format}")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
337
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
338 print(f"Model saved: {filepath}")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
339
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
340 except Exception as e:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
341 warning(f"Error saving model {filename}: {str(e)}")
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
342
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
343 def apply_bounds_to_model(model, bounds):
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
344 """
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
345 Apply bounds from a DataFrame to a COBRA model.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
346
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
347 Args:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
348 model (cobra.Model): The metabolic model to modify.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
349 bounds (pd.DataFrame): DataFrame with reaction bounds.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
350
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
351 Returns:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
352 cobra.Model: Modified model with new bounds.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
353 """
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
354 model_copy = model.copy()
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
355 for reaction_id in bounds.index:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
356 try:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
357 reaction = model_copy.reactions.get_by_id(reaction_id)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
358 reaction.lower_bound = bounds.loc[reaction_id, "lower_bound"]
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
359 reaction.upper_bound = bounds.loc[reaction_id, "upper_bound"]
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
360 except KeyError:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
361 # Reaction not found in model, skip
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
362 continue
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
363 return model_copy
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
364
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
365 def process_ras_cell(cellName, ras_row, model, rxns_ids, output_folder, save_models=False, save_models_path='saved_models/', save_models_format='csv'):
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
366 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
367 Process a single RAS cell, apply bounds, and save the bounds to a CSV file.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
368
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
369 Args:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
370 cellName (str): The name of the RAS cell (used for naming the output file).
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
371 ras_row (pd.Series): A row from a RAS DataFrame containing scaling factors for reaction bounds.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
372 model (cobra.Model): The metabolic model to be modified.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
373 rxns_ids (list of str): List of reaction IDs to which the scaling factors will be applied.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
374 output_folder (str): Folder path where the output CSV file will be saved.
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
375 save_models (bool): Whether to save models with applied bounds.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
376 save_models_path (str): Path where to save models.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
377 save_models_format (str): Format for saved models.
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
378
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
379 Returns:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
380 None
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
381 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
382 bounds = pd.DataFrame([(rxn.lower_bound, rxn.upper_bound) for rxn in model.reactions], index=rxns_ids, columns=["lower_bound", "upper_bound"])
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
383 new_bounds = apply_ras_bounds(bounds, ras_row)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
384 new_bounds.to_csv(output_folder + cellName + ".csv", sep='\t', index=True)
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
385
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
386 # Save model if requested
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
387 if save_models:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
388 modified_model = apply_bounds_to_model(model, new_bounds)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
389 save_model(modified_model, cellName, save_models_path, save_models_format)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
390
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
391 pass
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
392
414
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
393 def generate_bounds_model(model: cobra.Model, ras=None, output_folder='output/', save_models=False, save_models_path='saved_models/', save_models_format='csv') -> pd.DataFrame:
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
394 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
395 Generate reaction bounds for a metabolic model based on medium conditions and optional RAS adjustments.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
396
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
397 Args:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
398 model (cobra.Model): The metabolic model for which bounds will be generated.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
399 ras (pd.DataFrame, optional): RAS pandas dataframe. Defaults to None.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
400 output_folder (str, optional): Folder path where output CSV files will be saved. Defaults to 'output/'.
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
401 save_models (bool): Whether to save models with applied bounds.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
402 save_models_path (str): Path where to save models.
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
403 save_models_format (str): Format for saved models.
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
404
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
405 Returns:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
406 pd.DataFrame: DataFrame containing the bounds of reactions in the model.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
407 """
407
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
408 rxns_ids = [rxn.id for rxn in model.reactions]
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
409
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
410 # Perform Flux Variability Analysis (FVA) on this medium
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
411 df_FVA = cobra.flux_analysis.flux_variability_analysis(model, fraction_of_optimum=0, processes=1).round(8)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
412
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
413 # Set FVA bounds
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
414 for reaction in rxns_ids:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
415 model.reactions.get_by_id(reaction).lower_bound = float(df_FVA.loc[reaction, "minimum"])
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
416 model.reactions.get_by_id(reaction).upper_bound = float(df_FVA.loc[reaction, "maximum"])
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
417
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
418 if ras is not None:
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
419 Parallel(n_jobs=cpu_count())(delayed(process_ras_cell)(
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
420 cellName, ras_row, model, rxns_ids, output_folder,
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
421 save_models, save_models_path, save_models_format
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
422 ) for cellName, ras_row in ras.iterrows())
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
423 else:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
424 bounds = pd.DataFrame([(rxn.lower_bound, rxn.upper_bound) for rxn in model.reactions], index=rxns_ids, columns=["lower_bound", "upper_bound"])
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
425 newBounds = apply_ras_bounds(bounds, pd.Series([1]*len(rxns_ids), index=rxns_ids))
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
426 newBounds.to_csv(output_folder + "bounds.csv", sep='\t', index=True)
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
427
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
428 # Save model if requested
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
429 if save_models:
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
430 modified_model = apply_bounds_to_model(model, newBounds)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
431 save_model(modified_model, "model_with_bounds", save_models_path, save_models_format)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
432
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
433 pass
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
434
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
435 ############################# main ###########################################
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
436 def main(args:List[str] = None) -> None:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
437 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
438 Initializes everything and sets the program in motion based on the fronted input arguments.
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
439
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
440 Returns:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
441 None
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
442 """
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
443 if not os.path.exists('ras_to_bounds'):
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
444 os.makedirs('ras_to_bounds')
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
445
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
446 global ARGS
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
447 ARGS = process_args(args)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
448
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
449 if(ARGS.ras_selector == True):
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
450 ras_file_list = ARGS.input_ras.split(",")
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
451 ras_file_names = ARGS.name.split(",")
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
452 if len(ras_file_names) != len(set(ras_file_names)):
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
453 error_message = "Duplicated file names in the uploaded RAS matrices."
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
454 warning(error_message)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
455 raise ValueError(error_message)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
456 pass
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
457 ras_class_names = []
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
458 for file in ras_file_names:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
459 ras_class_names.append(file.rsplit(".", 1)[0])
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
460 ras_list = []
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
461 class_assignments = pd.DataFrame(columns=["Patient_ID", "Class"])
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
462 for ras_matrix, ras_class_name in zip(ras_file_list, ras_class_names):
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
463 ras = read_dataset(ras_matrix, "ras dataset")
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
464 ras.replace("None", None, inplace=True)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
465 ras.set_index("Reactions", drop=True, inplace=True)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
466 ras = ras.T
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
467 ras = ras.astype(float)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
468 if(len(ras_file_list)>1):
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
469 #append class name to patient id (dataframe index)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
470 ras.index = [f"{idx}_{ras_class_name}" for idx in ras.index]
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
471 else:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
472 ras.index = [f"{idx}" for idx in ras.index]
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
473 ras_list.append(ras)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
474 for patient_id in ras.index:
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
475 class_assignments.loc[class_assignments.shape[0]] = [patient_id, ras_class_name]
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
476
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
477
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
478 # Concatenate all ras DataFrames into a single DataFrame
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
479 ras_combined = pd.concat(ras_list, axis=0)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
480 # Normalize the RAS values by max RAS
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
481 ras_combined = ras_combined.div(ras_combined.max(axis=0))
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
482 ras_combined.dropna(axis=1, how='all', inplace=True)
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
483
408
f413b78d61bf Uploaded
francesco_lapi
parents: 407
diff changeset
484 model = utils.build_cobra_model_from_csv(ARGS.model_upload)
407
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
485
408
f413b78d61bf Uploaded
francesco_lapi
parents: 407
diff changeset
486 validation = utils.validate_model(model)
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
487
407
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
488 print("\n=== VALIDAZIONE MODELLO ===")
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
489 for key, value in validation.items():
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
490 print(f"{key}: {value}")
6619f237aebe Uploaded
francesco_lapi
parents: 406
diff changeset
491
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
492 if(ARGS.ras_selector == True):
414
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
493 generate_bounds_model(model, ras=ras_combined, output_folder=ARGS.output_path,
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
494 save_models=ARGS.save_models, save_models_path=ARGS.save_models_path,
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
495 save_models_format=ARGS.save_models_format)
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
496 class_assignments.to_csv(ARGS.cell_class, sep='\t', index=False)
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
497 else:
414
5086145cfb96 Uploaded
francesco_lapi
parents: 411
diff changeset
498 generate_bounds_model(model, output_folder=ARGS.output_path,
411
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
499 save_models=ARGS.save_models, save_models_path=ARGS.save_models_path,
6b015d3184ab Uploaded
francesco_lapi
parents: 408
diff changeset
500 save_models_format=ARGS.save_models_format)
406
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
501
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
502 pass
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
503
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
504 ##############################################################################
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
505 if __name__ == "__main__":
187cee1a00e2 Uploaded
francesco_lapi
parents:
diff changeset
506 main()