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

@@ -1588,7 +1588,11 @@ def content_header():
 
                 path_custom_prompts = os.path.join(st.session_state.dir_home,'custom_prompts',st.session_state.config['leafmachine']['project']['prompt_version'])
                 # Call the machine function.
-                st.session_state['formatted_json'], st.session_state['formatted_json_WFO'], st.session_state['formatted_json_GEO'], total_cost, n_failed_OCR, n_failed_LLM_calls, st.session_state['zip_filepath'] = voucher_vision(None,
+                total_cost = 0.00
+                n_failed_OCR = 0
+                n_failed_LLM_calls = 0
+                try:
+                    st.session_state['formatted_json'], st.session_state['formatted_json_WFO'], st.session_state['formatted_json_GEO'], total_cost, n_failed_OCR, n_failed_LLM_calls, st.session_state['zip_filepath'] = voucher_vision(None,
                                                                                                                 st.session_state.dir_home, 
                                                                                                                 path_custom_prompts, 
                                                                                                                 None, 
@@ -1597,24 +1601,32 @@ 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)
-                
+                except Exception as e:
+                    with col_run_4:
+                        st.error(f"Transcription failed. Error: {e}")
+
                 if n_failed_OCR > 0:
-                    st.error(f"Caution:heavy_exclamation_mark: :loudspeaker: {n_failed_LLM_calls} images had a no extractable OCR text :eyes:")
+                    with col_run_4:
+                        st.error(f"Caution:heavy_exclamation_mark: :loudspeaker: {n_failed_LLM_calls} images had a no extractable OCR text :eyes:")
 
                 if n_failed_LLM_calls > 0:
-                    st.error(f"Caution:heavy_exclamation_mark: :loudspeaker: {n_failed_LLM_calls} images had a failed LLM API call :eyes:")
-                    st.error(f"Make sure that you have access to the chosen LLM API model. Sometimes certain OpenAI accounts do not have access to all models, for example")
+                    with col_run_4:
+                        st.error(f"Caution:heavy_exclamation_mark: :loudspeaker: {n_failed_LLM_calls} images had a failed LLM API call :eyes:")
+                        st.error(f"Make sure that you have access to the chosen LLM API model. Sometimes certain OpenAI accounts do not have access to all models, for example")
                 
                 if total_cost:
-                    st.success(f":money_with_wings: This run cost :heavy_dollar_sign:{total_cost:.4f}")
+                    with col_run_4:
+                        st.success(f":money_with_wings: This run cost :heavy_dollar_sign:{total_cost:.4f}")
                 else:
-                    st.info(f":money_with_wings: This run cost :heavy_dollar_sign:{total_cost:.4f}")
+                    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:
             st.button("Start Processing", type='primary', disabled=True)
-            st.error(":heavy_exclamation_mark: Required API keys not set. Please visit the 'API Keys' tab and set the Google Vision OCR API key and at least one LLM key.")
+            with col_run_4:
+                st.error(":heavy_exclamation_mark: Required API keys not set. Please visit the 'API Keys' tab and set the Google Vision OCR API key and at least one LLM key.")
       
         if st.session_state['formatted_json']:
             if st.session_state['hold_output']:

vouchervision/API_validation.py

@@ -185,10 +185,9 @@ class APIvalidation:
     def check_google_vertex_genai_api_key(self):
         results = {"palm2": False, "gemini": False}
         if self.cfg_private:
-            try:
+            try: # Local
                 # Assuming genai and vertexai are clients for Google services
                 # os.environ["GOOGLE_API_KEY"] = self.cfg_private['google_palm']['google_palm_api']
-                os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = self.get_google_credentials()
                 # genai.configure(api_key=self.cfg_private['google_palm']['google_palm_api'])
                 vertexai.init(project= self.cfg_private['google_palm']['project_id'], location=self.cfg_private['google_palm']['location'])
 
@@ -218,10 +217,11 @@ class APIvalidation:
             except Exception as e:  # Replace with a more specific exception if possible
                 return results
         else:
+            ### is hugging face
             try:
                 # Assuming genai and vertexai are clients for Google services
                 # os.environ["GOOGLE_API_KEY"] = os.getenv('PALM_API_KEY')
-                genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
+                os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = self.get_google_credentials()
                 vertexai.init(project= os.getenv('GOOGLE_PROJECT_ID'), location=os.getenv('GOOGLE_LOCATION'))
 
                 try: