header with filename added to docx

src/runners/transcription_zipper.py

@@ -18,7 +18,7 @@ class TranscriptionZipper(bentoml.Runnable):
 
         for t_list in transcription_list:
             orig_filename = t_list[0]
-            new_content = create_word_content(t_list[1], t_list[2])
+            new_content = create_word_content(orig_filename, t_list[1], t_list[2])
             new_content.save(orig_filename + '.docx')
             # new_content = create_content(t_list[1], t_list[2])  # html string
             if ".mp3" in orig_filename:
@@ -26,18 +26,19 @@ class TranscriptionZipper(bentoml.Runnable):
             else:
                 file_name = orig_filename.removesuffix(".wav")
             with zipfile.ZipFile(zip_buffer, "a") as zip_file:
-                #zip_file.writestr(file_name + ".html", new_content)
+                # zip_file.writestr(file_name + ".html", new_content)
                 zip_file.write(orig_filename + '.docx')
 
         # Return the zip file as bytes
         return zip_buffer.getvalue()
 
-def create_word_content(old_content, new_content):
+
+def create_word_content(filename, old_content, new_content):
     from docx import Document
 
     document = Document()
 
-    # document.add_heading('Document Title', 0)
+    document.add_heading(filename, 0)
 
     p = document.add_paragraph()
 
@@ -51,7 +52,7 @@ def create_word_content(old_content, new_content):
         try:
             if changed_content[i] == old_content[i]:
                 p.add_run(" " + changed_content[i])
-                #new_content += " " + changed_content[i]
+                # new_content += " " + changed_content[i]
                 both.pop(0)
                 old_content.pop(0)
                 changed_content.pop(0)