updates

app.py

@@ -253,9 +253,8 @@ def load_gallery(converted_files, uploaded_file):
 
 
 
-# @st.cache_data
 def handle_image_upload_and_gallery_hf(uploaded_files):
-    # SAFE = SafetyCheck(st.session_state['is_hf'])
+    SAFE = SafetyCheck(st.session_state['is_hf'])
     if uploaded_files:
         
         # Clear input image gallery and input list
@@ -264,11 +263,11 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
         ind_small = 0
         for uploaded_file in uploaded_files:
 
-            # if SAFE.check_for_inappropriate_content(uploaded_file):
-            #     clear_image_uploads()
-            #     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
+            if SAFE.check_for_inappropriate_content(uploaded_file):
+                clear_image_uploads()
+                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
 
             
             # Determine the file type
@@ -319,10 +318,9 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
             images_to_display = st.session_state['input_list_small']
         show_gallery_small_hf(images_to_display)
     
-    # return False
+    return False
 
 
-# @st.cache_data
 def handle_image_upload_and_gallery():
     
     if st.session_state['view_local_gallery'] and st.session_state['input_list_small'] and (st.session_state['dir_images_local_TEMP'] == st.session_state.config['leafmachine']['project']['dir_images_local']):
@@ -371,9 +369,6 @@ def handle_image_upload_and_gallery():
 
 
 def content_input_images(col_left, col_right):
-    SAFE = SafetyCheck(st.session_state['is_hf'])
-    is_violation = False
-
     st.write('---')
     # col1, col2 = st.columns([2,8])
     with col_left:
@@ -396,17 +391,7 @@ def content_input_images(col_left, col_right):
     
     with col_right:
         if st.session_state.is_hf:
-            if uploaded_files:
-                is_violation = False
-
-                for uploaded_file in uploaded_files:
-                    if SAFE.check_for_inappropriate_content(uploaded_file):
-                        clear_image_uploads()
-                        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.")
-                        is_violation = True
-            if not is_violation:
-                handle_image_upload_and_gallery_hf(uploaded_files)
+            result = handle_image_upload_and_gallery_hf(uploaded_files)
 
         else:
             st.session_state['view_local_gallery'] = st.toggle("View Image Gallery",)