Update app.py

app.py

@@ -452,24 +452,70 @@ def clone(repo_git, repo_hf, sdk_type, skip_lfs, enable_smart_generation):
             yield f"❌ Authentication failed: {str(e)}"
             return
         
+        # URL 정규화 및 검증
+        repo_git = repo_git.strip()
+        
+        # .git 확장자 추가 (필요한 경우)
+        if not repo_git.endswith('.git'):
+            repo_git_with_git = repo_git + '.git'
+        else:
+            repo_git_with_git = repo_git
+        
+        # HTTPS URL로 변환
+        if repo_git.startswith('git@github.com:'):
+            repo_git = repo_git.replace('git@github.com:', 'https://github.com/')
+        elif not repo_git.startswith('http'):
+            yield "❌ Invalid repository URL. Please use HTTPS URL (e.g., https://github.com/username/repo)"
+            return
+        
         yield f"📥 Cloning repository from {repo_git}..."
         
         env = os.environ.copy()
         env['GIT_LFS_SKIP_SMUDGE'] = '1'
-        clone_cmd = ['git', 'clone', '--recurse-submodules', repo_git, folder]
-        subprocess.run(clone_cmd, check=True, env=env)
         
+        # 먼저 .git 없이 시도
+        clone_cmd = ['git', 'clone', '--depth', '1', repo_git, folder]
+        result = subprocess.run(clone_cmd, env=env, capture_output=True, text=True)
+        
+        # 실패하면 .git 추가하여 재시도
+        if result.returncode != 0:
+            if ".git" not in repo_git:
+                yield "   Retrying with .git extension..."
+                clone_cmd = ['git', 'clone', '--depth', '1', repo_git_with_git, folder]
+                result = subprocess.run(clone_cmd, env=env, capture_output=True, text=True)
+            
+            if result.returncode != 0:
+                error_msg = result.stderr
+                if "Repository not found" in error_msg or "fatal: repository" in error_msg:
+                    yield f"❌ Repository not found. Please check:"
+                    yield f"   1. The repository URL is correct"
+                    yield f"   2. The repository is public"
+                    yield f"   3. The repository exists"
+                    yield f"   URL tried: {repo_git}"
+                else:
+                    yield f"❌ Git clone failed: {error_msg[:500]}"
+                return
+        
+        yield "✅ Repository cloned successfully"
+        
+        # Git 클론 재시도 로직 개선
         if not skip_lfs:
             yield "📦 Attempting to download LFS files..."
             try:
-                subprocess.run(['git', 'lfs', 'install'], cwd=folder, check=True)
-                lfs_result = subprocess.run(['git', 'lfs', 'pull'], cwd=folder, capture_output=True, text=True)
-                
-                if lfs_result.returncode != 0:
-                    yield f"⚠️ Warning: LFS download failed"
+                # Git LFS 설치 확인
+                lfs_check = subprocess.run(['git', 'lfs', 'version'], capture_output=True, text=True)
+                if lfs_check.returncode != 0:
+                    yield "⚠️ Git LFS not installed. Skipping LFS files..."
                     skip_lfs = True
                 else:
-                    yield "✅ LFS files downloaded successfully"
+                    subprocess.run(['git', 'lfs', 'install'], cwd=folder, check=True)
+                    lfs_result = subprocess.run(['git', 'lfs', 'pull'], cwd=folder, capture_output=True, text=True)
+                    
+                    if lfs_result.returncode != 0:
+                        yield f"⚠️ Warning: LFS download failed - {lfs_result.stderr[:200]}"
+                        skip_lfs = True
+                    else:
+                        yield "✅ LFS files downloaded successfully"
             except Exception as e:
                 yield f"⚠️ LFS error: {str(e)}"
                 skip_lfs = True