Add response for successful file upload (#195)

deepset-ai · Jul 6, 2020 · 68d604d · 68d604d
1 parent 0eb8a29
commit 68d604d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -13,4 +13,5 @@ coverage
 langdetect # for PDF conversions
 # optional: sentence-transformers
 #temporarily (used for DPR downloads)
-wget
+wget
+python-multipart
diff --git a/rest_api/controller/file_upload.py b/rest_api/controller/file_upload.py
@@ -49,7 +49,7 @@ def upload_file_to_document_store(
     remove_empty_lines: Optional[bool] = Form(REMOVE_EMPTY_LINES),
     remove_header_footer: Optional[bool] = Form(REMOVE_HEADER_FOOTER),
     valid_languages: Optional[List[str]] = Form(VALID_LANGUAGES),
-) -> None:
+):
     try:
         file_path = Path(FILE_UPLOAD_PATH) / f"{uuid.uuid4().hex}_{file.filename}"
         with file_path.open("wb") as buffer:
@@ -78,6 +78,6 @@ def upload_file_to_document_store(
 
         document = {TEXT_FIELD_NAME: "\n".join(pages), "name": file.filename}
         document_store.write_documents([document])
-
+        return "File upload was successful."
     finally:
         file.file.close()