updates

vouchervision/utils_hf.py

@@ -150,27 +150,26 @@ def check_prompt_yaml_filename(fname):
 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"
+    violation_file_name = f"violation_{current_time}.yaml"  # Updated variable name to avoid confusion
     
-    # Create a temporary YAML file
-    with tempfile.NamedTemporaryFile(delete=False, suffix='.yaml') as temp_file:
+    # Create a temporary YAML file in text mode
+    with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.yaml') as temp_file:
         # Example content - customize as needed
         content = {
             'violation_time': current_time,
             'notes': 'This is an autogenerated violation report.',
-            'name_of_file': file_name,
+            'name_of_file': violation_file_name,
         }
         # Write the content to the temporary YAML file in text mode
         yaml.dump(content, temp_file, default_flow_style=False)
         temp_filepath = temp_file.name
 
     # Now upload the temporary file
-    upload_to_drive(temp_filepath, file_name, is_hf=is_hf)
+    upload_to_drive(temp_filepath, violation_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
 def upload_to_drive(filepath, filename, is_hf=True, cfg_private=None, do_upload = True):
     if do_upload: