updates

main.py

@@ -19,7 +19,7 @@ app = FastAPI()
 
 def file_extension(filename):
   filename_list = filename.split(".")
-  return filename_list[1]
+  return filename_list[1].lower()
 
 
 @app.get("/generate")
@@ -84,20 +84,22 @@ def verify_image(fileUpload: UploadFile):
 
     # check if the file has been uploaded
     if fileUpload.filename:
+        fileupload_extension = file_extension(fileUpload.filename)
+        input_filename = str(uuid.uuid4()) + "." + fileupload_extension
+        output_filename = str(uuid.uuid4()) + "." + fileupload_extension
+
         # strip the leading path from the file name
-        fn = os.path.basename(fileUpload.filename)
+        fn = os.path.basename(input_filename)
 
         # open read and write the file into the server
         open(fn, "wb").write(fileUpload.file.read())
 
-        fileupload_extension = file_extension(fileUpload.filename)
-        filename = str(uuid.uuid4()) + "." + fileupload_extension
-
+        
         response = subprocess.check_output(
             [
                 "./scripts/verify.sh",
-                fileUpload.filename,
-                filename
+                input_filename,
+                output_filename
             ]
         )
 
@@ -121,12 +123,12 @@ def verify_image(fileUpload: UploadFile):
             response = subprocess.check_output(
                 [
                     "cp",
-                    fileUpload.filename,
-                    "static/" + filename,
+                    input_filename,
+                    "static/" + output_filename,
                 ]
             )
 
-            result_media = filename
+            result_media = output_filename
 
         elif original_media != 'n/a':
             original_media_extension = file_extension(original_media)

static/index.html

@@ -524,6 +524,9 @@
         .then((data) => {
           console.log(data);
 
+          fileUpload.value = "";
+          setVerifyButtonStatus();
+
           if (data.contains_c2pa == "true") {
             contentCredentialsIcon.setAttribute("class", "check-icon");
           } else {