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:
|