updates

app.py

@@ -14,7 +14,7 @@ from vouchervision.vouchervision_main import voucher_vision
 from vouchervision.general_utils import test_GPU, get_cfg_from_full_path, summarize_expense_report, validate_dir
 from vouchervision.model_maps import ModelMaps
 from vouchervision.API_validation import APIvalidation
-from vouchervision.utils_hf import setup_streamlit_config, save_uploaded_file, save_uploaded_local, save_uploaded_file_local
+from vouchervision.utils_hf import setup_streamlit_config, save_uploaded_file, save_uploaded_local, save_uploaded_file_local, report_violation
 from vouchervision.data_project import convert_pdf_to_jpg
 from vouchervision.utils_LLM import check_system_gpus
 from vouchervision.OCR_google_cloud_vision import SafetyCheck
@@ -266,7 +266,8 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
 
             if SAFE.check_for_inappropriate_content(uploaded_file):
                 clear_image_uploads()
-                st.error("Warning: You have uploaded an inappropriate image")
+                report_violation(uploaded_file.name.lower(), 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

@@ -1,4 +1,4 @@
-import os, json, re
+import os, json, re, datetime, tempfile, yaml
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaFileUpload
 from google.oauth2 import service_account
@@ -146,7 +146,30 @@ def check_prompt_yaml_filename(fname):
         return True
     else:
         return False
+
+def report_violation(file_name, is_hf=True, cfg_private=None, do_upload=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"
     
+    # Create a temporary YAML file
+    with tempfile.NamedTemporaryFile(delete=False, 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,
+        }
+        yaml.dump(content, temp_file, default_flow_style=False)
+        temp_file_path = temp_file.name  # Save the temp file path to use after closing
+
+    # 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
+
+
 # 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: