Update split_files_to_excel.py

split_files_to_excel.py

@@ -527,101 +527,103 @@ def split_by_keywords(files, key_words,words_limit=1000):
 
     tabLine = []
     for file in files:
-
-        file = PdfReader(file)
-        pdfNumberPages = len(file.pages)
-        for pdfPage in range(0, pdfNumberPages):
-
-            load_page = file.get_page(pdfPage)
-            text = load_page.extract_text()
-            lines = text.split("\n")
-            sizeOfLines = len(lines) - 1
-
-            for index, line in enumerate(lines):
-                print(line)
-                for key in key_words:
-                    line = line.lower()
-
-                    if key in line:
-                        print("Found keyword")
-
-                        # Init variables for search
-                        lineBool = True
-                        lineIndex = index
-                        previousSelectedLines = []
-                        stringLength = 0
-                        linesForSelection = lines
-                        loadOnce = True
-                        selectedPdfPage = pdfPage
-
-                        # Loop while extracting text before keyword
-                        while lineBool:
-                            print(lineIndex)
-                            if stringLength > words_limit or lineIndex < 0:
-                                lineBool = False
-                            else:
-                                if lineIndex == 0:
-                                    if pdfPage == 0:
-                                        lineBool = False
-                                    
-                                    # Load previous page
-                                    else:
-                                        try:
-                                            selectedPdfPage -= 1
-                                            newLoad_page = file.get_page(selectedPdfPage)
-                                            newText = newLoad_page.extract_text()
-                                            newLines = newText.split("\n")
-                                            linesForSelection = newLines
-                                            lineIndex = len(newLines) - 1
-
-                                        except Exception as e:
-                                            print(f"Loading previous PDF page failed")
+        
+        if file.endswith('pdf'):
+            
+            file = PdfReader(file)
+            pdfNumberPages = len(file.pages)
+            for pdfPage in range(0, pdfNumberPages):
+    
+                load_page = file.get_page(pdfPage)
+                text = load_page.extract_text()
+                lines = text.split("\n")
+                sizeOfLines = len(lines) - 1
+    
+                for index, line in enumerate(lines):
+                    print(line)
+                    for key in key_words:
+                        line = line.lower()
+    
+                        if key in line:
+                            print("Found keyword")
+    
+                            # Init variables for search
+                            lineBool = True
+                            lineIndex = index
+                            previousSelectedLines = []
+                            stringLength = 0
+                            linesForSelection = lines
+                            loadOnce = True
+                            selectedPdfPage = pdfPage
+    
+                            # Loop while extracting text before keyword
+                            while lineBool:
+                                print(lineIndex)
+                                if stringLength > words_limit or lineIndex < 0:
+                                    lineBool = False
+                                else:
+                                    if lineIndex == 0:
+                                        if pdfPage == 0:
                                             lineBool = False
-
-                                previousSelectedLines.append(linesForSelection[lineIndex])
-                                stringLength += len(linesForSelection[lineIndex])
-
-                                lineIndex -= 1
-                        previousSelectedLines = ' '.join(previousSelectedLines[::-1])
-                        
-                        # Init variables for search
-                        lineBool = True
-                        lineIndex = index + 1
-                        nextSelectedLines = ""
-                        linesForSelection = lines
-                        loadOnce = True
-                        selectedPdfPage = pdfPage
-
-                        # Loop while extracting text after keyword
-                        while lineBool:
-
-                            if len(nextSelectedLines.split()) > words_limit:
-                                lineBool = False
-                            else:
-                                if lineIndex > sizeOfLines:
+                                        
+                                        # Load previous page
+                                        else:
+                                            try:
+                                                selectedPdfPage -= 1
+                                                newLoad_page = file.get_page(selectedPdfPage)
+                                                newText = newLoad_page.extract_text()
+                                                newLines = newText.split("\n")
+                                                linesForSelection = newLines
+                                                lineIndex = len(newLines) - 1
+    
+                                            except Exception as e:
+                                                print(f"Loading previous PDF page failed")
+                                                lineBool = False
+    
+                                    previousSelectedLines.append(linesForSelection[lineIndex])
+                                    stringLength += len(linesForSelection[lineIndex])
+    
+                                    lineIndex -= 1
+                            previousSelectedLines = ' '.join(previousSelectedLines[::-1])
+                            
+                            # Init variables for search
+                            lineBool = True
+                            lineIndex = index + 1
+                            nextSelectedLines = ""
+                            linesForSelection = lines
+                            loadOnce = True
+                            selectedPdfPage = pdfPage
+    
+                            # Loop while extracting text after keyword
+                            while lineBool:
+    
+                                if len(nextSelectedLines.split()) > words_limit:
                                     lineBool = False
-
-                                    if pdfPage == pdfNumberPages - 1:
+                                else:
+                                    if lineIndex > sizeOfLines:
                                         lineBool = False
-                                    
-                                    # Load next page
-                                    else:
-                                        try:
-                                            selectedPdfPage += 1
-                                            newLoad_page = file.get_page(selectedPdfPage)
-                                            newText = newLoad_page.extract_text()
-                                            newLines = newText.split("\n")
-                                            linesForSelection = newLines
-                                            lineIndex = 0
-                                        except Exception as e:
-                                            print(f"Loading next PDF page failed")
+    
+                                        if pdfPage == pdfNumberPages - 1:
                                             lineBool = False
-                                else:
-                                    nextSelectedLines += " " + linesForSelection[lineIndex]
-                                lineIndex += 1
-
-                        selectedText = previousSelectedLines + ' ' + nextSelectedLines
-                        tabLine.append([pdfPage, selectedText, key])
+                                        
+                                        # Load next page
+                                        else:
+                                            try:
+                                                selectedPdfPage += 1
+                                                newLoad_page = file.get_page(selectedPdfPage)
+                                                newText = newLoad_page.extract_text()
+                                                newLines = newText.split("\n")
+                                                linesForSelection = newLines
+                                                lineIndex = 0
+                                            except Exception as e:
+                                                print(f"Loading next PDF page failed")
+                                                lineBool = False
+                                    else:
+                                        nextSelectedLines += " " + linesForSelection[lineIndex]
+                                    lineIndex += 1
+    
+                            selectedText = previousSelectedLines + ' ' + nextSelectedLines
+                            tabLine.append([pdfPage, selectedText, key])
 
     for r in tabLine:
         text_joined = ''.join(r[1])