updates

main.py

@@ -59,6 +59,14 @@ def generate_image(prompt, model):
         ]
     )
 
+    subprocess.check_output(
+        [
+            "cp",
+            "output.jpg"
+            "static/" + filename,
+        ]
+    )
+
     return {"response": filename}
 
 
@@ -67,6 +75,7 @@ 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
@@ -82,9 +91,21 @@ def verify_image(fileUpload: UploadFile):
             ]
         )
 
+        
         logging.warning(response)
+        response_list = response.splitlines()
+        
+        c2pa_string = str(response_list[0])
+        c2pa = c2pa_string.split(":", 1)
+
+        watermark_string = str(response_list[1])
+        watermark = watermark_string.split(":", 1)
+
+        original_media_string = str(response_list[2])
+        original_media = original_media_string.split(":", 1)
+ 
+    return {"response": fileUpload.filename, "contains_c2pa" : c2pa[1], "contains_watermark" : watermark[1], "original_media" : original_media[1]}
 
-    return {"response": fileUpload.filename}
 
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")

static/index.html

@@ -142,7 +142,7 @@
                 <div class="form-group mb-3">
                   <label>Upload image</label>
                   <div class="custom-select">
-                    <input type="file" class="form-control" />
+                    <input type="file" class="form-control" name="fileUpload" id="fileUpload" />
                   </div>
                 </div>