Allow multiple file upload and add title

app/services/forms.py

@@ -1,25 +1,17 @@
 from flask_wtf import FlaskForm
-from flask_wtf.file import FileAllowed, FileRequired
+from wtforms import MultipleFileField, SelectField, StringField, SubmitField
-from wtforms import FileField, SelectField, StringField, SubmitField
 from wtforms.validators import DataRequired, Length
 
 
-class OCRJobForm(FlaskForm):
+class CreateOCRJobForm(FlaskForm):
     description = StringField(
         'Description',
         validators=[DataRequired(), Length(1, 64)]
     )
-    file = FileField(
+    files = MultipleFileField('Files', validators=[DataRequired()])
-        'File',
-        validators=[
-            FileAllowed(['pdf', 'tif', 'tiff']),
-            FileRequired()
-        ]
-    )
     language = SelectField(
         'Language',
-        choices=[
+        choices=[('', 'Choose your option'),
-            ('', 'Choose your option'),
                  ('eng', 'English'),
                  ('enm', 'English, Middle (1100-1500)'),
                  ('fra', 'French'),
@@ -32,4 +24,8 @@ class OCRJobForm(FlaskForm):
                  ],
         validators=[DataRequired()]
     )
-    submit = SubmitField('Submit')
+    submit = SubmitField('Create OCR job')
+    title = StringField(
+        'Title',
+        validators=[DataRequired(), Length(1, 32)]
+    )

app/services/views.py

@@ -2,7 +2,7 @@ from datetime import datetime
 from flask import current_app, flash, redirect, render_template, url_for
 from . import services
 from flask_login import current_user, login_required
-from .forms import OCRJobForm
+from .forms import CreateOCRJobForm
 from ..import swarm
 from threading import Thread
 import hashlib
@@ -12,8 +12,8 @@ import os
 @services.route('/ocr', methods=['GET', 'POST'])
 @login_required
 def ocr():
-    ocr_job_form = OCRJobForm()
+    create_ocr_job_form = CreateOCRJobForm()
-    if ocr_job_form.validate_on_submit():
+    if create_ocr_job_form.validate_on_submit():
         app = current_app._get_current_object()
         id = hashlib.md5(
             (current_user.username + '_' + datetime.now().isoformat()).encode()
@@ -27,7 +27,7 @@ def ocr():
                'requested_cpus': 2,
                'requested_memory': 2048,
                'service': 'ocr',
-               'service_args': {'lang': ocr_job_form.language.data,
+               'service_args': {'lang': create_ocr_job_form.language.data,
                                 'version': 'latest'
                                 },
                'status': 'queued'
@@ -39,7 +39,7 @@ def ocr():
         except OSError:
             flash('OSError!')
         else:
-            file = ocr_job_form.file.data
+            for file in create_ocr_job_form.files.data:
                 file.save(os.path.join(dir, file.filename))
             '''
             ' TODO: Let the scheduler run this job in the background.
@@ -55,5 +55,5 @@ def ocr():
     return render_template(
         'services/ocr.html.j2',
         title='Optical Character Recognition',
-        ocr_job_form=ocr_job_form
+        create_ocr_job_form=create_ocr_job_form
     )

app/templates/services/ocr.html.j2

@@ -79,44 +79,54 @@
   <div class="card">
     <div class="card-content">
       <form method="POST" enctype="multipart/form-data">
-        {{ ocr_job_form.hidden_tag() }}
+        {{ create_ocr_job_form.hidden_tag() }}
         <div class="row">
-          <div class="col s8">
+          <div class="col s12 m4">
+            <div class="input-field">
+              <i class="material-icons prefix">title</i>
+              {{ create_ocr_job_form.title() }}
+              {{ create_ocr_job_form.title.label }}
+              {% for error in create_ocr_job_form.title.errors %}
+                <span class="helper-text red-text">{{ error }}</span>
+              {% endfor %}
+            </div>
+          </div>
+          <div class="col s12 m8">
             <div class="input-field">
               <i class="material-icons prefix">description</i>
-              {{ ocr_job_form.description(placeholder='Job description') }}
+              {{ create_ocr_job_form.description() }}
-              {{ ocr_job_form.description.label }}
+              {{ create_ocr_job_form.description.label }}
-              {% for error in ocr_job_form.description.errors %}
+              {% for error in create_ocr_job_form.description.errors %}
                 <span class="helper-text red-text">{{ error }}</span>
               {% endfor %}
             </div>
           </div>
-          <div class="col s4">
+          <div class="col s12 m8">
-            <div class="input-field">
-              <i class="material-icons prefix">language</i>
-              {{ ocr_job_form.language() }}
-              {{ ocr_job_form.language.label }}
-              {% for error in ocr_job_form.language.errors %}
-                <span class="helper-text red-text">{{ error }}</span>
-              {% endfor %}
-            </div>
-          </div>
-          <div class="col s8">
             <div class="file-field input-field">
               <div class="btn">
-                <span>{{ ocr_job_form.file.label.text }}</span>
+                <span>{{ create_ocr_job_form.files.label.text }}</span>
-                {{ ocr_job_form.file() }}
+                {{ create_ocr_job_form.files(accept='application/pdf, image/tiff') }}
               </div>
               <div class="file-path-wrapper">
                 <input class="file-path validate" type="text">
               </div>
-              {% for error in ocr_job_form.file.errors %}
+              {% for error in create_ocr_job_form.files.errors %}
+                <span class="helper-text red-text">{{ error }}</span>
+              {% endfor %}
+            </div>
+          </div>
+          <div class="col s12 m4">
+            <div class="input-field">
+              <i class="material-icons prefix">language</i>
+              {{ create_ocr_job_form.language() }}
+              {{ create_ocr_job_form.language.label }}
+              {% for error in create_ocr_job_form.language.errors %}
                 <span class="helper-text red-text">{{ error }}</span>
               {% endfor %}
             </div>
           </div>
         </div>
-        {{ ocr_job_form.submit(class='btn') }}
+        {{ create_ocr_job_form.submit(class='btn') }}
       </form>
     </div>
   </div>