added Curie-Generate BETA

1 files changed, 5 insertions, 2 deletions

diff --git a/app/forms.py b/app/forms.py
index 47f423d..3a06f27 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
+from wtforms import StringField, DecimalField, IntegerField
 from wtforms.validators import DataRequired, Email
 
 
@@ -32,4 +32,7 @@ class curieForm(FlaskForm):
     email = StringField('Email', validators=[DataRequired(), Email()])
 
 class statusForm(FlaskForm):
-    jobID = StringField('Job ID',validators=[DataRequired()])
\ No newline at end of file
+    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