add additional handling for ocr-able files: pdf jpg png

diff --git a/karmaworld/apps/notes/gdrive.py b/karmaworld/apps/notes/gdrive.py
index a12b6c928ac289a1eabf76d51bc2fdf3b3c85c4e..4c4f5fb9d9b2cd386f427420ae5dbc860198b1b5 100644 (file)
--- a/karmaworld/apps/notes/gdrive.py
+++ b/karmaworld/apps/notes/gdrive.py
@@ -123,9 +123,14 @@ def convert_with_google_drive(note):
 
     service, http = build_api_service(creds)
 
+    # get the file extension
+    filename, extension = os.path.splitext(note.note_file.path)
     # Upload the file
-    # TODO: wrap this in a try loop that does a token refresh if it fails
-    file_dict = service.files().insert(body=resource, media_body=media, convert=True, ocr=True).execute()
+    if extension.lower() in ['.pdf', '.jpg', '.png']:
+        # include OCR on ocr-able files
+        file_dict = service.files().insert(body=resource, media_body=media, convert=True, ocr=True).execute()
+    else:
+        file_dict = service.files().insert(body=resource, media_body=media, convert=True).execute()
 
     if u'exportLinks' not in file_dict:
         # wait some seconds