mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-12-26 11:24:18 +00:00
101 lines
3.5 KiB
Python
101 lines
3.5 KiB
Python
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 NewOCRJobForm, NewNLPJobForm
|
|
from ..models import Job
|
|
from .. import db
|
|
import json
|
|
import os
|
|
|
|
|
|
@services.route('/nlp', methods=['GET', 'POST'])
|
|
@login_required
|
|
def nlp():
|
|
new_nlp_job_form = NewNLPJobForm()
|
|
if new_nlp_job_form.validate_on_submit():
|
|
nlp_job = Job(creator=current_user,
|
|
description=new_nlp_job_form.description.data,
|
|
mem_mb=4096,
|
|
n_cores=2,
|
|
service='nlp',
|
|
service_args=json.dumps(
|
|
['-l {}'.format(new_nlp_job_form.language.data)]
|
|
),
|
|
service_version=new_nlp_job_form.version.data,
|
|
status='preparing',
|
|
title=new_nlp_job_form.title.data)
|
|
|
|
db.session.add(nlp_job)
|
|
db.session.commit()
|
|
|
|
dir = os.path.join(current_app.config['OPAQUE_STORAGE_DIRECTORY'],
|
|
str(nlp_job.user_id),
|
|
'jobs',
|
|
str(nlp_job.id))
|
|
|
|
try:
|
|
os.makedirs(dir)
|
|
except OSError:
|
|
flash('OSError!')
|
|
db.session.remove(nlp_job)
|
|
db.session.commit()
|
|
else:
|
|
for file in new_nlp_job_form.files.data:
|
|
file.save(os.path.join(dir, file.filename))
|
|
nlp_job.status = 'submitted'
|
|
db.session.commit()
|
|
flash('Job created!')
|
|
return redirect(url_for('services.nlp'))
|
|
|
|
return render_template('services/nlp.html.j2',
|
|
title='Natrual Language Processing',
|
|
new_nlp_job_form=new_nlp_job_form)
|
|
|
|
|
|
@services.route('/ocr', methods=['GET', 'POST'])
|
|
@login_required
|
|
def ocr():
|
|
new_ocr_job_form = NewOCRJobForm()
|
|
if new_ocr_job_form.validate_on_submit():
|
|
ocr_job = Job(creator=current_user,
|
|
description=new_ocr_job_form.description.data,
|
|
mem_mb=8192,
|
|
n_cores=4,
|
|
service='ocr',
|
|
service_args=json.dumps(
|
|
[
|
|
'-l {}'.format(new_ocr_job_form.language.data),
|
|
'' if new_ocr_job_form.binarization.data
|
|
else '--skip-binarisation'
|
|
]
|
|
),
|
|
service_version=new_ocr_job_form.version.data,
|
|
status='preparing',
|
|
title=new_ocr_job_form.title.data)
|
|
|
|
db.session.add(ocr_job)
|
|
db.session.commit()
|
|
|
|
dir = os.path.join(current_app.config['OPAQUE_STORAGE_DIRECTORY'],
|
|
str(ocr_job.user_id),
|
|
'jobs',
|
|
str(ocr_job.id))
|
|
|
|
try:
|
|
os.makedirs(dir)
|
|
except OSError:
|
|
flash('OSError!')
|
|
db.session.remove(ocr_job)
|
|
db.session.commit()
|
|
else:
|
|
for file in new_ocr_job_form.files.data:
|
|
file.save(os.path.join(dir, file.filename))
|
|
ocr_job.status = 'submitted'
|
|
db.session.commit()
|
|
flash('Job created!')
|
|
return redirect(url_for('services.ocr'))
|
|
|
|
return render_template('services/ocr.html.j2',
|
|
title='Optical Character Recognition',
|
|
new_ocr_job_form=new_ocr_job_form)
|