oktakurniawan/removebg repository overview

  POST /remove-backgorund

from flask import Flask, request, send_file
import rembg
from PIL import Image
from io import BytesIO

app = Flask(__name__)

@app.route('/remove-background', methods=['POST'])
def remove_background():
    try:
        # Dapatkan file dari permintaan POST
        uploaded_file = request.files['file']

        # Baca file dan hapus latar belakang menggunakan rembg
        input_image = Image.open(uploaded_file)
        output_image = rembg.remove(input_image)

        # Simpan hasil penghapusan latar belakang di dalam BytesIO
        output_buffer = BytesIO()
        output_image.save(output_buffer, format="PNG")
        output_buffer.seek(0)

        return send_file(output_buffer, mimetype='image/png', as_attachment=True, download_name='output.png')
    except Exception as e:
        return str(e)

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000)

# Gunakan image Python yang telah dilengkapi dengan debian Linux sebagai base image
FROM python:3.8

# Set working directory di dalam container
WORKDIR /app

# Copy requirements.txt ke dalam container dan install dependensi
COPY requirements.txt .

RUN pip install --no-cache-dir -r requirements.txt

# Copy seluruh konten aplikasi ke dalam container
COPY . .

# Ekspose port 5000 yang digunakan oleh Flask
EXPOSE 5000

# Perintah untuk menjalankan aplikasi Flask
CMD ["python", "app.py"]

Flask
rembg
Pillow

version: '3'

services:
  remove-background-service:
    build:
      context: .
      dockerfile: Dockerfile
    ports:
      - "5000:5000"
    container_name: removebg
    hostname: removebg

<?php
defined('BASEPATH') OR exit('No direct script access allowed');

use GuzzleHttp\Client;

class RemoveBackgroundService {

    protected $client;

    public function __construct()
    {
        $this->client = new Client([
            'base_uri' => 'http://remove-background-service:5000/', // Ganti URL sesuai dengan konfigurasi Docker Anda
        ]);
    }

    public function removeBackground($file)
    {
        try {
            $response = $this->client->request('POST', 'remove-background', [
                'multipart' => [
                    [
                        'name'     => 'file',
                        'contents' => fopen($file, 'r'),
                    ],
                ],
            ]);

            return $response->getBody();
        } catch (\Exception $e) {
            return $e->getMessage();
        }
    }
}

<?php
defined('BASEPATH') OR exit('No direct script access allowed');

class RemoveBackgroundController extends CI_Controller {

    public function index() {
        $this->load->view('remove_background_form');
    }

    public function process_upload() {
        // Mendapatkan file yang diunggah
        $uploaded_file = $_FILES['userfile']['tmp_name'];

        // Load library RemoveBackgroundService
        $this->load->library('removebackgroundservice');

        // Mengirimkan file ke layanan penghapusan latar belakang
        $result = $this->removebackgroundservice->removeBackground($uploaded_file);

        // Menyimpan hasil penghapusan latar belakang (tersimpan di folder tempat file diunggah)
        $output_file = 'uploads/output.png';
        file_put_contents($output_file, $result);

        // Tampilkan hasil penghapusan latar belakang
        $data['output_file'] = $output_file;
        $this->load->view('remove_background_result', $data);
    }
}

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>Upload and Remove Background</title>
</head>
<body>
    <h2>Upload and Remove Background</h2>
    <?php echo form_open_multipart('removebackgroundcontroller/process_upload'); ?>
    <input type="file" name="userfile" size="20" />
    <br /><br />
    <input type="submit" value="Upload" />
    </form>
</body>
</html>


<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>Remove Background Result</title>
</head>
<body>
    <h2>Remove Background Result</h2>
    <img src="<?php echo base_url($output_file); ?>" alt="Removed Background" />
</body>
</html>