Update app.py

app.py

@@ -4,6 +4,8 @@ import re
 import fitz  # PyMuPDF
 import docx
 from io import BytesIO
+from reportlab.pdfgen import canvas
+from reportlab.lib.pagesizes import letter
 
 # Diccionario de idiomas y sus códigos
 lang_dict = {
@@ -82,12 +84,20 @@ def read_docx(file):
     text = "\n".join([para.text for para in doc.paragraphs])
     return text
 
-def save_pdf(text):
+def save_pdf(original_file, translated_text):
     buffer = BytesIO()
-    doc = fitz.open()
-    page = doc.new_page()
-    page.insert_text((72, 72), text)
-    doc.save(buffer)
+    original_doc = fitz.open(stream=original_file.read(), filetype="pdf")
+    
+    # Crear un nuevo documento PDF
+    translated_doc = fitz.open()
+    for page_num in range(len(original_doc)):
+        original_page = original_doc.load_page(page_num)
+        translated_page = translated_doc.new_page(width=original_page.rect.width, height=original_page.rect.height)
+        
+        # Insertar el texto traducido
+        translated_page.insert_text((72, 72), translated_text, fontsize=12, fontname="helv")
+    
+    translated_doc.save(buffer)
     buffer.seek(0)
     return buffer
 
@@ -127,7 +137,8 @@ def main():
 
         if uploaded_file:
             if uploaded_file.type == "application/pdf":
-                translated_file = save_pdf(translated_text)
+                uploaded_file.seek(0)  # Reiniciar el puntero del archivo
+                translated_file = save_pdf(uploaded_file, translated_text)
                 st.download_button(
                     label="Descargar PDF traducido",
                     data=translated_file,
@@ -144,4 +155,4 @@ def main():
                 )
 
 if __name__ == "__main__":
-    main()
+    main()