updates

app.py

@@ -255,7 +255,7 @@ 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 +264,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,7 +319,7 @@ 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
@@ -371,6 +371,8 @@ def handle_image_upload_and_gallery():
 
 
 def content_input_images(col_left, col_right):
+    SAFE = SafetyCheck(st.session_state['is_hf'])
+
     st.write('---')
     # col1, col2 = st.columns([2,8])
     with col_left:
@@ -393,7 +395,16 @@ def content_input_images(col_left, col_right):
     
     with col_right:
         if st.session_state.is_hf:
-            result = handle_image_upload_and_gallery_hf(uploaded_files)
+            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)
 
         else:
             st.session_state['view_local_gallery'] = st.toggle("View Image Gallery",)