Spaces:

Truepic
/

watermarked-content-credentials

Sleeping

jclyo1 commited on Sep 29, 2023

Commit

fd1dd45

•

1 Parent(s): be4dbd7

updates

Files changed (2) hide show

main.py CHANGED Viewed

@@ -1,7 +1,6 @@
-from fastapi import FastAPI, Request, UploadFile, HTTPException, status
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
-import aiofiles
 import subprocess
 import os
@@ -68,41 +67,49 @@ def generate_image(prompt, model):
     return {"response": filename}
 @app.post("/verify")
-def verify_image(file: UploadFile):
     logging.warning("in verify")
-    logging.warning(file.filename)
-    logging.warning('form')
-    logging.warning(form)
-    logging.warning('fileitem')
-    logging.warning(fileitem)
     # 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.post('/upload')
-async def upload(fileUpload: UploadFile):
-    try:
-        contents = await fileUpload.read()
-        async with aiofiles.open(fileUpload.filename, 'wb') as f:
-            await f.write(contents)
-    except Exception:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail='There was an error uploading the file',
-        )
-    finally:
-        await fileUpload.close()
-    return {'message': f'Successfuly uploaded {fileUpload.filename}'}
 app.mount("/", StaticFiles(directory="static", html=True), name="static")

+from fastapi import FastAPI, UploadFile
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
 import subprocess
 import os
     return {"response": filename}
 @app.post("/verify")
+def verify_image(fileUpload: UploadFile):
     logging.warning("in verify")
+    logging.warning(fileUpload.filename)
+    # check if the file has been uploaded
+    if fileUpload.filename:
+        # strip the leading path from the file name
+        fn = os.path.basename(fileUpload.filename)
+        # open read and write the file into the server
+        open(fn, 'wb').write(fileUpload.file.read())
+        subprocess.check_output(
+            [
+                "./truepic",
+                "sign",
+                fileUpload.filename,
+                "--output",
+                (os.getcwd() + "/static/" + fileUpload.filename),
+            ]
+        )
+    return {"response": fileUpload.filename}
+@app.post("/sign")
+def sign_image(fileUpload: UploadFile):
+    logging.warning("in verify")
+    logging.warning(fileUpload.filename)
     # check if the file has been uploaded
+    if fileUpload.filename:
         # strip the leading path from the file name
+        fn = os.path.basename(fileUpload.filename)
         # open read and write the file into the server
+        open(fn, 'wb').write(fileUpload.file.read())
+    return {"response": fileUpload.filename}
 app.mount("/", StaticFiles(directory="static", html=True), name="static")

static/index.html CHANGED Viewed

@@ -19,14 +19,20 @@
   ></script>
     <link rel="stylesheet" href="style.css" type="text/css" />
 </head>
   <body>
     <div class="container">
-        <form action="/upload" id="fileUploadForm" enctype = "multipart/form-data">
             <label class="form-label" for="customFile">Default file input example</label>
             <input type="file" class="form-control" name="fileUpload" id="fileUpload" />
             <button class="btn btn-primary" type="submit">Submit</button>
         </form>
     </div>
   </body>
@@ -57,6 +63,29 @@ function submitFormData(formData) {
   .then(response => response.json())
   .then(data => {
     console.log(data)
     // Handle response data
   })
   .catch(error => {

   ></script>
     <link rel="stylesheet" href="style.css" type="text/css" />
+    <script
+    type="module"
+    src="https://display.truepic.com/truepic_display.es.js"
+  ></script>
 </head>
   <body>
     <div class="container">
+        <form method="post" id="fileUploadForm" enctype = "multipart/form-data">
             <label class="form-label" for="customFile">Default file input example</label>
             <input type="file" class="form-control" name="fileUpload" id="fileUpload" />
             <button class="btn btn-primary" type="submit">Submit</button>
         </form>
+        <div id="image-container"></div>
     </div>
   </body>
   .then(response => response.json())
   .then(data => {
     console.log(data)
+    var resultsContainer = document.getElementById("image-container");
+    var truepicDisplay = document.createElement("truepic-display");
+    /*
+    truepicDisplay.addEventListener(
+      "validate",
+      setVerificationOutputFromValidation
+    );
+    truepicDisplay.addEventListener(
+      "validate",
+      setCertificateOutputFromValidation
+    );
+    */
+    truepicDisplay.setAttribute("id", "result");
+    truepicDisplay.setAttribute("active", "");
+    var truepic = document.createElement("img");
+    truepic.src = path;
+    truepicDisplay.appendChild(truepic);
+    resultsContainer.appendChild(truepicDisplay);
     // Handle response data
   })
   .catch(error => {