feat(core): Allow for txt reports to be uploaded

.gitignore

@@ -12,6 +12,7 @@ __pycache__
 !.gitkeep
 db.sqlite3
 *.pem
+.vscode/
 
 # pyenv
 .python-version

src/tram/tram/ml/base.py

@@ -128,6 +128,8 @@ def _extract_text(self, document):
             text = self._extract_docx_text(document)
         elif suffix == ".html":
             text = self._extract_html_text(document)
+        elif suffix == '.txt':
+            text = self._extract_plain_text(document)
         else:
             raise ValueError("Unknown file suffix: %s" % suffix)
 
@@ -228,6 +230,10 @@ def _extract_docx_text(self, document):
         text = " ".join([paragraph.text for paragraph in parsed_docx.paragraphs])
         return text
 
+    def _extract_plain_text(self, document):
+        text = document.docfile.read().decode('UTF-8')
+        return text
+
     def process_job(self, job):
         name = self._get_report_name(job)
         text = self._extract_text(job.document)

src/tram/tram/templates/base.html

@@ -45,7 +45,7 @@ <h1 style="color: white">
                     <label for="id-upload" class="btn btn btn-outline-light" title="Upload File">
                         <i class="fas fa-upload"></i>
                     </label>
-                    <input type="file" id="id-upload" style="display: none; height: max-content;" accept=".pdf,.docx,.html,.json">
+                    <input type="file" id="id-upload" style="display: none; height: max-content;" accept=".pdf,.docx,.html,.json,.txt">
                     </form>
                 </li>
                 <li class="navbar-item me-1">

src/tram/tram/views.py

@@ -98,6 +98,7 @@ def upload(request):
         "application/pdf",  # .pdf files
         "text/html",  # .html files
         "application/vnd.openxmlformats-officedocument.wordprocessingml.document",  # .docx files
+        "text/plain",  # .txt files
     ):
         DocumentProcessingJob.create_from_file(request.FILES["file"])
     elif file_content_type in ("application/json",):  # .json files