Update app.py

app.py

@@ -32,40 +32,30 @@ print(f"Diretório atual adicionado ao sys.path.")
 # --- ETAPA 3: Instalar Dependências Corretamente ---
 python_executable = sys.executable
 
-# CORREÇÃO FINAL: Atualiza o pip para garantir que a flag --no-pep517 seja reconhecida
-print("Atualizando o pip...")
-subprocess.run([python_executable, "-m", "pip", "install", "--upgrade", "pip"], check=True)
-
 # Instalar flash-attn
 print("Instalando flash-attn...")
 subprocess.run(
-    [
-        python_executable,
-        "-m",
-        "pip",
-        "install",
-        "flash-attn",
-        "--no-build-isolation"
-    ],
+    [python_executable, "-m", "pip", "install", "flash-attn", "--no-build-isolation"],
     check=True
 )
 print("✅ Configuração do flash-attn concluída.")
 
 
-# Instalar Apex, forçando o uso do setup.py para compilar as extensões CUDA
-print("Instalando Apex a partir do código-fonte...")
-subprocess.run(
-    [
-        python_executable,
-        "-m",
-        "pip",
-        "install",
-        "--no-build-isolation",
-        "--no-pep517",  # Esta flag requer um pip atualizado
-        "git+https://github.com/NVIDIA/apex.git"
-    ],
-    check=True
-)
+# CORREÇÃO FINAL: Usar um método de instalação robusto para o Apex que força a compilação das extensões CUDA.
+print("Clonando o repositório Apex para compilação manual...")
+subprocess.run("git clone https://github.com/NVIDIA/apex.git", shell=True, check=True)
+
+print("Instalando Apex a partir do código-fonte clonado com flags de compilação explícitas...")
+# Nota: --global-option é obsoleto, mas necessário para o setup.py do Apex.
+apex_install_command = [
+    python_executable, "-m", "pip", "install",
+    "--no-build-isolation",
+    "--no-cache-dir",
+    "--global-option=--cpp_ext",
+    "--global-option=--cuda_ext",
+    "./apex"
+]
+subprocess.run(apex_install_command, check=True)
 print("✅ Configuração do Apex concluída.")