Version 1.8RC

src/mcp_server/tools.py

@@ -195,9 +195,13 @@ def deploy_to_space(draft_id: str, visibility: str = "public", space_target: str
                  if hf_user:
                      repo_id_to_fetch = f"{hf_user}/{final_space_name}"
             
+            import requests
+            existing_reqs = set()
+            fetch_success = False
+
+            # Méthode 1: Via API HuggingFace (hf_hub_download)
             try:
-                print(f"DEBUG: Tentative de récupération des requirements existants sur {repo_id_to_fetch}...")
-                # Téléchargement du requirements.txt existant
+                print(f"DEBUG: Tentative de récupération des requirements via API sur {repo_id_to_fetch}...")
                 cached_path = hf_hub_download(
                     repo_id=repo_id_to_fetch,
                     filename="requirements.txt",
@@ -206,24 +210,44 @@ def deploy_to_space(draft_id: str, visibility: str = "public", space_target: str
                 )
                 with open(cached_path, 'r') as f:
                     existing_reqs = set(f.read().splitlines())
-                
-                # Fusion (Union)
-                merged_reqs = existing_reqs.union(new_reqs)
-                # Nettoyage (lignes vides)
-                merged_content = "\n".join(sorted([r for r in merged_reqs if r.strip()]))
-                
-                files_to_deploy["requirements.txt"] = merged_content
-                print(f"DEBUG: Requirements fusionnés avec succès ({len(existing_reqs)} existants + nouveaux).")
-                
+                fetch_success = True
+                print(f"DEBUG: Requirements récupérés via API ({len(existing_reqs)} items).")
             except (EntryNotFoundError, RepositoryNotFoundError):
-                print("DEBUG: Pas de requirements.txt existant ou repo introuvable, utilisation des nouveaux uniquement.")
-                # Le fichier n'existe pas encore, on garde celui du draft
-                pass
+                print("DEBUG: Pas de requirements.txt via API (404/Not Found).")
             except Exception as e:
-                print(f"DEBUG: Erreur lors de la fusion des requirements: {e}")
-                # En cas d'erreur, on garde les nouveaux pour ne pas bloquer, 
-                # mais on pourrait aussi choisir d'échouer. Ici on force l'ajout.
-                pass
+                print(f"DEBUG: Erreur API lors de la récupération requirements: {e}")
+
+            # Méthode 2: Via URL directe (Fallback "Raw")
+            # Utile si l'API échoue ou si le cache local est incohérent
+            if not fetch_success and "/" in repo_id_to_fetch:
+                try:
+                    raw_url = f"https://huggingface.co/spaces/{repo_id_to_fetch}/resolve/main/requirements.txt"
+                    print(f"DEBUG: Tentative de récupération via URL Raw: {raw_url}")
+                    headers = {}
+                    if deployer.token:
+                        headers["Authorization"] = f"Bearer {deployer.token}"
+                    
+                    resp = requests.get(raw_url, headers=headers)
+                    if resp.status_code == 200:
+                        existing_reqs = set(resp.text.splitlines())
+                        fetch_success = True
+                        print(f"DEBUG: Requirements récupérés via URL Raw ({len(existing_reqs)} items).")
+                    elif resp.status_code == 404:
+                         print("DEBUG: requirements.txt non trouvé via URL Raw (404).")
+                    else:
+                         print(f"DEBUG: Erreur HTTP {resp.status_code} lors de la récupération via URL Raw.")
+                except Exception as e:
+                    print(f"DEBUG: Exception lors de la récupération via URL Raw: {e}")
+
+            # Fusion finale
+            if fetch_success or existing_reqs:
+                merged_reqs = existing_reqs.union(new_reqs)
+                # Nettoyage
+                cleaned_reqs = sorted([r.strip() for r in merged_reqs if r.strip()])
+                files_to_deploy["requirements.txt"] = "\n".join(cleaned_reqs)
+                print(f"DEBUG: Fusion terminée. Total requirements: {len(cleaned_reqs)}")
+            else:
+                print("DEBUG: Aucun requirements existant trouvé, déploiement des nouveaux uniquement.")
             
     try:
         url = deployer.deploy_space(