Update main.py

main.py

@@ -6,7 +6,7 @@ import subprocess
 import os
 import json
 import uuid
-import logging
+import html
 
 import torch
 from diffusers import (
@@ -26,6 +26,9 @@ def file_extension(filename):
 def generate_image(prompt, model):
     torch.cuda.empty_cache()
 
+    prompt = html.escape(prompt)
+    model = html.escape(model)
+
     modelArray = model.split(",")
     modelName = modelArray[0]
     modelVersion = modelArray[1]
@@ -79,9 +82,6 @@ def generate_image(prompt, model):
 
 @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:
         fileupload_extension = file_extension(fileUpload.filename)
@@ -102,9 +102,7 @@ def verify_image(fileUpload: UploadFile):
                 output_filename
             ]
         )
-
-        
-        logging.warning(response)
+     
         response_list = response.splitlines()
         
         c2pa_string = str(response_list[0])