Update app.py

app.py

@@ -2,27 +2,27 @@ import streamlit as st
 import google.generativeai as genai
 import json
 import re
+import datetime
 
 # Configurações de chave secreta e API do Google Gemini
 API_KEY = st.secrets["GOOGLE_API_KEY"]
 
 # Configuração da página
-st.set_page_config(page_title="Google Gemini 2 PowerShell Command Gen", page_icon="🤖", layout="wide")
+st.set_page_config(page_title="🤖 Gemini2 PowerShell Command Gen", page_icon="🤖", layout="wide")
 
-def enviar_mensagem_para_modelo(mensagem):
+def enviar_mensagem_para_modelo(mensagem, modelo_nome, temperatura, max_tokens):
     """Envia uma mensagem para o modelo de IA e retorna a resposta."""
     try:
-        # Configurações do modelo de IA (dentro da função)
+        # Configurações do modelo de IA
         CONFIGURACAO_GERACAO = {
-            "temperature": 0.7,
+            "temperature": temperatura,
             "top_p": 0.8,
             "top_k": 40,
             "response_mime_type": "text/plain",
-            "max_output_tokens": 1024
+            "max_output_tokens": max_tokens,
         }
-        NOME_MODELO = "gemini-2.0-flash-exp"
         MODELO = genai.GenerativeModel(
-            model_name=NOME_MODELO,
+            model_name=modelo_nome,
             generation_config=CONFIGURACAO_GERACAO,
         )
         
@@ -32,7 +32,7 @@ def enviar_mensagem_para_modelo(mensagem):
         st.error(f"❌ Erro ao comunicar com a IA: {e}")
         return None
 
-def gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_seguranca):
+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."""
     prompt = f"""
     Você é um especialista em Powershell. Sua tarefa é gerar um único comando Powershell com base na seguinte descrição:
@@ -42,6 +42,7 @@ def gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_segu
     **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.
@@ -56,13 +57,16 @@ def gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_segu
     -   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.
     
+     **Log Level**:{log_level}
+     **Tratamento de Erros**:{add_error_handling}
+     
     **Importante:**
     -   Gere apenas um comando por vez.
     """
-    resposta = enviar_mensagem_para_modelo(prompt)
+    resposta = enviar_mensagem_para_modelo(prompt, modelo_nome, temperatura, max_tokens)
     return resposta
 
-def parsear_e_salvar_ps1(codigo_ia, titulo_curto):
+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)
     if match:
@@ -71,62 +75,94 @@ def parsear_e_salvar_ps1(codigo_ia, titulo_curto):
         codigo_ps1 = codigo_ia.strip()
 
     nome_arquivo = f"comando_{titulo_curto}.ps1"
-    with open(nome_arquivo, "w", encoding="utf-8") as f:
+
+    if add_header:
+        data_atual = 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
+#===============================================================================
+    """
+        codigo_ps1 = header + codigo_ps1
+
+    with open(nome_arquivo, "w", encoding=encoding) as f:
         f.write(codigo_ps1)
     return nome_arquivo, codigo_ps1
 
 def main():
-   
-    st.title("🤖 Google Gemini 2 PowerShell Command Gen by [Elias Andrade](https://github.com/chaos4455)")
-    st.markdown("Crie comandos PowerShell de forma fácil e rápida!")
+    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("---")
 
     # Layout em colunas (barra lateral e principal)
     col1, col2 = st.columns([1, 3])
-
+    
     with col1:
         st.header("⚙️ Configurações")
-        powershell_version = st.selectbox("Versão do PowerShell", ["7", "5.1"], index=0)
-        sistema_operacional = st.selectbox(
-            "Sistema Operacional",
-            [
-                "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"
-            ],
-            index=2,  # Define Windows Server 2016 como padrão
-        )
+        
+        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("📝 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",
+                [
+                    "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"
+                ],
+                index=2, help="Escolha o Sistema Operacional de destino."
+            )
+            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.")
 
-        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)
     with col2:
-        # Prompt base do usuário
+         # 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")
 
+        if prompt_presets != "Nenhum":
+            if prompt_base:
+               prompt_base = f"{prompt_presets} , {prompt_base}"
+            else:
+               prompt_base = prompt_presets;
         if st.button("✨ Gerar Comando PowerShell"):
             if not prompt_base:
                 st.error("⚠️ Por favor, insira uma descrição do comando.")
                 return
 
             with st.spinner("⏳ Gerando comando..."):
-                codigo_ia = gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_seguranca)
+                codigo_ia = 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
+                   )
+
                 if codigo_ia:
                     st.markdown("### ✅ Comando Gerado:")
                     st.code(codigo_ia, language="powershell")
 
                     titulo_curto = prompt_base[:30].strip().replace(" ", "_").lower()
-                    nome_arquivo_ps1, codigo_ps1 = parsear_e_salvar_ps1(codigo_ia, titulo_curto)
+                    nome_arquivo_ps1, codigo_ps1 = parsear_e_salvar_ps1(codigo_ia, titulo_curto, encoding, add_header)
 
                     st.download_button(
                         label="⬇️ Baixar Comando (.ps1)",