Update app.py

app.py

@@ -340,6 +340,7 @@ def run_single_pdf_preprocessing(pdf_path: str) -> Tuple[int, int, int, str, flo
     global GLOBAL_FIGURE_COUNT, GLOBAL_EQUATION_COUNT
     start_time = time.time()
     log_messages = []
+    all_saved_images = []
     all_base64_images: List[str] = []
     
     # Dictionary to store {page_number (int): equation_count (int)}
@@ -399,7 +400,7 @@ def run_single_pdf_preprocessing(pdf_path: str) -> Tuple[int, int, int, str, flo
         detect_start = time.time()
         # page_equations, _ = run_yolo_detection_and_count(original_img, model, page_num)
         page_equations, _, page_images = run_yolo_detection_and_count(original_img, model, page_num)
-        all_base64_images.extend(page_images)
+        all_saved_images.extend(page_images)
         
         detect_time = time.time() - detect_start
         
@@ -437,7 +438,8 @@ def run_single_pdf_preprocessing(pdf_path: str) -> Tuple[int, int, int, str, flo
 
     # Return the dictionary with string keys
     # return total_pages, GLOBAL_EQUATION_COUNT, GLOBAL_FIGURE_COUNT, report, total_execution_time, equation_counts_per_page_str_keys, []
-    return total_pages, GLOBAL_EQUATION_COUNT, GLOBAL_FIGURE_COUNT, report, total_execution_time, equation_counts_per_page_str_keys, all_base64_images
+    return total_pages, GLOBAL_EQUATION_COUNT, GLOBAL_FIGURE_COUNT, report, total_execution_time, equation_counts_per_page_str_keys, all_saved_images
+
 
 
 # ============================================================================
@@ -459,8 +461,10 @@ def gradio_process_pdf(pdf_file) -> Tuple[str, str, str, str, Dict[str, int], Li
         # num_pages, num_equations, num_figures, report, total_time, equation_counts_per_page, _ = run_single_pdf_preprocessing(
         #     pdf_path
         # )
+        # num_pages, num_equations, num_figures, report, total_time, equation_counts_per_page, images = run_single_pdf_preprocessing(pdf_path)
         num_pages, num_equations, num_figures, report, total_time, equation_counts_per_page, images = run_single_pdf_preprocessing(pdf_path)
 
+
         
         # Return results (6 items now)
         # return str(num_pages), str(num_equations), str(num_figures), report, equation_counts_per_page, []