Spaces:
Running
Running
Commit
·
6239bbc
1
Parent(s):
eded448
bug fixing
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
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:
|