update: app.py

app.py

@@ -5,25 +5,36 @@ import importlib.util
 import sys
 
 
-# プライベートリポジトリからコードを取得
 def load_private_code():
     api = HfApi(token=os.environ["PRIVATE_REPOSITORY_HF_TOKEN"])
-
+    
     with TemporaryDirectory() as tmp_dir:
+        # プライベートスペースからファイルをダウンロード
         api.snapshot_download(
-            repo_id=os.environ["PRIVATE_REPOSITORY_NAME"],
+            repo_id="neoai-inc-private/pj-andromeda",
             repo_type="space",
             local_dir=tmp_dir
         )
-
+        
+        # tmp_dirをPythonパスに追加
+        sys.path.append(tmp_dir)
+        
+        # app.pyをインポート
         spec = importlib.util.spec_from_file_location(
             "private_app",
             os.path.join(tmp_dir, "app.py")
         )
         private_module = importlib.util.module_from_spec(spec)
         sys.modules["private_app"] = private_module
-        spec.loader.exec_module(private_module)
-
+        
+        try:
+            spec.loader.exec_module(private_module)
+        except ModuleNotFoundError as e:
+            print(f"Error loading module: {e}")
+            print(f"Current sys.path: {sys.path}")
+            print(f"Contents of tmp_dir: {os.listdir(tmp_dir)}")
+            raise
+        
         return private_module