1 files changed, 10 insertions, 3 deletions
diff --git a/megapixels/app/server/api_task.py b/megapixels/app/server/api_task.py
index c9bc19ed..57ae9f7d 100644
--- a/megapixels/app/server/api_task.py
+++ b/megapixels/app/server/api_task.py
@@ -27,7 +27,8 @@ def task_status(task_name, task_id):
   if task_name in task_lookup:
     task = task_lookup[task_name]['task'].AsyncResult(task_id)
     # task = AsyncResult(task_id, app=celery)
-  else:
+
+  if task_name not in task_lookup or task.info is None:
     return jsonify({
       'state': 'error',
       'percent': 100,
@@ -75,10 +76,16 @@ def sleep_test():
 
 @api_task.route('/upload/blur', methods=['POST'])
 def upload():
+  return process('blur')
+
+@api_task.route('/upload/demo', methods=['POST'])
+def demo():
+  return process('demo')
+
+def process(style):
   """
-  Process a images in a particular style
+  Process an image in a particular style
   """
-  style = 'blur'
   print('style: {}'.format(style))
   if style in task_lookup:
     task = task_lookup[style]['task']