Update pdf_compressor_app.py

pdf_compressor_app.py

@@ -0,0 +1,86 @@
+import os
+import io
+import subprocess
+from flask import Flask, render_template, request, send_file, Response
+
+app = Flask(__name__)
+
+# --- CONFIGURACIÓN DE GHOSTSCRIPT ---
+# Definición del tamaño objetivo en Bytes (18 MB, solo como referencia de la agresividad)
+TARGET_SIZE_BYTES = 18 * 1024 * 1024 
+
+def compress_pdf_with_ghostscript(input_stream: io.BytesIO, quality: str) -> io.BytesIO:
+    """
+    Comprime un PDF usando Ghostscript. 
+    Aplica una compresión agresiva de imágenes y metadatos del PDF.
+    """
+    output_stream = io.BytesIO()
+    input_stream.seek(0)
+    
+    # Argumentos de Ghostscript para compresión
+    gs_command = [
+        "gs",
+        "-sDEVICE=pdfwrite",
+        "-dCompatibilityLevel=1.4",
+        "-dPDFSETTINGS=/" + quality,  # Usará 'screen', 'ebook', etc.
+        "-dNOPAUSE", "-dBATCH", "-dQUIET",
+        "-sOutputFile=-",  # Salida a stdout
+        "-"  # Entrada desde stdin
+    ]
+    
+    try:
+        # Ejecutar el comando Ghostscript
+        process = subprocess.run(
+            gs_command,
+            input=input_stream.read(),
+            capture_output=True,
+            check=True
+        )
+        output_stream.write(process.stdout)
+        output_stream.seek(0)
+        return output_stream
+    except subprocess.CalledProcessError as e:
+        raise Exception(f"Error de Ghostscript: {e.stderr.decode()}")
+    except FileNotFoundError:
+        raise Exception("Ghostscript no está instalado en el servidor. (Verifica el Dockerfile)")
+    except Exception as e:
+        raise Exception(f"Error desconocido durante la compresión: {e}")
+
+# --- RUTAS ---
+@app.route('/')
+def index():
+    """Muestra la interfaz del compresor."""
+    # Asegúrate de crear la carpeta templates/ y poner el HTML dentro
+    return render_template('compressor_index.html') 
+
+@app.route('/compress', methods=['POST'])
+def compress_pdf_file():
+    """Ruta para recibir y comprimir el PDF subido."""
+    if 'pdf_file' not in request.files:
+        return Response('No se seleccionó ningún archivo PDF', status=400)
+
+    pdf_file = request.files['pdf_file']
+    quality_setting = request.form.get('quality', 'screen')
+    
+    try:
+        input_stream = io.BytesIO(pdf_file.read())
+        original_size = len(input_stream.getvalue())
+
+        # 1. Comprimir el PDF subido
+        compressed_stream = compress_pdf_with_ghostscript(input_stream, quality_setting)
+        
+        # 2. Devolver el archivo comprimido
+        return send_file(
+            compressed_stream,
+            as_attachment=True,
+            download_name=f"comprimido_{pdf_file.filename}",
+            mimetype='application/pdf'
+        )
+        
+    except Exception as e:
+        return Response(f"Error en la compresión: {str(e)}", status=500)
+
+
+if __name__ == '__main__':
+    port = int(os.environ.get('PORT', 7860)) 
+    app.run(host='0.0.0.0', port=port, debug=False)