Update page_files/Upload_Data.py

page_files/Upload_Data.py

@@ -119,7 +119,7 @@ def extract_images(pdf_path: str) -> list:
     try:
         with open(pdf_path, "rb") as f:
             pdf_bytes = f.read()
-        plot_results, coverage = image2.extract_and_verify_plots(
+            plot_results, coverage = Pdf_ImageExtraction.extract_and_verify_plots(
             pdf_bytes, verify_engines=["gemini"]
         )
         # Drop captions naming photos/micrographs/logos and crops the verifier
@@ -656,13 +656,13 @@ def render_plots_tab(pdf_path: str, paper_id: str):
     d1, d2 = st.columns(2)
     d1.download_button(
         "⬇ Images + metadata (ZIP)",
-        data=image2.create_plot_zip(plot_results, include_json=True),
+        data=Pdf_ImageExtraction.create_plot_zip(plot_results, include_json=True),
         file_name=f"{paper_id}_plots.zip", mime="application/zip",
         use_container_width=True, key="dl_plots_zip",
     )
     d2.download_button(
         "⬇ Images only (ZIP)",
-        data=image2.create_plot_zip(plot_results, include_json=False),
+        data=Pdf_ImageExtraction.create_plot_zip(plot_results, include_json=False),
         file_name=f"{paper_id}_images.zip", mime="application/zip",
         use_container_width=True, key="dl_images_zip",
     )