phyloforfun commited on
Commit
d719f71
1 Parent(s): f83f83b

fix safety check

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -295,15 +295,18 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
295
  else:
296
  ind_small += 1
297
  # Handle JPG/JPEG files (existing process)
298
- file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file, image=None) ######### Yale TODO
299
  # file_path = os.path.join(st.session_state['dir_uploaded_images'], uploaded_file.name)
300
-
 
 
 
301
  st.session_state['input_list'].append(file_path)
302
- if ind_small < MAX_GALLERY_IMAGES +5:
303
- img = Image.open(file_path)
304
- img.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)
305
- file_path_small = save_uploaded_file(st.session_state['dir_uploaded_images_small'], uploaded_file, img)
306
- st.session_state['input_list_small'].append(file_path_small)
307
 
308
  # After processing all files
309
  st.session_state.config['leafmachine']['project']['dir_images_local'] = st.session_state['dir_uploaded_images']
 
295
  else:
296
  ind_small += 1
297
  # Handle JPG/JPEG files (existing process)
298
+ # file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file, image=None) ######### Yale TODO
299
  # file_path = os.path.join(st.session_state['dir_uploaded_images'], uploaded_file.name)
300
+ image = Image.open(uploaded_file)
301
+ full_path = os.path.join(st.session_state['dir_uploaded_images'], uploaded_file.name)
302
+ image.save(full_path, "JPEG")
303
+
304
  st.session_state['input_list'].append(file_path)
305
+ # if ind_small < MAX_GALLERY_IMAGES +5:
306
+ # img = Image.open(file_path)
307
+ # img.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)
308
+ # file_path_small = save_uploaded_file(st.session_state['dir_uploaded_images_small'], uploaded_file, img)
309
+ # st.session_state['input_list_small'].append(file_path_small)
310
 
311
  # After processing all files
312
  st.session_state.config['leafmachine']['project']['dir_images_local'] = st.session_state['dir_uploaded_images']