Change structure of validate()

main.py

@@ -49,26 +49,26 @@ def setup_credential_file(self):
 
     def validate(self):
         """Validate that we can run the OCR"""
-        print(self.get_document_count())
-        if self.get_document_count() == 0:
+        if self.get_document_count() is None:
             self.set_message(
                 "It looks like no documents were selected. Search for some or "
                 "select them and run again."
             )
             return False
-        if not self.org_id:
+        elif not self.org_id:
             self.set_message("No organization to charge.")
             return False
-        num_pages = 0
-        for document in self.get_documents():
-            num_pages += document.page_count
-        resp = self.client.post(
-            f"organizations/{self.org_id}/ai_credits/",
-            json={"ai_credits": num_pages},
-        )
-        if resp.status_code != 200:
-            self.set_message("Error charging AI credits.")
-            return False
+        else:
+            num_pages = 0
+            for document in self.get_documents():
+                num_pages += document.page_count
+            resp = self.client.post(
+                f"organizations/{self.org_id}/ai_credits/",
+                json={"ai_credits": num_pages},
+            )
+            if resp.status_code != 200:
+                self.set_message("Error charging AI credits.")
+                return False
         return True
 
     def json_ocr(self, input_dir, filename):