Content disposition header

1 files changed, 2 insertions, 2 deletions

diff --git a/main.py b/main.py
index a4766b1..2309bf8 100644
--- a/main.py
+++ b/main.py
@@ -50,7 +50,7 @@ def create_app(test_config=None):
         def remove_file(response):
             os.remove(complete_path)
             return response
-        return send_from_directory(app.config['UPLOAD_FOLDER'], filepath)
+        return send_from_directory(app.config['UPLOAD_FOLDER'], filepath, as_attachment=True)
 
     @app.route('/', methods=['GET', 'POST'])
     def upload_file():
@@ -180,7 +180,7 @@ def create_app(test_config=None):
                     os.remove(complete_path)
                     return response
 
-            return send_from_directory(app.config['UPLOAD_FOLDER'], filepath)
+            return send_from_directory(app.config['UPLOAD_FOLDER'], filepath, as_attachment=True)
 
     class APIBulkDownloadCreator(Resource):
         schema = {