1 files changed, 14 insertions, 3 deletions
diff --git a/app.py b/app.py
index 430942c..51249de 100644
--- a/app.py
+++ b/app.py
@@ -62,8 +62,9 @@ class UploadForm(Form):
 
 
 
-@app.route('/<study>/upload', methods=['GET', 'POST'])
-def upload(study):
+@app.route('/<study>/upload/', methods=['GET', 'POST'])
+@app.route('/<study>/upload/<module>', methods=['GET', 'POST'])
+def upload(study, module = None):
   form = UploadForm()
   form.study.data = study
 
@@ -95,7 +96,17 @@ def upload(study):
 
       return redirect(url_for('study_index', study = study, module = module))
 
-  return render_template('upload.html', study = study, form = form)
+  module_list = None
+  if module:
+    try: form.module.data = [k for (k,v) in form.module.choices if v == module][0]
+    except: pass
+
+    print(form.module.data)
+    module_list = fit[study].get_module(module)
+
+  return render_template('upload.html', 
+            study = study, form = form, module_list = module_list, module=module
+         )