Update app.py

app.py

@@ -4,166 +4,166 @@ import json
 import re
 import datetime
 
-# Configurações de chave secreta e API do Google Gemini
+# Secret key and Google Gemini API configuration
 API_KEY = st.secrets["GOOGLE_API_KEY"]
 
-# Configuração da página
+# Page configuration
 st.set_page_config(page_title="🤖 Gemini2 PowerShell Command Gen", page_icon="🤖", layout="wide")
 
-def enviar_mensagem_para_modelo(mensagem, modelo_nome, temperatura, max_tokens):
-    """Envia uma mensagem para o modelo de IA e retorna a resposta."""
+def send_message_to_model(message, model_name, temperature, max_tokens):
+    """Sends a message to the AI model and returns the response."""
     try:
-        # Configurações do modelo de IA
-        CONFIGURACAO_GERACAO = {
-            "temperature": temperatura,
+        # AI model configurations
+        GENERATION_CONFIG = {
+            "temperature": temperature,
             "top_p": 0.8,
             "top_k": 40,
             "response_mime_type": "text/plain",
             "max_output_tokens": max_tokens,
         }
-        MODELO = genai.GenerativeModel(
-            model_name=modelo_nome,
-            generation_config=CONFIGURACAO_GERACAO,
+        MODEL = genai.GenerativeModel(
+            model_name=model_name,
+            generation_config=GENERATION_CONFIG,
         )
         
-        resposta = MODELO.start_chat(history=[]).send_message(mensagem)
-        return resposta.text
+        response = MODEL.start_chat(history=[]).send_message(message)
+        return response.text
     except Exception as e:
-        st.error(f"❌ Erro ao comunicar com a IA: {e}")
+        st.error(f"❌ Error communicating with the AI: {e}")
         return None
 
-def gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_seguranca, modelo_nome, temperatura, max_tokens, prompt_detail, encoding, add_header, add_error_handling, log_level):
-    """Gera um comando Powershell com base nas configurações do usuário."""
+def generate_powershell_command(prompt_base, detail_level, script_type, security_level, model_name, temperature, max_tokens, prompt_detail, encoding, add_header, add_error_handling, log_level):
+    """Generates a PowerShell command based on user settings."""
     prompt = f"""
-    Você é um especialista em Powershell. Sua tarefa é gerar um único comando Powershell com base na seguinte descrição:
-
-    **Objetivo:** Criar o comando Powershell mais completo, detalhado e eficiente possível, considerando todas as variáveis e cenários.
-
-    **Descrição do Comando:** {prompt_base}
-
-    **Nível de Detalhe:** {nivel_detalhe}
-    **Tipo de Script:** {tipo_script}
-    **Nível de Segurança:** {nivel_seguranca}
-    **Nível de Detalhamento do Prompt**:{prompt_detail}
-
-    **Formato da Resposta:**
-    - Responda em formato Markdown, incluindo um bloco de código Powershell com a formatação original, sem quebrar linhas.
-    - O bloco de código Powershell deve estar delimitado por ```powershell e ``` .
-    - Não inclua comentários, explicações ou qualquer outro texto fora do bloco de código.
-    - O código Powershell deve manter a formatação vertical completa, respeitando indentação e quebras de linha.
-    - O código deve ser realista, usando exemplos, dados e situações do mundo real.
-    - Explore diferentes abordagens, técnicas e práticas avançadas.
-    - Se for gerar um comando com uma cadeia de comandos, não use ponto e virgula no final ou inicio.
-    - Se precisar use pipe "|" para encadear.
-    - Não use nenhum tipo de formatação especial no resultado, apenas o código.
-    - Se a descrição pedir para criar um arquivo, o comando deverá criar o arquivo diretamente no sistema de arquivos e não usar um output em tela para isso.
-    - Se a descrição pedir para ler um arquivo, o comando deverá ler o arquivo diretamente do sistema de arquivos e não usar um input em tela para isso.
-    - Use recursos avançados do PowerShell, como pipelines, variáveis, funções e blocos de script, quando necessário.
-     - O sistema operacional padrão é Windows Server 2016 e a versão do Powershell padrão é a 7, a não ser que o usuário especifique o contrário.
-     - Certifique-se de que o comando gerado seja seguro e siga as melhores práticas do Powershell.
+    You are a Powershell expert. Your task is to generate a single Powershell command based on the following description:
+
+    **Goal:** Create the most complete, detailed, and efficient Powershell command possible, considering all variables and scenarios.
+
+    **Command Description:** {prompt_base}
+
+    **Detail Level:** {detail_level}
+    **Script Type:** {script_type}
+    **Security Level:** {security_level}
+    **Prompt Detail Level**:{prompt_detail}
+
+    **Response Format:**
+    - Respond in Markdown format, including a Powershell code block with its original formatting, without line breaks.
+    - The Powershell code block must be delimited by ```powershell and ```.
+    - Do not include comments, explanations, or any other text outside the code block.
+    - The Powershell code must maintain its full vertical formatting, respecting indentation and line breaks.
+    - The code must be realistic, using real-world examples, data, and situations.
+    - Explore different approaches, techniques, and advanced practices.
+    - If generating a command with a chain of commands, do not use semicolons at the end or beginning.
+    - If necessary, use the pipe "|" to chain commands.
+    - Do not use any special formatting in the result, only the code.
+    - If the description asks to create a file, the command should create the file directly in the file system and not use a screen output for this.
+    - If the description asks to read a file, the command should read the file directly from the file system and not use a screen input for this.
+    - Use advanced PowerShell resources, such as pipelines, variables, functions, and script blocks, when necessary.
+     - The default operating system is Windows Server 2016, and the default Powershell version is 7, unless the user specifies otherwise.
+     - Make sure that the generated command is secure and follows the best Powershell practices.
 
     **Log Level**:{log_level}
-    **Tratamento de Erros**:{add_error_handling}
+    **Error Handling**:{add_error_handling}
 
-     **Importante:**
-    -   Gere apenas um comando por vez.
-    -   Crie o código o mais longo, completo e detalhado possível.
-    -   Considere todos os detalhes do pedido, expandindo a resposta e melhorando o comando.
-     -  Use as informações contextuais (como versão do PowerShell e sistema operacional) para gerar o comando.
-     - Se possível, use um raciocínio incremental para adicionar melhorias, expansões e considerações ao seu código
-     - Use o historico das conversas para que a resposta seja incremental
+     **Important:**
+    - Generate only one command at a time.
+    - Create the longest, most complete, and detailed code possible.
+    - Consider all the details of the request, expanding the response and improving the command.
+     - Use contextual information (such as PowerShell version and operating system) to generate the command.
+     - If possible, use incremental reasoning to add improvements, expansions, and considerations to your code.
+     - Use the history of the conversations so that the response is incremental.
 
     """
-    resposta = enviar_mensagem_para_modelo(prompt, modelo_nome, temperatura, max_tokens)
-    return resposta
+    response = send_message_to_model(prompt, model_name, temperature, max_tokens)
+    return response
 
-def parsear_e_salvar_ps1(codigo_ia, titulo_curto, encoding, add_header):
-    """Parseia o markdown e salva o código Powershell como .ps1."""
-    match = re.search(r'```powershell\s*(.*?)\s*```', codigo_ia, re.DOTALL | re.IGNORECASE)
+def parse_and_save_ps1(ai_code, short_title, encoding, add_header):
+    """Parses the markdown and saves the Powershell code as .ps1."""
+    match = re.search(r'```powershell\s*(.*?)\s*```', ai_code, re.DOTALL | re.IGNORECASE)
     if match:
-        codigo_ps1 = match.group(1).strip()
+        ps1_code = match.group(1).strip()
     else:
-        codigo_ps1 = codigo_ia.strip()
+        ps1_code = ai_code.strip()
 
-    nome_arquivo = f"comando_{titulo_curto}.ps1"
+    file_name = f"command_{short_title}.ps1"
 
     if add_header:
-        data_atual = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        current_date = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         header = f"""
 #===============================================================================
-# Script Gerado pelo Google Gemini 2 PowerShell Command Gen
-# Data: {data_atual}
-# Autor: Elias Andrade
+# Script Generated by Google Gemini 2 PowerShell Command Gen
+# Date: {current_date}
+# Author: Elias Andrade AKA Chaos4455
 #===============================================================================
     """
-        codigo_ps1 = header + codigo_ps1
+        ps1_code = header + ps1_code
 
-    with open(nome_arquivo, "w", encoding=encoding) as f:
-        f.write(codigo_ps1)
-    return nome_arquivo, codigo_ps1
+    with open(file_name, "w", encoding=encoding) as f:
+        f.write(ps1_code)
+    return file_name, ps1_code
 
 def main():
     st.title("🤖 Gemini2 PowerShell Command Gen by [Elias Andrade](https://github.com/chaos4455)")
-    st.markdown("Crie comandos PowerShell de forma fácil e rápida! 🚀")
+    st.markdown("Create PowerShell commands easily and quickly! 🚀")
     st.markdown("---")
 
-    # Layout em colunas (barra lateral e principal)
+    # Layout in columns (sidebar and main area)
     col1, col2 = st.columns([1, 3])
     
     with col1:
-        st.header("⚙️ Configurações")
+        st.header("⚙️ Settings")
         
-        with st.expander("✨ Configurações da IA"):
-            modelo_nome = st.selectbox("🤖 Modelo da IA", ["gemini-2.0-flash-exp", "gemini-1.5-flash-exp"], index=0, help="Escolha o modelo da IA.")
-            temperatura = st.slider("🌡️ Temperatura", min_value=0.1, max_value=1.0, value=0.7, step=0.1, help="Ajuste a criatividade da IA.")
-            max_tokens = st.number_input("📏 Max Tokens", min_value=128, max_value=8192, value=8192, step=128, help="Ajuste o tamanho máximo da resposta.")
+        with st.expander("✨ AI Settings"):
+            model_name = st.selectbox("🤖 AI Model", ["gemini-2.0-flash-exp", "gemini-1.5-flash-exp"], index=0, help="Choose the AI model.")
+            temperature = st.slider("🌡️ Temperature", min_value=0.1, max_value=1.0, value=0.7, step=0.1, help="Adjust the AI's creativity.")
+            max_tokens = st.number_input("📏 Max Tokens", min_value=128, max_value=8192, value=8192, step=128, help="Adjust the maximum size of the response.")
             
-        with st.expander("📝 Configurações do Prompt"):
-            prompt_presets = st.selectbox("🎯 Prompts Predefinidos", ["Nenhum", "Listar arquivos", "Gerenciar processos", "Gerenciar serviços"], index=0, help="Escolha um prompt predefinido.")
-            prompt_detail = st.selectbox("🧐 Detalhe do Prompt", ["Mais descritivo", "Padrão", "Conciso"], index=1, help="Define o nível de detalhe do prompt")
-
-        with st.expander("🛠️ Configurações do PowerShell"):
-            powershell_version = st.selectbox("🎛️ Versão do PowerShell", ["7", "5.1"], index=0, help="Escolha a versão do PowerShell")
-            sistema_operacional = st.selectbox(
-                "💻 Sistema Operacional",
+        with st.expander("📝 Prompt Settings"):
+            prompt_presets = st.selectbox("🎯 Predefined Prompts", ["None", "List files", "Manage processes", "Manage services"], index=0, help="Choose a predefined prompt.")
+            prompt_detail = st.selectbox("🧐 Prompt Detail", ["More descriptive", "Default", "Concise"], index=1, help="Defines the level of detail of the prompt")
+
+        with st.expander("🛠️ PowerShell Settings"):
+            powershell_version = st.selectbox("🎛️ PowerShell Version", ["7", "5.1"], index=0, help="Choose the PowerShell version.")
+            operating_system = st.selectbox(
+                "💻 Operating System",
                 [
                     "Windows Server 2022", "Windows Server 2019", "Windows Server 2016",
                     "Windows Server 2012 R2", "Windows Server 2012", "Windows Server 2008 R2",
-                    "Windows Server 2008", "Windows 11", "Windows 10","Windows 8.1", "Windows 8","Windows 7", "Outro"
+                    "Windows Server 2008", "Windows 11", "Windows 10", "Windows 8.1", "Windows 8", "Windows 7", "Other"
                 ],
-                index=2, help="Escolha o Sistema Operacional de destino."
+                index=2, help="Choose the target operating system."
             )
-            encoding = st.selectbox("🔤 Encoding", ["utf-8", "ansi"], index=0, help="Escolha o encoding do arquivo .ps1")
-            add_header = st.checkbox("📜 Adicionar Cabeçalho", value=True, help="Adicionar um cabeçalho com informações no arquivo .ps1")
-            add_error_handling = st.checkbox("🛡️ Tratamento de Erros", value=True, help="Adicionar tratamento de erros padrão no script")
-            log_level = st.selectbox("🗂️ Nível de Logging", ["Detalhado", "Padrão", "Mínimo"], index=1, help="Define o nível de detalhe de logs.")
+            encoding = st.selectbox("🔤 Encoding", ["utf-8", "ansi"], index=0, help="Choose the encoding of the .ps1 file.")
+            add_header = st.checkbox("📜 Add Header", value=True, help="Add a header with information in the .ps1 file.")
+            add_error_handling = st.checkbox("🛡️ Error Handling", value=True, help="Add standard error handling to the script.")
+            log_level = st.selectbox("🗂️ Logging Level", ["Detailed", "Default", "Minimum"], index=1, help="Defines the detail level of logs.")
             
-            nivel_detalhe = st.selectbox(" Nível de Detalhe", ["Mais detalhado", "Padrão", "Mais conciso"], index=1)
-            tipo_script = st.selectbox(" Tipo de Script", ["Mais automático", "Mais interativo"], index=0)
-            nivel_seguranca = st.radio(" Nível de Segurança", ["Alto", "Médio", "Baixo"], index=1)
+            detail_level = st.selectbox("Detail Level", ["More detailed", "Default", "More concise"], index=1)
+            script_type = st.selectbox("Script Type", ["More automatic", "More interactive"], index=0)
+            security_level = st.radio("Security Level", ["High", "Medium", "Low"], index=1)
 
     with col2:
-         # Prompt base do usuário
-        prompt_base = st.text_input("Descreva o Comando PowerShell:", placeholder="Ex: Liste todos os processos em execução", key="prompt_base")
+         # User's base prompt
+        prompt_base = st.text_input("Describe the PowerShell Command:", placeholder="Ex: List all running processes", key="prompt_base")
 
-        if prompt_presets != "Nenhum":
+        if prompt_presets != "None":
             if prompt_base:
                prompt_base = f"{prompt_presets} , {prompt_base}"
             else:
                prompt_base = prompt_presets;
-        if st.button("✨ Gerar Comando PowerShell"):
+        if st.button("✨ Generate PowerShell Command"):
             if not prompt_base:
-                st.error("⚠️ Por favor, insira uma descrição do comando.")
+                st.error("⚠️ Please enter a command description.")
                 return
 
-            with st.spinner("⏳ Gerando comando..."):
-                codigo_ia = gerar_comando_powershell(
+            with st.spinner("⏳ Generating command..."):
+                ai_code = generate_powershell_command(
                    prompt_base, 
-                   nivel_detalhe, 
-                   tipo_script, 
-                   nivel_seguranca,
-                   modelo_nome,
-                   temperatura,
+                   detail_level, 
+                   script_type, 
+                   security_level,
+                   model_name,
+                   temperature,
                    max_tokens,
                    prompt_detail,
                    encoding,
@@ -172,21 +172,21 @@ def main():
                    log_level
                    )
 
-                if codigo_ia:
-                    st.markdown("### ✅ Comando Gerado:")
-                    st.code(codigo_ia, language="powershell")
+                if ai_code:
+                    st.markdown("### ✅ Generated Command:")
+                    st.code(ai_code, language="powershell")
 
-                    titulo_curto = prompt_base[:30].strip().replace(" ", "_").lower()
-                    nome_arquivo_ps1, codigo_ps1 = parsear_e_salvar_ps1(codigo_ia, titulo_curto, encoding, add_header)
+                    short_title = prompt_base[:30].strip().replace(" ", "_").lower()
+                    file_name_ps1, ps1_code = parse_and_save_ps1(ai_code, short_title, encoding, add_header)
 
                     st.download_button(
-                        label="⬇️ Baixar Comando (.ps1)",
-                        data=codigo_ps1,
-                        file_name=nome_arquivo_ps1,
+                        label="⬇️ Download Command (.ps1)",
+                        data=ps1_code,
+                        file_name=file_name_ps1,
                         mime="application/powershell",
                     )
                 else:
-                    st.error("❌ Erro ao gerar o comando. Verifique a conexão com a IA e tente novamente.")
+                    st.error("❌ Error generating the command. Check the connection with the AI and try again.")
 
 if __name__ == "__main__":
     main()