Adds app and dockerfile

Dockerfile

@@ -0,0 +1,16 @@
+FROM tiangolo/uvicorn-gunicorn-fastapi:python3.9
+
+RUN mkdir src
+RUN mkdir app
+
+WORKDIR /src/app
+
+COPY ./requirements.txt ./
+
+RUN pip3 install -r requirements.txt
+
+COPY . .
+
+EXPOSE 7860
+
+CMD [ "python3", "app.py" ]

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Sharan Thakur
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

app.py

@@ -0,0 +1,108 @@
+import base64
+
+from rembg import remove, new_session
+
+from typing import Annotated
+from fastapi import FastAPI, UploadFile, Request, File, Form
+from fastapi.templating import Jinja2Templates
+from fastapi.responses import RedirectResponse, HTMLResponse, Response
+
+import uvicorn
+
+import os
+
+os.environ['U2NET_HOME'] = './models/'
+
+model_names = [
+    'u2net',
+    'u2net_human_seg',
+    'u2net_cloth_seg',
+    'isnet-general-use',
+]
+
+sessions = {
+    'u2net': new_session(model_name=model_names[0]),
+}
+
+app = FastAPI()
+templates = Jinja2Templates(directory="templates")
+
+
+@app.get("/")
+def health(request: Request):
+    return templates.TemplateResponse('dynamic.html', { "request": request })
+
+
+@app.post("/remove")
+async def remove_bg(
+    request: Request,
+    file: Annotated[UploadFile, File()],
+    mask_only: Annotated[str, Form()] = 'off',
+    name_of_model: Annotated[str, Form()] = 'u2net'
+) -> HTMLResponse:
+    try:
+        if name_of_model not in sessions.keys():
+            sessions[name_of_model] = new_session(model_name=name_of_model)
+        
+        current_session = sessions[name_of_model]
+
+        only_mask = mask_only == 'on'
+        data = file.file.read()
+
+        output_array = remove(data, only_mask=only_mask, session=current_session)
+
+        output_img = base64.b64encode(output_array).decode('utf-8')
+
+        file.file.close()
+
+        encoded_image = base64.b64encode(data).decode('utf-8')
+        
+        return templates.TemplateResponse(
+            'dynamic.html',
+            {
+                "request" : request,
+                "image" : encoded_image,
+                "output_img" : output_img
+            }
+        )
+    except Exception as error_msg:
+        return templates.TemplateResponse(
+            "error.html",
+            {
+                "request" : request,
+                "error_msg" : str(error_msg),
+            }
+        )
+
+
+@app.post("/remove_bg")
+async def remove_bg(
+    request: Request,
+    file: UploadFile,
+    mask_only: str = 'off',
+    name_of_model: str = 'u2net'
+):
+    try:
+        if name_of_model not in sessions.keys():
+            sessions[name_of_model] = new_session(model_name=name_of_model)
+        
+        current_session = sessions[name_of_model]
+
+        only_mask = mask_only == 'on'
+        data = file.file.read()
+
+        output_array = remove(data, only_mask=only_mask, session=current_session)
+
+        file.file.close()
+        
+        return Response(content=output_array, media_type="image/png")
+    except Exception as error:
+        return f"Oopss!!!! {error}"
+
+
+@app.get("/remove")
+def remove_bg_redirect():
+    return RedirectResponse('/')
+
+if __name__ == '__main__':
+    uvicorn.run(app, host='0.0.0.0', port=7860)

models/isnet-general-use.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:60920e99c45464f2ba57bee2ad08c919a52bbf852739e96947fbb4358c0d964a
+size 178648008

models/u2net.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8d10d2f3bb75ae3b6d527c77944fc5e7dcd94b29809d47a739a7a728a912b491
+size 175997641

models/u2net_cloth_seg.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d2cbc27bfbdc989e1fd325656d65902ecc6a3ccbe94b2d3655ec114efcb128e
+size 176194565

models/u2net_human_seg.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01eb6a29a5c4d8edb30b56adad9bb3a2a0535338e480724a213e0acfd2d1c73c
+size 175997641

requirements.txt

@@ -0,0 +1,4 @@
+rembg==2.0.56
+python-multipart==0.0.9
+fastapi
+uvicorn

templates/dynamic.html

@@ -0,0 +1,116 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <title>Remove Image Background</title>
+  </head>
+  <style>
+    hr {
+      border: none;
+      border-top: 3px double #333;
+      color: #333;
+      overflow: visible;
+      text-align: center;
+      height: 5px;
+    }
+
+    hr::after {
+      background: #fff;
+      content: "📷";
+      padding: 0 4px;
+      position: relative;
+      top: -13px;
+    }
+
+    form {
+      padding: 10px;
+      border: 1px solid #ddd;
+      margin-bottom: 20px;
+    }
+
+    input[type="submit"] {
+      background-color: #333;
+      color: #fff;
+      padding: 5px 10px;
+      border: none;
+      border-radius: 3px;
+      cursor: pointer;
+    }
+
+    .drop-zone {
+      border: 2px dashed #ccc;
+      padding: 20px;
+      display: flex;
+      flex-direction: column;
+      align-items: center;
+      justify-content: center;
+      cursor: pointer;
+    }
+
+    .drop-zone.dragover {
+      background-color: #f0f0f0;
+    }
+  </style>
+
+  <body>
+    <div id="dropZone" class="drop-zone">
+      <h1>Remove Background from Images!</h1>
+      <form
+        id="upload_form"
+        action="/remove"
+        enctype="multipart/form-data"
+        method="POST"
+      >
+        <input id="input_image" name="file" type="file" accept="image/*" />
+
+        <label for="mask_only">Mask Only</label>
+        <input
+          type="checkbox"
+          id="mask_only"
+          name="mask_only"
+          type="checkbox"
+        />
+
+        <label for="name_of_model">Choose a model:</label>
+
+        <select name="name_of_model" id="name_of_model">
+          <option value="u2net">U2Net</option>
+          <option value="u2net_human_seg">U2Net Human Seg</option>
+          <option value="u2net_cloth_seg">U2Net Cloth Seg</option>
+          <option value="isnet-general-use">ISNET</option>
+        </select>
+
+        <input type="submit" />
+      </form>
+
+      <hr />
+
+      {% if image %}
+      <h1>Input Image</h1>
+      <img src="data:image/jpeg;base64,{{ image }}" height="200" />
+      {% else %} No Image Selected {% endif %} {% if output_img %}
+      <h1>Output Image</h1>
+      <img src="data:image/jpeg;base64,{{ output_img }}" height="200" />
+      {% endif %}
+
+      <h4>The available models are:</h4>
+      <ul>
+        <li><b>u2net:</b> A pre-trained model for general use cases.</li>
+        <li>
+          <b>u2net_human_seg:</b> A pre-trained model for human segmentation.
+        </li>
+        <li>
+          <b>u2net_cloth_seg:</b> A pre-trained model for Cloths Parsing from
+          human portrait. Here clothes are parsed into 3 category: Upper body,
+          Lower body and Full body. isnet-general-use: A new pre-trained model
+          for general use cases.
+        </li>
+        <li>
+          <a
+            href="https://github.com/danielgatis/rembg?tab=readme-ov-file#models"
+            >Learn More</a
+          >
+        </li>
+      </ul>
+    </div>
+  </body>
+</html>

templates/error.html

@@ -0,0 +1,13 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <title>Remove Image Background</title>
+  </head>
+  <body>
+    <h1>Error!!!!</h1>
+    {% if error_msg %}
+    <h3>{{error_msg}}</h3>
+    {% endif %}
+    <a href="/">Home</a>
+  </body>
+</html>