updates

main.py

@@ -17,6 +17,10 @@ from diffusers import (
 
 app = FastAPI()
 
+def file_extension(filename):
+  filename_list = filename.split(".")
+  return filename_list[1]
+
 
 @app.get("/generate")
 def generate_image(prompt, model):
@@ -109,6 +113,20 @@ def verify_image(fileUpload: UploadFile):
         original_media = original_media_string.split(":", 1)
         original_media = original_media[1].strip(" ").strip("'")
 
+        if original_media != 'n/a':
+            original_media_extension = file_extension(original_media)
+            logging.warning(original_media_extension)
+    
+            filename = str(uuid.uuid4()) + original_media_extension
+            
+            response = subprocess.check_output(
+                [
+                    "cp",
+                    original_media,
+                    "static/" + filename,
+                ]
+            ) 
+
  
     return {"response": fileUpload.filename, "contains_c2pa" : c2pa, "contains_watermark" : watermark, "original_media" : original_media}
 
@@ -120,3 +138,6 @@ 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

@@ -80,11 +80,11 @@
               <p>Contains Credentials: <span id="contentCredentialResults"></span></p>
               <p>Digital Watermark: <span id="digitalWatermarkResults"></span></p>
               <p>Original Media: <span id="originalMedia"></span></p>
-              <!--
+              
                 <div class="image-container">
 
                 </div>
-              -->
+              
             </div>
           </div>
           <div class="col right-column">
@@ -202,7 +202,7 @@
     const imagePrompt = document.getElementById("prompt");
     const model = document.getElementById("model");
     const generateImageContainer = document.querySelector(".display-generate .image-container");
-    
+    const verifyImageContainer = document.querySelector(".display-verify .image-container");
 
     generateTab.addEventListener("click", (event) => {
       event.target.classList.add("active");
@@ -264,32 +264,32 @@
           document.getElementById("digitalWatermarkResults").innerHTML =  data.contains_watermark;
           document.getElementById("originalMedia").innerHTML =  data.original_media;
 
-/*
-          const path = "/" + data.response;
-
-          var resultsContainer = document.getElementById("image-container");
+          if (data.original_media != 'n/a') {
+            const path = "/" + data.original_media;
 
-          var truepicDisplay = document.createElement("truepic-display");
+            var truepicDisplay = document.createElement("truepic-display");
           
-    truepicDisplay.addEventListener(
-      "validate",
-      setVerificationOutputFromValidation
-    );
-    truepicDisplay.addEventListener(
-      "validate",
-      setCertificateOutputFromValidation
-    );
+            /*
+            truepicDisplay.addEventListener(
+              "validate",
+              setVerificationOutputFromValidation
+            );
+            truepicDisplay.addEventListener(
+              "validate",
+              setCertificateOutputFromValidation
+            );
+            */
     
 
-          truepicDisplay.setAttribute("id", "result");
-          truepicDisplay.setAttribute("active", "");
-          var truepic = document.createElement("img");
-          truepic.src = path;
+            truepicDisplay.setAttribute("id", "result");
+            truepicDisplay.setAttribute("active", "");
+            var truepic = document.createElement("img");
+            truepic.src = path;
 
-          truepicDisplay.appendChild(truepic);
+            truepicDisplay.appendChild(truepic);
 
-          resultsContainer.appendChild(truepicDisplay);
-          */
+            verifyImageContainer.appendChild(truepicDisplay);
+          }
           // Handle response data
         })
         .catch((error) => {