Se implemento la descarga del archvio de la transcripcion

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 from streamlit_tools.tools import load_audio_file
 from openai_models.whisper import whisper_os
 import time
+from documents.docs import generate_docx
+import os
 
 
 def main():
@@ -16,6 +18,16 @@ def main():
                     yield word + ' '
             st.write_stream(transcription_generator(), )
             # st.write(transcription)
+            #
+            generate_docx(transcription["text"])
+
+        if os.path.exists(st.session_state.full_path_docx):
+            with open(st.session_state.full_path_docx, "rb") as file:
+                btn = st.download_button(
+                    label="Download docx",
+                    data=file,
+                    file_name=st.session_state.full_path_docx,
+                )
 
 
 if __name__ == "__main__":

documents/docs.py

@@ -0,0 +1,26 @@
+import os
+import docx
+import streamlit as st
+
+
+def generate_docx(text: str):
+    file_name: str = st.session_state.audio_file_name.replace(".mp3", ".docx")
+
+    file_path: str = os.path.join("documents", "docs")
+
+    # Ensure the directory exists
+    os.makedirs(file_path, exist_ok=True)
+
+    # full_path: str = os.path.join(file_path, file_name)
+
+    if "full_path_docx" not in st.session_state.keys():
+        st.session_state.full_path_docx = os.path.join(file_path, file_name)
+
+    # Crea un nuevo documento
+    doc = docx.Document()
+
+    # Agrega un párrafo al documento
+    doc.add_paragraph(f"Texto del audio transcrito:\n\n{text}")
+
+    # Guarda el documento en un archivo .docx
+    doc.save(st.session_state.full_path_docx)