Update app.py

app.py

@@ -62,11 +62,11 @@ class HFToMSConverter:
                 src.rename(dst)
                 logger.info(f"移动文件成功: {src.name}")
             elif operation_type == "push":
-                print(clone_dir)
                 ms_repo_id, clone_dir = args
+                logger.info(f"开始推送文件夹: {clone_dir}")
                 self.ms_api.upload_folder(
-                    repo_id=f"{self.config['username']}/{ms_repo_id}",
-                    folder_path=clone_dir,
+                    repo_id=f"{ms_repo_id}",
+                    folder_path=str(clone_dir),  # 确保路径是字符串
                     commit_message='upload dataset folder',
                     repo_type='dataset'
                 )
@@ -76,42 +76,44 @@ class HFToMSConverter:
             logger.error(f"{operation_type}操作失败: {e}")
             return False
 
-    def move_file(self, filename: str, clone_dir: str) -> bool:
-        """移动文件到目标目录"""
-        return self.handle_file_operation(
-            "move",
-            Path(self.local_dir) / filename,
-            Path(clone_dir) / filename
-        )
-
-    def push_to_ms(self, ms_repo_id: str, clone_dir: str) -> bool:
-        """推送到ModelScope"""
-        return self.handle_file_operation("push", ms_repo_id, clone_dir)
-
     def process_files(self, hf_repo: str, ms_repo: str, files: List[str]) -> bool:
         """处理所有文件的完整流程"""
-        # 获取绝对路径
-        print(files)
-        os_path = os.path.abspath('.')
-        
-        clone_dir = os.path.join(os_path,  ms_repo.split("/")[-1])
-        
         try:
-            # # 下载并移动所有文件
+            # 获取绝对路径并创建目录
+            clone_dir = Path(os.path.abspath('.')) / ms_repo.split("/")[-1]
+            clone_dir.mkdir(parents=True, exist_ok=True)
+            
+            # 下载并移动所有文件
             for filename in files:
                 if not all([
                     self.download_file(hf_repo, filename),
-                    self.move_file(filename, clone_dir)
+                    self.move_file(filename, str(clone_dir))
                 ]):
                     return False
             
-            print('111'*10)
-            print(clone_dir)
             # 统一推送整个文件夹
             return self.push_to_ms(ms_repo, clone_dir)
+            
         except Exception as e:
             logger.error(f"处理文件失败: {e}")
             return False
+        finally:
+            # 清理临时文件
+            if clone_dir.exists():
+                import shutil
+                shutil.rmtree(clone_dir)
+
+    def move_file(self, filename: str, clone_dir: str) -> bool:
+        """移动文件到目标目录"""
+        return self.handle_file_operation(
+            "move",
+            Path(self.local_dir) / filename,
+            Path(clone_dir) / filename
+        )
+
+    def push_to_ms(self, ms_repo_id: str, clone_dir: str) -> bool:
+        """推送到ModelScope"""
+        return self.handle_file_operation("push", ms_repo_id, clone_dir)
 
 def create_ui() -> gr.Blocks:
     """创建Gradio界面"""