Update app.py

app.py

@@ -1,11 +1,6 @@
 import streamlit as st
 from mtranslate import translate
 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 = {
@@ -72,87 +67,24 @@ def translate_text(lang, text):
     
     return translated_text
 
-def read_pdf(file):
-    doc = fitz.open(stream=file.read(), filetype="pdf")
-    text = ""
-    for page in doc:
-        text += page.get_text()
-    return text
-
-def read_docx(file):
-    doc = docx.Document(file)
-    text = "\n".join([para.text for para in doc.paragraphs])
-    return text
-
-def save_pdf(original_file, translated_text):
-    buffer = BytesIO()
-    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
-
-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")
     
     text = st.text_area("Texto a traducir", height=200)
     lang = st.selectbox("Idioma", lang_list)
-
-    uploaded_file = st.file_uploader("Sube un documento (PDF o DOCX)", type=["pdf", "docx"])
-    
-    if uploaded_file:
-        if uploaded_file.type == "application/pdf":
-            text = read_pdf(uploaded_file)
-        elif uploaded_file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
-            text = read_docx(uploaded_file)
-        st.text_area("Contenido del documento", text, height=200)
     
     if st.button("Traducir"):
         # Realizar la traducción
         translated_text = translate_text(lang, text)
         
-        # Mostrar el resultado
-        st.write("Texto traducido:")
-        st.write(translated_text)
-
-        # 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":
-                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,
-                    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"
-                )
+        # Mostrar el resultado con el botón de copia
+        html_output = f"""
+        <div>
+            <span id='translated_text'>{translated_text}</span>
+            <button class="lg secondary svelte-cmf5ev" style="font-size: small; padding: 2px; color: #808080ba; border: none; margin-left: 5px;" onclick='navigator.clipboard.writeText(document.getElementById("translated_text").innerText).then(() => alert("Texto copiado al portapapeles"))'>&nbsp;✂&nbsp;</button>
+        </div>
+        """
+        st.markdown(html_output, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()