Update app.py

app.py

@@ -9,44 +9,44 @@ import json
 # #files = [f for f in os.listdir("/Users/andreeabodea/") if f.endswith(".pdf")]
 # #print(files)
 
-# """
-# Extract the text from a section of a PDF file  between 'wanted_section' and 'next_section'.
-# Parameters: 
-# - path (str): The file path to the PDF file.
-# - wanted_section (str): The section to start extracting text from.
-# - next_section (str): The section to stop extracting text at.
-# Returns:
-# - text (str): The extracted text from the specified section range.
-# """
-# def get_section(path, wanted_section, next_section):
-#     print(wanted_section)
+"""
+Extract the text from a section of a PDF file  between 'wanted_section' and 'next_section'.
+Parameters: 
+- path (str): The file path to the PDF file.
+- wanted_section (str): The section to start extracting text from.
+- next_section (str): The section to stop extracting text at.
+Returns:
+- text (str): The extracted text from the specified section range.
+"""
+def get_section(path, wanted_section, next_section):
+    print(wanted_section)
 
-#     # Open the PDF file
-#     doc = pdfplumber.open(path)
-#     start_page = []
-#     end_page = []
+    # Open the PDF file
+    doc = pdfplumber.open(path)
+    start_page = []
+    end_page = []
 
-#     # Find the all the pages for the specified sections
-#     for page in range(len(doc.pages)):
-#         if len(doc.pages[page].search(wanted_section, return_chars = False, case = False)) > 0:
-#             start_page.append(page)
-#         if len(doc.pages[page].search(next_section, return_chars = False, case = False)) > 0:
-#             end_page.append(page)
-#     print(max(start_page))
-#     print(max(end_page))
+    # Find the all the pages for the specified sections
+    for page in range(len(doc.pages)):
+        if len(doc.pages[page].search(wanted_section, return_chars = False, case = False)) > 0:
+            start_page.append(page)
+        if len(doc.pages[page].search(next_section, return_chars = False, case = False)) > 0:
+            end_page.append(page)
+    print(max(start_page))
+    print(max(end_page))
 
-#     # Extract the text between the start and end page of the wanted section
-#     text = []
-#     for page_num in range(max(start_page), max(end_page)):
-#         page = doc.pages[page_num]
-#         text.append(page.extract_text())
-#     text = " ".join(text)
-#     new_text = text.replace("\n", " ")
-#     special_char_unicode_list = ["\u00e4", "\u00f6", "\u00fc", "\u00df"]
-#     special_char_replacement_list = ["ae", "oe", "ue", "ss"]
-#     for index, special_char in enumerate(special_char_unicode_list):
-#         final_text = new_text.replace(special_char, special_char_replacement_list[index])
-#     return final_text
+    # Extract the text between the start and end page of the wanted section
+    text = []
+    for page_num in range(max(start_page), max(end_page)):
+        page = doc.pages[page_num]
+        text.append(page.extract_text())
+    text = " ".join(text)
+    new_text = text.replace("\n", " ")
+    special_char_unicode_list = ["\u00e4", "\u00f6", "\u00fc", "\u00df"]
+    special_char_replacement_list = ["ae", "oe", "ue", "ss"]
+    for index, special_char in enumerate(special_char_unicode_list):
+        final_text = new_text.replace(special_char, special_char_replacement_list[index])
+    return final_text
 
 
 def process_pdf(path):
@@ -85,8 +85,6 @@ def process_pdf(path):
     #json_string = json.dumps(results_dict, indent=4)
     #print(json_string)
 
-
-
 # Define the Gradio interface
 iface = gr.Interface(fn=process_pdf, 
                      inputs=gr.File(type="binary", label="Upload PDF"),