Update app.py

app.py

@@ -15,23 +15,38 @@ def executar_script(script):
         saida = resultado.stdout + "\n" + resultado.stderr
 
         if resultado.returncode != 0:
-            return f"ERRO AO EXECUTAR {script}\n\n{saida}"
+            return (
+                f"❌ ERRO AO EXECUTAR {script}\n\n"
+                + saida
+            )
 
         return saida
 
     except subprocess.TimeoutExpired:
-        return "O processo ultrapassou o limite de 30 minutos."
+        return (
+            "⏰ O processo ultrapassou o limite de 30 minutos."
+        )
 
     except Exception as e:
-        return f"ERRO INESPERADO:\n\n{e}"
+        return f"❌ ERRO INESPERADO:\n\n{e}"
 
 
 def explorar_corpus():
-    return executar_script("src/01_explore_corpus.py")
+    return executar_script(
+        "src/01_explore_corpus.py"
+    )
 
 
 def preparar_dataset():
-    return executar_script("src/02_prepare_dataset.py")
+    return executar_script(
+        "src/02_prepare_dataset.py"
+    )
+
+
+def treinar_modelo():
+    return executar_script(
+        "src/03_train.py"
+    )
 
 
 with gr.Blocks(title="SEEDS Lab") as demo:
@@ -39,19 +54,25 @@ with gr.Blocks(title="SEEDS Lab") as demo:
     gr.Markdown("""
     # 🌱 SEEDS Lab
 
-    Pipeline de preparação do corpus
-    **English → Macushi**
+    ## Pipeline de tradução English → Macushi
+
+    Este espaço executa as etapas do projeto SEEDS.
     """)
 
+    gr.Markdown("### 🔬 Etapas do projeto")
+
     with gr.Row():
 
         botao_explorar = gr.Button(
-            "🔎 Explorar corpus",
-            variant="secondary"
+            "🔎 Explorar corpus"
         )
 
         botao_preparar = gr.Button(
-            "📦 Preparar dataset",
+            "📦 Preparar dataset"
+        )
+
+        botao_treinar = gr.Button(
+            "🧠 Treinar modelo",
             variant="primary"
         )
 
@@ -73,5 +94,11 @@ with gr.Blocks(title="SEEDS Lab") as demo:
         outputs=resultado
     )
 
+    botao_treinar.click(
+        fn=treinar_modelo,
+        inputs=[],
+        outputs=resultado
+    )
+
 
 demo.launch()