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

@@ -242,14 +242,14 @@ def use_test_image():
         st.session_state['input_list_small'].append(file_path_small)
 
 
-def create_download_button_yaml(file_path, selected_yaml_file):
+def create_download_button_yaml(file_path, selected_yaml_file, key_val):
     file_label = f"Download {selected_yaml_file}"
     with open(file_path, 'rb') as f:
         st.download_button(
             label=file_label,
             data=f,
             file_name=os.path.basename(file_path),
-            mime='application/x-yaml',use_container_width=True,key=237895,
+            mime='application/x-yaml',use_container_width=True,key=key_val,
         )
 
 
@@ -1015,7 +1015,7 @@ def save_prompt_yaml(filename, col):
     upload_to_drive(filepath, filename) # added
 
     with col: # added
-        create_download_button_yaml(filepath, filename) # added
+        create_download_button_yaml(filepath, filename,key_val=2456237465) # added
 
 def check_unique_mapping_assignments():
     print(st.session_state['assigned_columns'])
@@ -1147,7 +1147,7 @@ def build_LLM_prompt_config():
                 download_file_path = os.path.join(dir_prompt, st.session_state['selected_yaml_file'] )
                 # Create the download button
                 st.write('##')
-                create_download_button_yaml(download_file_path, st.session_state['selected_yaml_file'] )
+                create_download_button_yaml(download_file_path, st.session_state['selected_yaml_file'],key_val=345798)
 
         # Prompt Author Information
         st.write("---")
@@ -1435,23 +1435,23 @@ def build_LLM_prompt_config():
 
 
 
-        st.write('---')
-        st.header("Save and Download Custom Prompt")
-        st.write('Once you click save, validation checks will verify the formatting and then a download button will appear so that you can ***save a local copy of your custom prompt.***')
-        col_left_save, col_right_save, _ = st.columns([2,2,8])
-        with col_left_save:
-            # Button to save the new YAML file
-            if st.button('Save YAML', type='primary',key=3450798):
-                if st.session_state['prompt_name']:
-                    if check_unique_mapping_assignments():
-                        if check_prompt_yaml_filename(st.session_state['prompt_name']):
-                            save_prompt_yaml(st.session_state['prompt_name'], col_right_save)
-                        else:
-                            st.error("File name can only contain letters, numbers, underscores, and dashes. Cannot contain spaces.")
-                    else:
-                        st.error("Mapping contains an error. Make sure that each column is assigned to only ***one*** category.")
-                else:
-                    st.error("Please enter a filename.")
+        # st.write('---')
+        # st.header("Save and Download Custom Prompt")
+        # st.write('Once you click save, validation checks will verify the formatting and then a download button will appear so that you can ***save a local copy of your custom prompt.***')
+        # col_left_save, col_right_save, _ = st.columns([2,2,8])
+        # with col_left_save:
+        #     # Button to save the new YAML file
+        #     if st.button('Save YAML', type='primary',key=3450798):
+        #         if st.session_state['prompt_name']:
+        #             if check_unique_mapping_assignments():
+        #                 if check_prompt_yaml_filename(st.session_state['prompt_name']):
+        #                     save_prompt_yaml(st.session_state['prompt_name'], col_right_save)
+        #                 else:
+        #                     st.error("File name can only contain letters, numbers, underscores, and dashes. Cannot contain spaces.")
+        #             else:
+        #                 st.error("Mapping contains an error. Make sure that each column is assigned to only ***one*** category.")
+        #         else:
+        #             st.error("Please enter a filename.")
 
     # with col_prompt_main_right:
     #     st.subheader('All Prompt Components')
@@ -1751,6 +1751,7 @@ def content_header():
                                                                                                                 path_api_cost=os.path.join(st.session_state.dir_home,'api_cost','api_cost.yaml'),
                                                                                                                 is_hf = st.session_state['is_hf'], 
                                                                                                                 is_real_run=True)
+                    st.balloons()
                 except Exception as e:
                     with col_run_4:
                         st.error(f"Transcription failed. Error: {e}")
@@ -1770,7 +1771,6 @@ def content_header():
                 else:
                     with col_run_4:
                         st.info(f":money_with_wings: This run cost :heavy_dollar_sign:{total_cost:.4f}")
-                st.balloons()
             if st.session_state['zip_filepath']:
                 create_download_button(st.session_state['zip_filepath'], col_run_1)
         else:
@@ -2089,7 +2089,7 @@ def content_collage_overlay():
             raise
         
         st.write("Supplement Google Vision OCR with trOCR (handwriting OCR) using `microsoft/trocr-base-handwritten`. This option requires Google Vision API and a GPU.")
-        do_use_trOCR = st.checkbox("Enable trOCR", value=st.session_state.config['leafmachine']['project']['do_use_trOCR'],disabled=st.session_state['lacks_GPU'])
+        do_use_trOCR = st.checkbox("Enable trOCR", value=st.session_state.config['leafmachine']['project']['do_use_trOCR'])#,disabled=st.session_state['lacks_GPU'])
         st.session_state.config['leafmachine']['project']['do_use_trOCR'] = do_use_trOCR
 
 
@@ -2407,13 +2407,13 @@ def main():
         content_prompt_and_llm_version()
     with col4:
         content_api_check()
-    content_llm_cost()
     content_collage_overlay()
+    content_llm_cost()
     content_processing_options()
     content_less_used()
-    content_archival_components()
-    content_space_saver()
-    # content_tab_domain()
+    with st.expander("View additional settings"):
+        content_archival_components()
+        content_space_saver()