Major update. Support for 15 LLMs, World Flora Online taxonomy validation, geolocation, 2 OCR methods, significant UI changes, stability improvements, consistent JSON parsing

app.py

@@ -255,6 +255,7 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
         # Clear input image gallery and input list
         clear_image_uploads()
 
+        ind_small = 0
         for uploaded_file in uploaded_files:
             # Determine the file type
             if uploaded_file.name.lower().endswith('.pdf'):
@@ -266,26 +267,30 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
                 converted_files = os.listdir(st.session_state['dir_uploaded_images'])
                 for file_name in converted_files:   
                     if file_name.split('.')[1].lower() in ['jpg','jpeg']:
+                        ind_small += 1
                         jpg_file_path = os.path.join(st.session_state['dir_uploaded_images'], file_name)
                         st.session_state['input_list'].append(jpg_file_path)
 
-                        # Optionally, create a thumbnail for the gallery
-                        img = Image.open(jpg_file_path)
-                        img.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)
-                        try:
-                            file_path_small = save_uploaded_file(st.session_state['dir_uploaded_images_small'], file_name, img)
-                        except:
-                            file_path_small = save_uploaded_file_local(st.session_state['dir_uploaded_images_small'],st.session_state['dir_uploaded_images_small'], file_name, img)
-                        st.session_state['input_list_small'].append(file_path_small)
+                        if ind_small < MAX_GALLERY_IMAGES +5:
+                            # Optionally, create a thumbnail for the gallery
+                            img = Image.open(jpg_file_path)
+                            img.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)
+                            try:
+                                file_path_small = save_uploaded_file(st.session_state['dir_uploaded_images_small'], file_name, img)
+                            except:
+                                file_path_small = save_uploaded_file_local(st.session_state['dir_uploaded_images_small'],st.session_state['dir_uploaded_images_small'], file_name, img)
+                            st.session_state['input_list_small'].append(file_path_small)
                 
             else:
+                ind_small += 1
                 # Handle JPG/JPEG files (existing process)
                 file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file)
                 st.session_state['input_list'].append(file_path)
-                img = Image.open(file_path)
-                img.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)
-                file_path_small = save_uploaded_file(st.session_state['dir_uploaded_images_small'], uploaded_file, img)
-                st.session_state['input_list_small'].append(file_path_small)
+                if ind_small < MAX_GALLERY_IMAGES +5:
+                    img = Image.open(file_path)
+                    img.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)
+                    file_path_small = save_uploaded_file(st.session_state['dir_uploaded_images_small'], uploaded_file, img)
+                    st.session_state['input_list_small'].append(file_path_small)
 
         # After processing all files
         st.session_state.config['leafmachine']['project']['dir_images_local'] = st.session_state['dir_uploaded_images']
@@ -334,27 +339,7 @@ def handle_image_upload_and_gallery():
                 file_path = save_uploaded_file_local(st.session_state.config['leafmachine']['project']['dir_images_local'], st.session_state['dir_uploaded_images'], input_file)
                 # Convert each page of the PDF to an image
                 n_pages = convert_pdf_to_jpg(file_path, st.session_state['dir_uploaded_images'], dpi=200)#st.session_state.config['leafmachine']['project']['dir_images_local'])
-                
-                
-                # pdf_files_pattern = os.path.join(st.session_state['dir_uploaded_images'], '*.pdf')
-                # for pdf_file in glob.glob(pdf_files_pattern):
-                #     os.remove(pdf_file)
-
-                # # Update the input list for each page image
-                # converted_files = os.listdir(st.session_state['dir_uploaded_images'])
-                # for file_name in converted_files:   
-                #     if file_name.lower().endswith('.jpg'):
-                #         jpg_file_path = os.path.join(st.session_state['dir_uploaded_images'], file_name)
-                #         st.session_state['input_list'].append(jpg_file_path)
-
-                #         # Optionally, create a thumbnail for the gallery
-                #         img = Image.open(jpg_file_path)
-                #         img.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)
-                #         file_path_small = save_uploaded_file_local(st.session_state['dir_uploaded_images'], st.session_state['dir_uploaded_images_small'], file_name, img)
-                #         st.session_state['input_list_small'].append(file_path_small)
-                
-                # st.session_state.config['leafmachine']['project']['dir_images_local'] = st.session_state['dir_uploaded_images']
-                
+
             else:
                 pass
                 # st.warning("Inputs must be '.PDF' or '.jpg' or '.jpeg'")
@@ -419,7 +404,6 @@ def list_jpg_files(directory_path):
             break
             
     for simg in jpg_files:
-
         simg2 = Image.open(simg)
         simg2.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)  
         file_path_small = save_uploaded_local(st.session_state['dir_uploaded_images_small'], simg, simg2)