Update app.py

app.py

@@ -132,63 +132,169 @@ def executar_equipe_interface(disciplina, assunto, topicos_str):
     
     return saida_final, pdf_path
 
-# Main application function with both login and main interfaces
 def main_app():
-    with gr.Blocks(theme="soft") as app:
+    # Custom CSS for better styling
+    custom_css = """
+    .main-container {
+        background-color: #f8f9fa;
+        border-radius: 10px;
+        padding: 20px;
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+        margin: 10px;
+    }
+    .header {
+        text-align: center;
+        margin-bottom: 20px;
+    }
+    .header h1 {
+        color: #4a4a4a;
+        font-weight: bold;
+    }
+    .login-container {
+        max-width: 500px;
+        margin: 0 auto;
+        padding: 30px;
+        background-color: white;
+        border-radius: 8px;
+        box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
+    }
+    .input-group {
+        margin-bottom: 15px;
+    }
+    .btn-primary {
+        background-color: #3498db;
+        color: white;
+        border: none;
+        padding: 10px 20px;
+        border-radius: 5px;
+        cursor: pointer;
+        transition: background-color 0.3s;
+    }
+    .btn-primary:hover {
+        background-color: #2980b9;
+    }
+    .user-info {
+        display: flex;
+        justify-content: space-between;
+        align-items: center;
+        margin-bottom: 20px;
+        padding: 10px;
+        background-color: #e8f4fd;
+        border-radius: 5px;
+    }
+    .form-card {
+        padding: 20px;
+        background-color: white;
+        border-radius: 8px;
+        margin-bottom: 20px;
+        box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
+    }
+    .tab-content {
+        padding: 20px;
+        background-color: white;
+        border-radius: 0 0 8px 8px;
+        min-height: 400px;
+    }
+    /* Hide Gradio footer */
+    footer {
+        display: none !important;
+    }
+    """
+    
+    with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as app:
         # Authentication state
         is_logged_in = gr.State(False)
         current_user = gr.State("")
 
         # Login Interface
         with gr.Group(visible=True) as login_block:
-            gr.Markdown("# Sistema de Material de Estudo")
-            gr.Markdown("Por favor, faça login para continuar")
-            username_input = gr.Textbox(label="Usuário")
-            password_input = gr.Textbox(label="Senha", type="password")
-            login_button = gr.Button("Login")
-            login_message = gr.Textbox(label="Mensagem", interactive=False)
-            
-            # Login credentials info (for demo purposes)
-            gr.Markdown("""
-            ### Credenciais para teste:
-            - Usuário: admin, Senha: admin
-            - Usuário: user, Senha: user123
-            - Usuário: test, Senha: test
-            """)
+            with gr.Column(elem_classes=["main-container"]):
+                with gr.Row(elem_classes=["header"]):
+                    gr.Markdown("# Sistema de Material de Estudo")
+                    gr.Markdown("### Assistente Inteligente para Geração de Material Educacional")
+                
+                with gr.Column(elem_classes=["login-container"]):
+                    gr.Markdown("### Login")
+                    with gr.Group():
+                        username_input = gr.Textbox(label="Usuário", placeholder="Digite seu nome de usuário")
+                        password_input = gr.Textbox(label="Senha", type="password", placeholder="Digite sua senha")
+                        login_message = gr.Textbox(label="", interactive=False, visible=False)
+                        login_button = gr.Button("Entrar", variant="primary")
+                    
+                    with gr.Accordion("Credenciais para teste", open=False):
+                        gr.Markdown("""
+                        - Usuário: admin, Senha: admin
+                        - Usuário: user, Senha: user123
+                        - Usuário: test, Senha: test
+                        """)
 
         # Main Application Interface
         with gr.Group(visible=False) as main_block:
