Update src/streamlit_app.py

src/streamlit_app.py

@@ -304,6 +304,21 @@ def find_actual_file(file_name):
     if not file_name:
         return None
     
+    # FIRST: Check if this is a generated filename from a multi-page PDF (e.g., "12_page1.png")
+    if file_name.endswith('.png') and '_page' in file_name:
+        base_name = os.path.splitext(file_name)[0]  # Remove .png
+        # Extract the base name before _pageN
+        if '_page' in base_name:
+            pdf_base = base_name.split('_page')[0]
+            # Try to find the PDF file
+            if pdf_base + '.pdf' in st.session_state.images:
+                return pdf_base + '.pdf'
+            # Also try without any extension in case original didn't have it
+            for uploaded_name in st.session_state.images.keys():
+                uploaded_base = os.path.splitext(uploaded_name)[0]
+                if uploaded_base == pdf_base and uploaded_name.lower().endswith('.pdf'):
+                    return uploaded_name
+    
     # Try exact match
     if file_name in st.session_state.images:
         return file_name
@@ -319,15 +334,6 @@ def find_actual_file(file_name):
         if uploaded_base == file_name:
             return uploaded_name
     
-    # Try matching any filename in file_names list
-    if file_name.endswith('.png'):
-        base_name = os.path.splitext(file_name)[0]
-        # Check if this is a page from a multi-page PDF
-        if '_page' in base_name:
-            pdf_base = base_name.split('_page')[0]
-            if pdf_base + '.pdf' in st.session_state.images:
-                return pdf_base + '.pdf'
-    
     return None
 
 # Initialize session state