Update app.py

app.py

@@ -3,7 +3,9 @@ from mtranslate import translate
 import re
 import fitz  # PyMuPDF
 import docx
+from io import BytesIO
 
+# Diccionario de idiomas y sus códigos
 lang_dict = {
     'Español': 'es',
     'English': 'en',
@@ -80,6 +82,23 @@ def read_docx(file):
     text = "\n".join([para.text for para in doc.paragraphs])
     return text
 
+def save_pdf(text):
+    buffer = BytesIO()
+    doc = fitz.open()
+    page = doc.new_page()
+    page.insert_text((72, 72), text)
+    doc.save(buffer)
+    buffer.seek(0)
+    return buffer
+
+def save_docx(text):
+    buffer = BytesIO()
+    doc = docx.Document()
+    doc.add_paragraph(text)
+    doc.save(buffer)
+    buffer.seek(0)
+    return buffer
+
 def main():
     st.title("Traducción sin límites")
     
@@ -106,5 +125,23 @@ def main():
         # Botón para copiar el texto traducido
         st.markdown(f'<button onclick="navigator.clipboard.writeText(`{translated_text}`)">Copiar al portapapeles</button>', unsafe_allow_html=True)
 
+        if uploaded_file:
+            if uploaded_file.type == "application/pdf":
+                translated_file = save_pdf(translated_text)
+                st.download_button(
+                    label="Descargar PDF traducido",
+                    data=translated_file,
+                    file_name="documento_traducido.pdf",
+                    mime="application/pdf"
+                )
+            elif uploaded_file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
+                translated_file = save_docx(translated_text)
+                st.download_button(
+                    label="Descargar DOCX traducido",
+                    data=translated_file,
+                    file_name="documento_traducido.docx",
+                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+                )
+
 if __name__ == "__main__":
-    main()
+    main()