cafierom commited on
Commit
535ef26
·
verified ·
1 Parent(s): 7f66e42

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from rdkit import Chem
3
+ from rdkit.Chem import AllChem
4
+
5
+ def get_sdf(smile: str) -> str:
6
+ """
7
+ Convert a SMILES string to an SDF string
8
+ Args:
9
+ smile: The SMILES string of the molecule
10
+ Returns:
11
+ The SDF string of the molecule
12
+ """
13
+
14
+ mol = Chem.MolFromSmiles(smile)
15
+ molH = Chem.AddHs(mol)
16
+ AllChem.EmbedMolecule(molH)
17
+ AllChem.MMFFOptimizeMolecule(molH)
18
+
19
+ img = Chem.Draw.MolToImage(molH)
20
+
21
+ sdf = Chem.MolToMolBlock(molH)
22
+ return sdf, img
23
+
24
+
25
+ gradio_app = gr.Interface(
26
+ get_sdf,
27
+ inputs=[gr.Textbox(label="SMILES to convert to SDF")],
28
+ outputs=[gr.Textbox(label="SDF: "), gr.Image(label="Image: ")],
29
+ title="Convert SMILES to SDF",
30
+ )
31
+
32
+ if __name__ == "__main__":
33
+ gradio_app.launch(mcp_server=True)