Fixed issues with incorrect pages getting converted

deadmantfa · Feb 22, 2021 · f75bc8d · f75bc8d
1 parent 6eff813
commit f75bc8d
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/main.py b/main.py
@@ -20,15 +20,9 @@
 def convert(pdf_document, start, end):
     try:
         text = ''
-        start -= 1
-        end -= 1
-        if start == end:
-            page_current = pdf_document.create_page(start)
+        for x in range(start - 1, end):
+            page_current = pdf_document.create_page(x)
             text += page_current.text()
-        else:
-            for x in range(start, end):
-                page_current = pdf_document.create_page(x)
-                text += page_current.text()
         # initialize tts, create mp3 and play
         mp3_fp = io.BytesIO()
         tts = gTTS(text=text, lang='en', slow=False, lang_check=False)