Update app.py

app.py

@@ -190,7 +190,7 @@ def run_yolo_detection_and_count(
 
 
 # ============================================================================
-# --- MAIN DOCUMENT PROCESSING FUNCTION ---
+# --- MAIN DOCUMENT PROCESSING FUNCTION (Fixed for unique filenames) ---
 # ============================================================================
 
 def run_single_pdf_preprocessing(pdf_path: str, temp_output_dir: str) -> Tuple[int, int, int, str, List[str]]:
@@ -227,6 +227,9 @@ def run_single_pdf_preprocessing(pdf_path: str, temp_output_dir: str) -> Tuple[i
     mat = fitz.Matrix(SCALE_FACTOR, SCALE_FACTOR)
     all_equation_images = [] # Stores file paths (strings) for Gradio gallery
     
+    # NEW LOCAL COUNTER: Tracks total equations processed for unique filename creation
+    equation_save_count = 0
+    
     for page_num_0_based in range(doc.page_count):
         fitz_page = doc.load_page(page_num_0_based)
         page_num = page_num_0_based + 1
@@ -257,7 +260,11 @@ def run_single_pdf_preprocessing(pdf_path: str, temp_output_dir: str) -> Tuple[i
                 
                 # Save to a temporary file path
                 img_bytes = eq_pix.tobytes("png")
-                filename = f"eq_{GLOBAL_EQUATION_COUNT}_p{page_num}.png"
+                
+                # FIX APPLIED: Increment and use local counter for unique filename
+                equation_save_count += 1
+                filename = f"eq_{equation_save_count}_p{page_num}.png"
+                
                 output_path = os.path.join(temp_output_dir, filename)
 
                 with open(output_path, 'wb') as f:
@@ -270,7 +277,7 @@ def run_single_pdf_preprocessing(pdf_path: str, temp_output_dir: str) -> Tuple[i
 
     doc.close()
 
-    # Final Report Generation
+    # Final Report Generation (GLOBAL_EQUATION_COUNT is correct here)
     report = (
         f"✅ **YOLO Counting Complete!**\n\n"
         f"**1) Total Pages Detected in PDF:** **{total_pages}**\n"
@@ -283,22 +290,18 @@ def run_single_pdf_preprocessing(pdf_path: str, temp_output_dir: str) -> Tuple[i
 
 
 # ============================================================================
-# --- GRADIO INTERFACE FUNCTION (Fixed for temp dir cleanup) ---
+# --- GRADIO INTERFACE FUNCTION ---
 # ============================================================================
 
 def gradio_process_pdf(pdf_file) -> Tuple[str, str, str, str, List[str]]:
     """
     Gradio wrapper function to handle file upload, manage temporary directory, and return file paths.
-    
-    The cleanup block is REMOVED to allow Gradio's front-end to access the images
-    before the files are deleted. Gradio will eventually handle the cleanup.
     """
     if pdf_file is None:
         return "N/A", "N/A", "N/A", "Please upload a PDF file.", []
     
     pdf_path = pdf_file.name
-    # Create temp directory
-    temp_output_dir = tempfile.mkdtemp() 
+    temp_output_dir = tempfile.mkdtemp() # Create temp directory
 
     try:
         # Run the core logic, passing the temp directory