a

app.py

@@ -46,51 +46,48 @@ def flujo_principal(pdf_file, idioma="spa"):
     if not pdf_file:
         raise gr.Error("No se subió ningún archivo.")
 
-    # Asegúrate de que los nombres de archivo sean seguros
-    input_pdf = pdf_file.name
-    input_pdf_fixed = input_pdf.replace(" ", "_")  # Renombra temporalmente para evitar problemas con espacios
-    os.rename(input_pdf, input_pdf_fixed)
+    # Guardar el archivo subido en el directorio /tmp
+    temp_dir = tempfile.mkdtemp()
+    input_pdf_path = os.path.join(temp_dir, pdf_file.name)
+    with open(input_pdf_path, "wb") as f:
+        f.write(pdf_file.read())
 
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_output:
-        output_pdf = temp_output.name
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf", dir=temp_dir) as temp_output:
+        output_pdf_path = temp_output.name
 
-    texto_original = leer_pdf(input_pdf_fixed)
+    texto_original = leer_pdf(input_pdf_path)
 
     try:
-        crear_pdf_con_texto_incrustado(input_pdf_fixed, output_pdf, idioma)
-        texto_ocr = leer_pdf(output_pdf)
-        return gr.File(input_pdf_fixed, label="PDF Original"), texto_original, gr.File(output_pdf, label="PDF con OCR"), texto_ocr
+        crear_pdf_con_texto_incrustado(input_pdf_path, output_pdf_path, idioma)
+        texto_ocr = leer_pdf(output_pdf_path)
+        return gr.File(input_pdf_path, label="PDF Original"), texto_original, gr.File(output_pdf_path, label="PDF con OCR"), texto_ocr
     except gr.Error as e:
-        if os.path.exists(output_pdf):
-            os.remove(output_pdf)
         raise e
     finally:
-        if os.path.exists(input_pdf_fixed):
-            os.remove(input_pdf_fixed)
-        if os.path.exists(output_pdf):
-            os.remove(output_pdf)
+        # Limpiar archivos temporales
+        if os.path.exists(input_pdf_path):
+            os.remove(input_pdf_path)
+        if os.path.exists(output_pdf_path):
+            os.remove(output_pdf_path)
+        os.rmdir(temp_dir)
 
 # Interfaz Gradio
 with gr.Blocks() as interfaz:
     gr.Markdown("## Procesador OCR para PDFs")
-
     with gr.Row():
         archivo_pdf = gr.File(label="Sube tu archivo PDF")
         idioma_ocr = gr.Dropdown(["spa", "eng", "fra", "deu"], label="Idioma OCR", value="spa")
         boton_procesar = gr.Button("Procesar OCR")
-
     with gr.Row():
         texto_original = gr.Textbox(label="Texto Original", lines=10, interactive=False)
         texto_ocr = gr.Textbox(label="Texto con OCR", lines=10, interactive=False)
-
     with gr.Row():
         pdf_original_vista = gr.File(label="Descargar PDF Original", interactive=False)
         pdf_ocr_vista = gr.File(label="Descargar PDF con OCR", interactive=False)
-
     boton_procesar.click(
         fn=flujo_principal,
         inputs=[archivo_pdf, idioma_ocr],
         outputs=[pdf_original_vista, texto_original, pdf_ocr_vista, texto_ocr]
     )
 
-interfaz.launch(share=True)
+interfaz.launch()