Spaces:

Truepic
/

watermarked-content-credentials

Sleeping

App Files Files Community

jclyo1 commited on Sep 28, 2023

Commit

e158b55

•

1 Parent(s): 34ee3ab

first update

Browse files

Files changed (4) hide show

Dockerfile +45 -0
main.py +88 -0
static/index.html +69 -0
static/style.css +8 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,45 @@

+FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
+# Using the Ubuntu image (our OS)
+# Update package manager (apt-get)
+# and install (with the yes flag `-y`)
+# Python and Pip
+RUN apt-get update && apt-get install -y \
+    python3-pip
+WORKDIR /code
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+# Switch to the "user" user
+USER user
+# Set home to the user's home directory
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+# Download and uzip truepic-sign from Google Drive
+RUN pip install gdown
+RUN --mount=type=secret,id=truepic_sign_gdoc_id,mode=0444,required=true \
+    gdown --id $(cat /run/secrets/truepic_sign_gdoc_id)
+RUN tar -xf truepic-lens-cli-v0.4.0-ubuntu-20.04.tar.gz
+RUN chmod +x truepic
+RUN --mount=type=secret,id=api_key,mode=0444,required=true \
+  ./truepic enroll file-system --api-key $(cat /run/secrets/api_key)
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+WORKDIR $HOME/app
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py ADDED Viewed

	@@ -0,0 +1,88 @@

+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+import subprocess
+import os
+import json
+import uuid
+import cgi
+import cgitb; cgitb.enable()
+import torch
+from diffusers import (
+    StableDiffusionPipeline,
+    DPMSolverMultistepScheduler,
+    EulerDiscreteScheduler,
+)
+app = FastAPI()
+@app.get("/generate")
+def generate_image(prompt, model):
+    torch.cuda.empty_cache()
+    modelArray = model.split(",")
+    modelName = modelArray[0]
+    modelVersion = modelArray[1]
+    pipeline = StableDiffusionPipeline.from_pretrained(
+        str(modelName), torch_dtype=torch.float16
+    )
+    pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
+    pipeline = pipeline.to("cuda")
+    image = pipeline(prompt, num_inference_steps=50, height=512, width=512).images[0]
+    filename = str(uuid.uuid4()) + ".jpg"
+    image.save(filename)
+    assertion = {
+        "assertions": [
+            {
+                "label": "com.truepic.custom.ai",
+                "data": {
+                    "model_name": modelName,
+                    "model_version": modelVersion,
+                    "prompt": prompt,
+                },
+            }
+        ]
+    }
+    json_object = json.dumps(assertion)
+    subprocess.check_output(
+        [
+            "./truepic",
+            "sign",
+            filename,
+            "--assertions-inline",
+            json_object,
+            "--output",
+            (os.getcwd() + "/static/" + filename),
+        ]
+    )
+    return {"response": filename}
+@app.get("/verify")
+def verify_image():
+    form = cgi.FieldStorage()
+    fileitem = form['fileUpload']
+    # check if the file has been uploaded
+    if fileitem.filename:
+        # strip the leading path from the file name
+        fn = os.path.basename(fileitem.filename)
+        # open read and write the file into the server
+        open(fn, 'wb').write(fileitem.file.read())
+    return {"response": fileitem.filename}
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
+@app.get("/")
+def index() -> FileResponse:
+    return FileResponse(path="/app/static/index.html", media_type="text/html")

static/index.html ADDED Viewed

	@@ -0,0 +1,69 @@

+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="utf-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1" />
+    <!-- Bootstrap CSS -->
+    <link
+      href="https://cdn.jsdelivr.net/npm/bootstrap@5.0.2/dist/css/bootstrap.min.css"
+      rel="stylesheet"
+      integrity="sha384-EVSTQN3/azprG1Anm3QDgpJLIm9Nao0Yz1ztcQTwFspd3yD65VohhpuuCOmLASjC"
+      crossorigin="anonymous"
+    />
+    <script
+    src="https://cdn.jsdelivr.net/npm/bootstrap@5.0.2/dist/js/bootstrap.bundle.min.js"
+    integrity="sha384-MrcW6ZMFYlzcLA8Nl+NtUVF0sA7MsXsP1UyJoMp4YLEuNSfAP+JcXn/tWtIaxVXM"
+    crossorigin="anonymous"
+  ></script>
+    <link rel="stylesheet" href="style.css" type="text/css" />
+</head>
+  <body>
+    <div class="container">
+        <form id="fileUploadForm">
+            <label class="form-label" for="customFile">Default file input example</label>
+            <input type="file" class="form-control" name="fileUpload" id="fileUpload" />
+        </form>
+    </div>
+  </body>
+  <script>
+const form = document.querySelector('#fileUploadForm');
+form.addEventListener('submit', (e) => {
+  e.preventDefault();
+  submitForm();
+});
+function submitForm() {
+  const form = document.querySelector('#myForm');
+  const formData = new FormData(form);
+  // Add additional form data as needed
+  //formData.append('additionalData', 'additionalValue');
+  // Call function to submit form data
+  submitFormData(formData);
+}
+function submitFormData(formData) {
+  fetch('/verify', {
+    method: 'POST',
+    body: formData
+  })
+  .then(response => response.json())
+  .then(data => {
+    console.log(data)
+    // Handle response data
+  })
+  .catch(error => {
+    console.log(error)
+    // Handle errors
+  });
+}
+  </script>
+</html>

static/style.css ADDED Viewed

	@@ -0,0 +1,8 @@

+#uploaded-image-container {
+    border-bottom: 1px solid black;
+    border-top: 1px solid black;
+}
+.results {
+    display: none;
+}