Update page_files/Categorized_Search.py

page_files/Categorized_Search.py

@@ -32,16 +32,38 @@ def _load_plot_bytes(selected_abbr, property_choice, result_df):
         if "material_name" in result_df.columns:
             material_name = str(result_df.iloc[0]["material_name"])
     tables = [table] if table else list(_CLASS_TABLE.values())  # fallback: try all
+    mapper5 = None
     try:
+        import sys, os
+        # 1) flat layout (local dev: mapper5.py on sys.path / same dir)
         try:
-            from categorized.Backend import mapper5
+            import mapper5 as _m5
+            mapper5 = _m5
         except ModuleNotFoundError:
-            import sys, os
-            sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
+            pass
+        # 2) HF Space package layout: page_files/categorized/Backend/mapper5.py
+        if mapper5 is None:
             try:
-                from categorized.Backend import mapper5
+                from categorized.Backend import mapper5 as _m5
+                mapper5 = _m5
             except ModuleNotFoundError:
-                import mapper5   # flat-layout fallback
+                pass
+        # 3) add candidate dirs to sys.path and retry flat import
+        if mapper5 is None:
+            here = os.path.dirname(__file__)
+            candidates = [
+                here,
+                os.path.join(here, ".."),
+                os.path.join(here, "..", ".."),
+                os.path.join(here, "categorized", "Backend"),
+                os.environ.get("MAPPER5_DIR", ""),
+                r"C:\Users\999te\InDesDatabaseTool\langchain-env\Lib",
+            ]
+            for cand in candidates:
+                if cand and os.path.isdir(cand):
+                    sys.path.insert(0, os.path.abspath(cand))
+            import mapper5 as _m5
+            mapper5 = _m5
         from sqlalchemy import text
         engine = mapper5.get_db_engine()
     except Exception as e: