added chem drawer

1 files changed, 3 insertions, 2 deletions

diff --git a/app/forms.py b/app/forms.py
index 3a06f27..44dadfc 100644
--- a/app/forms.py
+++ b/app/forms.py
@@ -1,6 +1,6 @@
 from flask_wtf import FlaskForm
 from flask_wtf.file import FileField, FileRequired, FileAllowed
-from wtforms import StringField, DecimalField, IntegerField
+from wtforms import StringField, DecimalField, IntegerField, SelectField
 from wtforms.validators import DataRequired, Email
 
 
@@ -35,4 +35,5 @@ class statusForm(FlaskForm):
     jobID = StringField('Job ID',validators=[DataRequired()])
 
 class generateSMILES(FlaskForm):
-    n = IntegerField('Number of Molecules to Generate',default=1,validators=[DataRequired()])
\ No newline at end of file
+    n = IntegerField('Number of Molecules to Generate',default=1,validators=[DataRequired()])
+    #modelSelection = SelectField('Model',choices=[("alpha","Alpha"),("beta","Beta")])
\ No newline at end of file