Update app.py

app.py

@@ -5,23 +5,92 @@ import re
 
 # Configurações de chave secreta e API do Google Gemini
 API_KEY = st.secrets["GOOGLE_API_KEY"]
-genai.configure(api_key=API_KEY)
-
-# Configuração do modelo de IA
-CONFIGURACAO_GERACAO = {
-    "temperature": 0.2,
-    "top_p": 0.8,
-    "top_k": 40,
-    "response_mime_type": "text/plain",
-}
-NOME_MODELO = "gemini-2.0-flash-exp"
-MODELO = genai.GenerativeModel(
-    model_name=NOME_MODELO,
-    generation_config=CONFIGURACAO_GERACAO,
-)
 
 # Estilos CSS para o Streamlit
-st.markdown(
+#  Removido do escopo global
+#  st.markdown( ... )
+
+#  Removido do escopo global
+#  st.set_page_config( ... )
+
+
+def enviar_mensagem_para_modelo(mensagem):
+    """Envia uma mensagem para o modelo de IA e retorna a resposta."""
+    try:
+        #  Removido do escopo global
+        #  MODELO = genai.GenerativeModel(model_name=NOME_MODELO, generation_config=CONFIGURACAO_GERACAO)
+        resposta = MODELO.start_chat(history=[]).send_message(mensagem)
+        return resposta.text
+    except Exception as e:
+        st.error(f"❌ Erro ao comunicar com a IA: {e}")
+        return None
+
+def gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_seguranca):
+    """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:
+    
+    **Descrição do Comando:** {prompt_base}
+    
+    **Nível de Detalhe:** {nivel_detalhe}
+    **Tipo de Script:** {tipo_script}
+    **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.
+    -   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.
+    -   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.
+    
+    **Importante:**
+    -   Gere apenas um comando por vez.
+    """
+    resposta = enviar_mensagem_para_modelo(prompt)
+    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)
+    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
+    
+    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,
+        "top_p": 0.8,
+        "top_k": 40,
+        "response_mime_type": "text/plain",
+    }
+    NOME_MODELO = "gemini-2.0-flash-exp"
+    
+    #  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="🤖")
+
+    
+    # Configuração do modelo de IA
+    global MODELO
+    MODELO = genai.GenerativeModel(
+    model_name=NOME_MODELO,
+    generation_config=CONFIGURACAO_GERACAO,
+    )
+    
+    # Estilos CSS para o Streamlit
+    st.markdown(
     """
     <style>
     body {
@@ -108,63 +177,6 @@ st.markdown(
     """,
     unsafe_allow_html=True,
 )
-
-
-#  Chamada de st.set_page_config fora da função main
-st.set_page_config(page_title="Google Gemini 2 PowerShell Command Gen", page_icon="🤖")
-
-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:
-        st.error(f"❌ Erro ao comunicar com a IA: {e}")
-        return None
-
-def gerar_comando_powershell(prompt_base, nivel_detalhe, tipo_script, nivel_seguranca):
-    """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:
-    
-    **Descrição do Comando:** {prompt_base}
-    
-    **Nível de Detalhe:** {nivel_detalhe}
-    **Tipo de Script:** {tipo_script}
-    **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.
-    -   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.
-    -   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.
-    
-    **Importante:**
-    -   Gere apenas um comando por vez.
-    """
-    resposta = enviar_mensagem_para_modelo(prompt)
-    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)
-    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
-    
-    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():
    
     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!")