Tieu Long Phan commited on
Commit
4a3f046
1 Parent(s): a072d22
Files changed (3) hide show
  1. Dockerfile +12 -0
  2. app.py +18 -0
  3. requirements.txt +4 -0
Dockerfile ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.9
2
+
3
+ RUN apt-get update && apt-get install -y python3-rdkit
4
+
5
+ COPY . /app
6
+ WORKDIR /app
7
+
8
+ RUN pip install -r requirements.txt
9
+
10
+ EXPOSE 7860
11
+ CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]
12
+
app.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from rdkit import Chem
2
+ from rdkit.Chem import Draw
3
+ from flask import Flask, request, jsonify
4
+
5
+ app = Flask(__name__)
6
+
7
+ @app.route('/convert', methods=['POST'])
8
+ def convert_smiles():
9
+ data = request.json
10
+ smiles = data['smiles']
11
+ mol = Chem.MolFromSmiles(smiles)
12
+ img = Draw.MolToImage(mol)
13
+ img.save('molecule.png')
14
+ # Add PyMol code for 3D structure
15
+ return jsonify({"message": "Conversion complete"})
16
+
17
+ if __name__ == '__main__':
18
+ app.run(debug=False)
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ gunicorn==21.2.0
2
+ flask==3.0.0
3
+ rdkit==2023.9.2
4
+ # Replace with the actual version numbers and add pymol or other packages as needed.