updates

app.py

@@ -266,7 +266,7 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
 
             if SAFE.check_for_inappropriate_content(uploaded_file):
                 clear_image_uploads()
-                report_violation(uploaded_file.name.lower(), is_hf=st.session_state['is_hf'])
+                report_violation(uploaded_file.name, is_hf=st.session_state['is_hf'])
                 st.error("Warning: You uploaded an image that violates our terms of service.")
                 return True
 

vouchervision/utils_hf.py

@@ -147,7 +147,7 @@ def check_prompt_yaml_filename(fname):
     else:
         return False
 
-def report_violation(file_name, is_hf=True, cfg_private=None, do_upload=True):
+def report_violation(file_name, is_hf=True):
     # Format the current date and time
     current_time = datetime.datetime.now().strftime("%Y_%m_%d__%H_%M_%S")
     file_name = f"violation_{current_time}.yaml"
@@ -160,14 +160,15 @@ def report_violation(file_name, is_hf=True, cfg_private=None, do_upload=True):
             'notes': 'This is an autogenerated violation report.',
             'name_of_file': file_name,
         }
+        # Write the content to the temporary YAML file in text mode
         yaml.dump(content, temp_file, default_flow_style=False)
-        temp_file_path = temp_file.name  # Save the temp file path to use after closing
+        temp_filepath = temp_file.name
 
-    # Now, upload the temporary file
-    try:
-        upload_to_drive(temp_file_path, file_name, is_hf=is_hf, cfg_private=cfg_private, do_upload=do_upload)
-    finally:
-        os.remove(temp_file_path)  # Ensure the temporary file is deleted after uploading
+    # Now upload the temporary file
+    upload_to_drive(temp_filepath, file_name, is_hf=is_hf)
+
+    # Optionally, delete the temporary file if you don't want it to remain on disk after uploading
+    os.remove(temp_filepath)
 
 
 # Function to upload files to Google Drive