Update app.py

app.py

@@ -1,35 +1,35 @@
 import os
-import tempfile
 import subprocess
+import tempfile
 from androguard.misc import AnalyzeAPK
 from transformers import pipeline
 from sentence_transformers import SentenceTransformer, util
 import gradio as gr
 
-# Global storage for APK data
+# Contexto global para armazenar dados do APK
 apk_context = {"smali": {}, "java": {}, "info": ""}
 
 def check_java():
-    """Function to check if Java is installed and accessible."""
+    """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 is available:", result.stderr.decode())  # Java version information
+        print("Java está disponível:", result.stderr.decode())  # Exibe informações sobre a versão do Java
     except FileNotFoundError:
-        raise EnvironmentError("Java is not installed or not found in the PATH.")
+        raise EnvironmentError("Java não está instalado ou não foi encontrado no PATH.")
     except Exception as e:
-        raise EnvironmentError(f"Unexpected error checking Java installation: {str(e)}")
+        raise EnvironmentError(f"Erro inesperado ao verificar a instalação do Java: {str(e)}")
 
 def install_tools():
-    """Install Baksmali and JADX tools."""
+    """Instalar ferramentas como Baksmali e JADX."""
     baksmali_path = "/usr/local/bin/baksmali.jar"
     jadx_path = "/usr/local/bin/jadx/bin/jadx"
 
-    # Check for Java availability
+    # Verificar se o Java está disponível
     check_java()
 
-    # Install Baksmali if not present
+    # Instalar o Baksmali caso não esteja presente
     if not os.path.exists(baksmali_path):
-        print("Installing Baksmali...")
+        print("Instalando o Baksmali...")
         subprocess.run(
             [
                 "curl",
@@ -41,10 +41,10 @@ def install_tools():
             check=True,
         )
 
-    # Install JADX if not present
+    # Instalar o JADX caso não esteja presente
     jadx_zip_path = "/usr/local/bin/jadx.zip"
     if not os.path.exists(jadx_path):
-        print("Installing JADX...")
+        print("Instalando o JADX...")
         subprocess.run(
             [
                 "curl",
@@ -59,32 +59,32 @@ def install_tools():
         if os.path.exists(jadx_path):
             subprocess.run(["chmod", "+x", jadx_path], check=True)
         else:
-            raise FileNotFoundError("JADX executable not found in the expected path.")
+            raise FileNotFoundError("Executável do JADX não encontrado no caminho esperado.")
 
 install_tools()
 
 def decompile_apk(apk_file):
     if apk_file is None:
-        return "No file uploaded. Please upload an APK file."
-    
-    temp_apk_path = apk_file.name  # Use the file path directly
+        return "Nenhum arquivo enviado. Por favor, envie um arquivo APK."
+
+    temp_apk_path = apk_file.name  # Usar o caminho do arquivo diretamente
     output_dir = tempfile.mkdtemp()
     try:
-        # Decompile using Baksmali
+        # Decompilar usando o Baksmali
         smali_output = os.path.join(output_dir, "smali")
         subprocess.run(
             ["java", "-jar", "/usr/local/bin/baksmali.jar", "d", temp_apk_path, "-o", smali_output], 
             check=True
         )
 
-        # Decompile using JADX
+        # Decompilar usando o JADX
         java_output = os.path.join(output_dir, "java")
         subprocess.run(
             ["/usr/local/bin/jadx/bin/jadx", "-d", java_output, temp_apk_path], 
             check=True
         )
 
-        # Collect decompiled Smali files
+        # Coletar arquivos Smali decompilados
         smali_files = {}
         for root, _, files in os.walk(smali_output):
             for file in files:
@@ -92,7 +92,7 @@ def decompile_apk(apk_file):
                     with open(os.path.join(root, file), "r") as f:
                         smali_files[file] = f.read()
 
-        # Collect decompiled Java files
+        # Coletar arquivos Java decompilados
         java_files = {}
         for root, _, files in os.walk(java_output):
             for file in files:
@@ -100,16 +100,16 @@ def decompile_apk(apk_file):
                     with open(os.path.join(root, file), "r") as f:
                         java_files[file] = f.read()
 
-        # Store results in global context
+        # Armazenar resultados no contexto global
         apk_context["smali"] = smali_files
         apk_context["java"] = java_files
 
-        return f"Decompilation successful. Extracted {len(smali_files)} Smali files and {len(java_files)} Java files."
+        return f"Decompilação bem-sucedida. Extraídos {len(smali_files)} arquivos Smali e {len(java_files)} arquivos Java."
 
     except subprocess.CalledProcessError as e:
-        return f"Error during decompilation: {e.stderr.decode('utf-8') if e.stderr else str(e)}"
+        return f"Erro durante a decompilação: {e.stderr.decode('utf-8') if e.stderr else str(e)}"
     except Exception as e:
-        return f"Error during decompilation: {str(e)}"
+        return f"Erro durante a decompilação: {str(e)}"
 
 def build_search_index():
     smali_texts = [f"{k}\n{v}" for k, v in apk_context["smali"].items()]
@@ -121,7 +121,7 @@ def build_search_index():
 
 def query_apk_chat(user_message):
     if not apk_context["smali"] and not apk_context["java"]:
-        return "No decompiled APK available. Please upload and decompile an APK first."
+        return "Nenhum APK decompilado disponível. Por favor, envie e decompile um APK primeiro."
 
     try:
         model, smali_texts, smali_embeddings, java_texts, java_embeddings = build_search_index()
@@ -130,28 +130,28 @@ def query_apk_chat(user_message):
         java_scores = util.pytorch_cos_sim(query_embedding, java_embeddings).squeeze(0)
         smali_result = smali_texts[smali_scores.argmax().item()]
         java_result = java_texts[java_scores.argmax().item()]
-        response = f"**Relevant Smali Code:**\n\n{smali_result[:1000]}\n\n"
-        response += f"**Relevant Java Code:**\n\n{java_result[:1000]}"
+        response = f"**Código Smali relevante:**\n\n{smali_result[:1000]}\n\n"
+        response += f"**Código Java relevante:**\n\n{java_result[:1000]}"
         return response
 
     except Exception as e:
-        return f"Error during search: {str(e)}"
+        return f"Erro durante a busca: {str(e)}"
 
 apk_upload_interface = gr.Interface(
     fn=decompile_apk,
-    inputs=gr.File(label="Upload APK File", file_types=[".apk"]),
+    inputs=gr.File(label="Enviar arquivo APK", file_types=[".apk"]),
     outputs="text",
-    title="APK Analyzer",
-    description="Upload an APK file to decompile it into Smali and Java source code.",
+    title="Analisador de APK",
+    description="Envie um arquivo APK para decompilá-lo em código Smali e Java.",
 )
 
 chat_interface = gr.Interface(
     fn=query_apk_chat,
-    inputs=gr.Textbox(lines=3, placeholder="Ask a question about the APK code..."),
-    outputs=gr.Textbox(lines=10, label="Response from AI"),
-    title="APK Chatbot",
-    description="Ask questions about the APK's source code in Smali or Java.",
+    inputs=gr.Textbox(lines=3, placeholder="Faça uma pergunta sobre o código do APK..."),
+    outputs=gr.Textbox(lines=10, label="Resposta do AI"),
+    title="Chat com APK",
+    description="Faça perguntas sobre o código-fonte do APK em Smali ou Java.",
 )
 
-iface = gr.TabbedInterface([apk_upload_interface, chat_interface], ["Upload & Analyze", "Chat with AI"])
+iface = gr.TabbedInterface([apk_upload_interface, chat_interface], ["Enviar & Analisar", "Conversar com AI"])
 iface.launch()