Update app.py

app.py

@@ -6,7 +6,6 @@ import re
 # Configurações de chave secreta e API do Google Gemini
 API_KEY = st.secrets["GOOGLE_API_KEY"]
 
-
 # Estilos CSS para o Streamlit
 st.markdown(
     """
@@ -14,61 +13,54 @@ st.markdown(
     body {
         background-color: #f0f2f6;
         color: #262730;
-        font-family: 'Arial', sans-serif; /* Fonte padrão */
+        font-family: 'Arial', sans-serif;
     }
     .stApp {
         max-width: 1200px;
         margin: 0 auto;
         padding: 20px;
         display: flex;
-        flex-wrap: wrap; /* Quebra em telas menores */
-        
+        flex-wrap: wrap;
     }
-    
-    /* Barra Lateral */
     .stSidebar {
         background-color: #e9ecef;
         padding: 20px;
         border-radius: 10px;
         margin-bottom: 20px;
-        width: 250px; /* Largura da barra lateral */
-         min-width:250px;
-        
+        width: 250px;
+        min-width:250px;
     }
-     @media (max-width: 768px) {
+    @media (max-width: 768px) {
         .stSidebar {
             width: 100%;
             margin-bottom: 10px;
-             min-width:auto;
+            min-width:auto;
         }
     }
-    /* Conteúdo Principal */
-     .main{
-         flex: 1;
-          padding: 20px; /* Espaço interno */
-          display: flex;
-          flex-direction:column; /* empilhamento do conteudo */
-         
-     }
+    .main{
+        flex: 1;
+        padding: 20px;
+        display: flex;
+        flex-direction:column;
+    }
     .stTextInput>div>div>input {
-         width: 100%;
+        width: 100%;
         border-radius: 8px;
         padding: 12px;
         border: 1px solid #ced4da;
         font-size: 1rem;
     }
-     .stSelectbox>div>div>div {
+    .stSelectbox>div>div>div {
         border-radius: 8px;
         padding: 10px;
         border: 1px solid #ced4da;
         font-size: 1rem;
         margin-bottom: 10px;
-        
     }
     .stRadio>div{
         display:flex;
         flex-direction: row;
-          margin-bottom: 10px;
+        margin-bottom: 10px;
     }
     .stRadio>div>div{
         margin-right:15px
@@ -86,54 +78,52 @@ st.markdown(
     .stButton>button:hover {
         background-color: #0056b3;
     }
-      .stDownloadButton>button{
-       background-color: #28a745;
+    .stDownloadButton>button{
+        background-color: #28a745;
         color: white;
         border: none;
         border-radius: 8px;
         padding: 12px 24px;
         cursor: pointer;
         transition: background-color 0.3s ease;
-         font-size: 1rem;
+        font-size: 1rem;
     }
     .stDownloadButton>button:hover {
         background-color: #1e7e34;
     }
-
     .stAlert {
-    padding: 15px;
-    margin-bottom: 20px;
-    border: 1px solid transparent;
-    border-radius: 4px;
-     font-size: 1rem;
+        padding: 15px;
+        margin-bottom: 20px;
+        border: 1px solid transparent;
+        border-radius: 4px;
+        font-size: 1rem;
     }
     .stAlert.stAlert-danger {
         color: #721c24;
         background-color: #f8d7da;
         border-color: #f5c6cb;
     }
-     .stSpinner {
+    .stSpinner {
         display: flex;
         justify-content: center;
         align-items: center;
         margin-top: 20px;
     }
     .stSpinner div {
-       border: 8px solid #f3f3f3;
-       border-top: 8px solid #3498db;
-       border-radius: 50%;
-       width: 60px;
-       height: 60px;
-       animation: spin 2s linear infinite;
+        border: 8px solid #f3f3f3;
+        border-top: 8px solid #3498db;
+        border-radius: 50%;
+        width: 60px;
+        height: 60px;
+        animation: spin 2s linear infinite;
     }
-
     @keyframes spin {
         0% { transform: rotate(0deg); }
         100% { transform: rotate(360deg); }
     }
-   code.language-powershell {
-         display: block; /* Para ocupar toda a largura */
-        overflow-x: auto; /* Barra de rolagem horizontal se necessário */
+    code.language-powershell {
+        display: block;
+        overflow-x: auto;
         background-color: #f5f5f5;
         color: #333;
         padding: 15px;
@@ -144,11 +134,9 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
-
 def enviar_mensagem_para_modelo(mensagem):
     """Envia uma mensagem para o modelo de IA e retorna a resposta."""
     try:
-       
         resposta = MODELO.start_chat(history=[]).send_message(mensagem)
         return resposta.text
     except Exception as e:
@@ -167,8 +155,10 @@ def gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_segu
     **Nível de Segurança:** {nivel_seguranca}
     
     **Formato da resposta:**
-    -   Responda apenas o código do comando Powershell, sem texto adicional.
-    -   Não inclua comentários, explicações ou markdown.
+    -   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.
     -   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.
@@ -184,21 +174,19 @@ def gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_segu
     return resposta
 
 def parsear_e_salvar_ps1(codigo_ia, titulo_curto):
-    """Parseia o código retornado pela IA para extrair o bloco de comando ps1 e salva como .ps1."""
-    # Regex para encontrar blocos de código Powershell (ignorando case)
+    """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:
         codigo_ps1 = match.group(1).strip()
     else:
-        codigo_ps1 = codigo_ia.strip() # Se não houver bloco, usa o código inteiro
-    
+        codigo_ps1 = codigo_ia.strip()
+
     nome_arquivo = f"comando_{titulo_curto}.ps1"
     with open(nome_arquivo, "w", encoding="utf-8") as f:
         f.write(codigo_ps1)
     return nome_arquivo, codigo_ps1
 
 def main():
-   
     # Configuração do modelo de IA
     CONFIGURACAO_GERACAO = {
         "temperature": 0.2,
@@ -207,24 +195,21 @@ def main():
         "response_mime_type": "text/plain",
     }
     NOME_MODELO = "gemini-2.0-flash-exp"
-    
-    #  Move st.set_page_config para o começo da função
+
+    # Move st.set_page_config para o começo da função
     st.set_page_config(page_title="Google Gemini 2 PowerShell Command Gen", page_icon="🤖", layout="wide")
 
-    
     # Configuração do modelo de IA
     global MODELO
     MODELO = genai.GenerativeModel(
-    model_name=NOME_MODELO,
-    generation_config=CONFIGURACAO_GERACAO,
+        model_name=NOME_MODELO,
+        generation_config=CONFIGURACAO_GERACAO,
     )
 
-    
-   
     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.markdown("---")
-    
+
     # Layout em colunas (barra lateral e principal)
     col1, col2 = st.columns([1, 3])
 
@@ -237,8 +222,8 @@ def main():
         tipo_script = st.sidebar.selectbox("Tipo de Script", ["Mais automático", "Mais interativo"], index=0)
         nivel_seguranca = st.sidebar.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 st.button("✨ Gerar Comando PowerShell"):
@@ -249,13 +234,12 @@ def main():
             with st.spinner("⏳ Gerando comando..."):
                 codigo_ia = gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_seguranca)
                 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)
+
                     st.download_button(
                         label="⬇️ Baixar Comando (.ps1)",
                         data=codigo_ps1,
@@ -263,7 +247,7 @@ def main():
                         mime="application/powershell",
                     )
                 else:
-                     st.error("❌ Erro ao gerar o comando. Verifique a conexão com a IA e tente novamente.")
-                     
+                    st.error("❌ Erro ao gerar o comando. Verifique a conexão com a IA e tente novamente.")
+
 if __name__ == "__main__":
     main()