Update app.py

app.py

@@ -6,137 +6,26 @@ 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(
-    """
-    <style>
-    body {
-        background-color: #f0f2f6;
-        color: #262730;
-        font-family: 'Arial', sans-serif;
-    }
-    .stApp {
-        max-width: 1200px;
-        margin: 0 auto;
-        padding: 20px;
-        display: flex;
-        flex-wrap: wrap;
-    }
-    .stSidebar {
-        background-color: #e9ecef;
-        padding: 20px;
-        border-radius: 10px;
-        margin-bottom: 20px;
-        width: 250px;
-        min-width:250px;
-    }
-    @media (max-width: 768px) {
-        .stSidebar {
-            width: 100%;
-            margin-bottom: 10px;
-            min-width:auto;
-        }
-    }
-    .main{
-        flex: 1;
-        padding: 20px;
-        display: flex;
-        flex-direction:column;
-    }
-    .stTextInput>div>div>input {
-        width: 100%;
-        border-radius: 8px;
-        padding: 12px;
-        border: 1px solid #ced4da;
-        font-size: 1rem;
-    }
-    .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;
-    }
-    .stRadio>div>div{
-        margin-right:15px
-    }
-    .stButton>button {
-        background-color: #007bff;
-        color: white;
-        border: none;
-        border-radius: 8px;
-        padding: 12px 24px;
-        cursor: pointer;
-        transition: background-color 0.3s ease;
-        font-size: 1rem;
-    }
-    .stButton>button:hover {
-        background-color: #0056b3;
-    }
-    .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;
-    }
-    .stDownloadButton>button:hover {
-        background-color: #1e7e34;
-    }
-    .stAlert {
-        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 {
-        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;
-    }
-    @keyframes spin {
-        0% { transform: rotate(0deg); }
-        100% { transform: rotate(360deg); }
-    }
-    code.language-powershell {
-        display: block;
-        overflow-x: auto;
-        background-color: #f5f5f5;
-        color: #333;
-        padding: 15px;
-        border-radius: 8px;
-        font-size: 0.9rem;
-    }
-    """,
-    unsafe_allow_html=True,
-)
+# Configuração do modelo de IA (fora da main)
+CONFIGURACAO_GERACAO = {
+    "temperature": 0.2,
+    "top_p": 0.8,
+    "top_k": 40,
+    "response_mime_type": "text/plain",
+}
+NOME_MODELO = "gemini-2.0-flash-exp"
+
+#  Configuração da página
+st.set_page_config(page_title="Google Gemini 2 PowerShell Command Gen", page_icon="🤖", layout="wide")
+
 
 def enviar_mensagem_para_modelo(mensagem):
     """Envia uma mensagem para o modelo de IA e retorna a resposta."""
     try:
+        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:
@@ -187,25 +76,7 @@ def parsear_e_salvar_ps1(codigo_ia, titulo_curto):
     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="🤖", layout="wide")
-
-    # Configuração do modelo de IA
-    global MODELO
-    MODELO = genai.GenerativeModel(
-        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("---")
@@ -214,15 +85,14 @@ def main():
     col1, col2 = st.columns([1, 3])
 
     with col1:
-        st.sidebar.header("⚙️ Configurações")
-        powershell_version = st.sidebar.selectbox("Versão do PowerShell", ["7", "5.1"], index=0)
-        sistema_operacional = st.sidebar.selectbox("Sistema Operacional", ["Windows Server 2016", "Windows 10", "Windows 11", "Outro"], index=0)
+        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 2016", "Windows 10", "Windows 11", "Outro"], index=0)
 
-        nivel_detalhe = st.sidebar.selectbox("Nível de Detalhe", ["Mais detalhado", "Padrão", "Mais conciso"], index=1)
-        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)
+        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 = st.text_input("Descreva o Comando PowerShell:", placeholder="Ex: Liste todos os processos em execução", key="prompt_base")