-            gr.Markdown("# Gerador de Material de Estudos")
-            user_info = gr.Markdown("Bem-vindo!")
-            logout_button = gr.Button("Logout")
-            
-            with gr.Row():
-                disciplina_input = gr.Textbox(label="Disciplina", value="Matemática")
-                assunto_input = gr.Textbox(label="Assunto", value="Funções")
-                topicos_input = gr.Textbox(label="Tópicos (separados por vírgula)", 
-                                           value="Função quadrática, Função exponencial, Função logarítmica")
-            
-            generate_button = gr.Button("Gerar Material de Estudo")
-            
-            with gr.Row():
-                markdown_output = gr.Markdown(label="Material Completo")
-                pdf_output = gr.File(label="Download PDF")
+            with gr.Column(elem_classes=["main-container"]):
+                with gr.Row(elem_classes=["user-info"]):
+                    user_display = gr.Markdown("Bem-vindo!")
+                    logout_button = gr.Button("Sair", size="sm")
+                
+                gr.Markdown("# Gerador Inteligente de Material de Estudos")
+                gr.Markdown("### Personalize seu material educacional completo em minutos")
+                
+                with gr.Column(elem_classes=["form-card"]):
+                    gr.Markdown("### Dados do Material")
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            disciplina_input = gr.Textbox(
+                                label="Disciplina", 
+                                placeholder="Ex: Matemática, Física, História...",
+                                value="Matemática"
+                            )
+                        with gr.Column(scale=1):
+                            assunto_input = gr.Textbox(
+                                label="Assunto", 
+                                placeholder="Ex: Funções, Leis de Newton, Segunda Guerra Mundial...",
+                                value="Funções"
+                            )
+                    
+                    topicos_input = gr.Textbox(
+                        label="Tópicos (separados por vírgula)", 
+                        placeholder="Ex: Função quadrática, Função exponencial, Função logarítmica...",
+                        value="Função quadrática, Função exponencial, Função logarítmica",
+                        lines=2
+                    )
+                
+                with gr.Row():
+                    generate_button = gr.Button("Gerar Material de Estudo", variant="primary", size="lg")
+                
+                # Progress indicator during generation
+                with gr.Group(visible=False) as progress_group:
+                    progress_bar = gr.Progress()
+                    progress_text = gr.Markdown("Gerando material, por favor aguarde...")
+                
+                # Output tabs
+                with gr.Tabs() as tabs:
+                    with gr.TabItem("Visualizar Material"):
+                        markdown_output = gr.Markdown(label="Material Completo")
+                    
+                    with gr.TabItem("Download"):
+                        with gr.Column():
+                            gr.Markdown("### Download do Material")
+                            pdf_output = gr.File(label="Download PDF")
+                            gr.Markdown("Clique no botão acima para baixar seu material em formato PDF.")
 
-        # Login function
+        # Login function with improved feedback
         def login_fn(username, password):
             success, message = login(username, password)
             if success:
                 return {
                     login_block: gr.update(visible=False),
                     main_block: gr.update(visible=True),
-                    login_message: message,
-                    user_info: f"### Usuário: {username}",
+                    login_message: gr.update(value=message, visible=False),
+                    user_display: f"**Usuário:** {username}",
                     is_logged_in: True,
                     current_user: username
                 }
             else:
                 return {
-                    login_message: message,
+                    login_message: gr.update(value=message, visible=True),
                     is_logged_in: False
                 }
 
@@ -199,21 +305,48 @@ def main_app():
                 main_block: gr.update(visible=False),
                 username_input: "",
                 password_input: "",
-                login_message: "",
+                login_message: gr.update(value="", visible=False),
                 is_logged_in: False,
                 current_user: ""
             }
         
-        # Generate material function
-        def generate_material(disciplina, assunto, topicos):
+        # Generate material function with progress tracking
+        def generate_material(disciplina, assunto, topicos, progress=gr.Progress()):
+            # Show progress during generation
+            progress(0, desc="Iniciando geração do material...")
+            
+            # First phase - motivational content
+            progress(0.25, desc="Gerando conteúdo motivacional...")
+            
+            # Second phase - study plan
+            progress(0.5, desc="Elaborando plano de estudos...")
+            
+            # Third phase - gathering video resources
+            progress(0.75, desc="Pesquisando recursos de vídeo...")
+            
+            # Execute the actual material generation
             markdown, pdf = executar_equipe_interface(disciplina, assunto, topicos)
-            return markdown, pdf
+            
+            # Complete
+            progress(1.0, desc="Material gerado com sucesso!")
+            
+            return {
+                markdown_output: markdown,
+                pdf_output: pdf,
+                progress_group: gr.update(visible=False)
+            }
+
+        # Show progress during generation
+        def show_progress():
+            return {
+                progress_group: gr.update(visible=True)
+            }
 
         # Connect buttons to functions
         login_button.click(
             login_fn,
             inputs=[username_input, password_input],
-            outputs=[login_block, main_block, login_message, user_info, is_logged_in, current_user]
+            outputs=[login_block, main_block, login_message, user_display, is_logged_in, current_user]
         )
         
         logout_button.click(
@@ -223,9 +356,12 @@ def main_app():
         )
         
         generate_button.click(
+            show_progress,
+            outputs=[progress_group]
+        ).then(
             generate_material,
             inputs=[disciplina_input, assunto_input, topicos_input],
-            outputs=[markdown_output, pdf_output]
+            outputs=[markdown_output, pdf_output, progress_group]
         )
 
     return app