Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
 import subprocess
 import tempfile
-import tarfile
 from androguard.misc import AnalyzeAPK
 from transformers import pipeline
 from sentence_transformers import SentenceTransformer, util
@@ -11,23 +10,20 @@ import gradio as gr
 apk_context = {"smali": {}, "java": {}, "info": ""}
 
 def extract_and_set_java():
-    """Extrai o JDK compactado e configura as variáveis de ambiente para o Java."""
-    jdk_tar_path = "jdk-11.0.1.tar.gz"  # Nome do arquivo compactado do JDK
-    extraction_dir = "/tmp/jdk-11"  # Diretório temporário para extração
+    """Configura as variáveis de ambiente para o JDK especificado."""
+    jdk_path = "jdk-11.0.1"  # Nome do diretório do JDK
 
-    if not os.path.exists(extraction_dir):
-        os.makedirs(extraction_dir)
-        with tarfile.open(jdk_tar_path, "r:gz") as tar:
-            tar.extractall(path=extraction_dir)
-        print(f"JDK extraído para {extraction_dir}")
+    if not os.path.exists(jdk_path):
+        raise FileNotFoundError(f"O JDK '{jdk_path}' não foi encontrado. Certifique-se de que ele está presente no caminho correto.")
 
     # Configura a variável de ambiente JAVA_HOME
-    java_home = os.path.join(extraction_dir, "jdk-11.0.1")
+    java_home = os.path.abspath(jdk_path)
     os.environ["JAVA_HOME"] = java_home
     os.environ["PATH"] = f"{os.path.join(java_home, 'bin')}:{os.environ['PATH']}"
+    print(f"JAVA_HOME configurado para {java_home}")
 
 def check_java():
-    """Função para verificar se o Java está configurado e acessível."""
+    """Função para verificar se o Java está instalado e acessível."""
     try:
         result = subprocess.run(["java", "-version"], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         print("Java está disponível:", result.stderr.decode())  # Exibe informações sobre a versão do Java
@@ -37,11 +33,11 @@ def check_java():
         raise EnvironmentError(f"Erro inesperado ao verificar a instalação do Java: {str(e)}")
 
 def install_tools():
-    """Instala ferramentas como Baksmali e JADX."""
+    """Instalar ferramentas como Baksmali e JADX."""
     baksmali_path = "/usr/local/bin/baksmali.jar"
     jadx_path = "/usr/local/bin/jadx/bin/jadx"
 
-    # Configura e verifica a presença do Java
+    # Configura e verifica o Java
     extract_and_set_java()
     check_java()
 
@@ -85,7 +81,7 @@ def decompile_apk(apk_file):
     if apk_file is None:
         return "Nenhum arquivo enviado. Por favor, envie um arquivo APK."
 
-    temp_apk_path = apk_file.name  # Usar o caminho do arquivo diretamente
+    temp_apk_path = apk_file.name
     output_dir = tempfile.mkdtemp()
     try:
         # Decompilar usando o Baksmali