yinuozhang commited on
Commit
6239bbc
·
1 Parent(s): eded448

bug fixing

Browse files
Files changed (1) hide show
  1. app.py +19 -2
app.py CHANGED
@@ -1062,13 +1062,25 @@ def process_input(smiles_input=None, file_obj=None, show_linear=False,
1062
  # Generate ETKDG structure
1063
  mol_etkdg = generator.generate_structure_etkdg(smiles)
1064
  etkdg_bytes = generator.mol_to_sdf_bytes(mol_etkdg)
1065
- structure_files.append(("structure_etkdg.sdf", etkdg_bytes))
 
 
 
 
 
 
1066
 
1067
  # Generate UFF structure if requested
1068
  if use_uff:
1069
  mol_uff = generator.generate_structure_uff(smiles)
1070
  uff_bytes = generator.mol_to_sdf_bytes(mol_uff)
1071
- structure_files.append(("structure_uff.sdf", uff_bytes))
 
 
 
 
 
 
1072
 
1073
  except Exception as e:
1074
  return f"Error generating 3D structures: {str(e)}", None, None, []
@@ -1141,6 +1153,11 @@ def process_input(smiles_input=None, file_obj=None, show_linear=False,
1141
  #summary += f"Peptide Cycles: {', '.join(peptide_cycles)}\n"
1142
  #summary += f"Aromatic Cycles: {', '.join(aromatic_cycles)}\n"
1143
 
 
 
 
 
 
1144
  return summary + output_text, img_cyclic, img_linear, structure_files
1145
 
1146
  except Exception as e:
 
1062
  # Generate ETKDG structure
1063
  mol_etkdg = generator.generate_structure_etkdg(smiles)
1064
  etkdg_bytes = generator.mol_to_sdf_bytes(mol_etkdg)
1065
+
1066
+ # Create temporary file path for ETKDG structure
1067
+ etkdg_file = {
1068
+ "name": "structure_etkdg.sdf",
1069
+ "data": etkdg_bytes
1070
+ }
1071
+ structure_files.append(etkdg_file)
1072
 
1073
  # Generate UFF structure if requested
1074
  if use_uff:
1075
  mol_uff = generator.generate_structure_uff(smiles)
1076
  uff_bytes = generator.mol_to_sdf_bytes(mol_uff)
1077
+
1078
+ # Create temporary file path for UFF structure
1079
+ uff_file = {
1080
+ "name": "structure_uff.sdf",
1081
+ "data": uff_bytes
1082
+ }
1083
+ structure_files.append(uff_file)
1084
 
1085
  except Exception as e:
1086
  return f"Error generating 3D structures: {str(e)}", None, None, []
 
1153
  #summary += f"Peptide Cycles: {', '.join(peptide_cycles)}\n"
1154
  #summary += f"Aromatic Cycles: {', '.join(aromatic_cycles)}\n"
1155
 
1156
+ if structure_files:
1157
+ summary += "\n3D Structures Generated:\n"
1158
+ for file_info in structure_files:
1159
+ summary += f"- {file_info['name']}\n"
1160
+
1161
  return summary + output_text, img_cyclic, img_linear, structure_files
1162
 
1163
  except Exception as e: