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

vouchervision/utils_hf.py

@@ -72,15 +72,16 @@ def check_prompt_yaml_filename(fname):
 # Function to upload files to Google Drive
 def upload_to_drive(filepath, filename):
     # Parse the service account info from the environment variable
-    creds_info = json.loads(os.environ.get('GDRIVE_API'))
+    creds_info = os.environ.get('GDRIVE_API')
     if creds_info:
+        creds_info = json.loads(creds_info)
         creds = service_account.Credentials.from_service_account_info(
             creds_info, scopes=["https://www.googleapis.com/auth/drive"]
         )
         service = build('drive', 'v3', credentials=creds)
 
         # Get the folder ID from the environment variable
-        folder_id = os.environ.get('GDRIVE')
+        folder_id = os.environ.get('GDRIVE_FOLDER_ID')  # Renamed for clarity
 
         if folder_id:
             file_metadata = {
@@ -94,12 +95,17 @@ def upload_to_drive(filepath, filename):
             elif filename.endswith('.zip'):
                 mimetype = 'application/zip'
             else:
-                mimetype = None  # or set a default mimetype
+                # Set a default mimetype if desired or handle the unsupported file type
+                print("Unsupported file type")
+                return None
 
-            if mimetype:
-                media = MediaFileUpload(filepath, mimetype=mimetype)
-                service.files().create(
-                    body=file_metadata,
-                    media_body=media,
-                    fields='id'
-                ).execute()
+            # Upload the file
+            media = MediaFileUpload(filepath, mimetype=mimetype)
+            file = service.files().create(
+                body=file_metadata,
+                media_body=media,
+                fields='id'
+            ).execute()
+            print(f"Uploaded file with ID: {file.get('id')}")
+    else:
+        print("GDRIVE_API environment variable not set.")