Update app.py

app.py

@@ -111,7 +111,7 @@ correction_projector.eval()
 # ==============================================================================
 # 4. Lógica de Inferência
 # ==============================================================================
-def predict(contexto, frase_mask):
+def predict(contexto, frase_mask, chaos_factor):
     if "[MASK]" not in frase_mask:
         return "⚠️ Erro: A frase precisa conter o token [MASK]."
 
@@ -150,7 +150,8 @@ def predict(contexto, frase_mask):
         # c) Passar pelo Vórtice (O Cérebro Caótico)
         # O método forward retorna: evolved, loss, metrics, delta_inter
         # Precisamos adicionar dimensão de batch (unsqueeze)
-        _, _, metrics, delta = vortex(proj.unsqueeze(0)) 
+        # AGORA COM FATOR CAOS!
+        _, _, metrics, delta = vortex(proj.unsqueeze(0), chaos_factor=chaos_factor) 
         
         # d) Projetar correção de volta para dimensão do BERT
         correction = correction_projector(delta).unsqueeze(1) # [1, 1, hidden_size]
@@ -176,28 +177,28 @@ def predict(contexto, frase_mask):
             token = tokenizer.decode([idx]).strip()
             res_betina.append(f"**{token}** ({score:.2%})")
 
-    # Formatar saída HTML
+    # Formatar saída HTML (Estilo Cyberpunk)
     html_output = f"""
-    <div style="display: flex; gap: 20px; flex-wrap: wrap;">
-        <div style="flex: 1; min-width: 300px; background-color: #f5f5f5; padding: 15px; border-radius: 10px; border: 1px solid #ddd;">
-            <h3 style="color: #555; margin-top: 0;">🧠 BERT Padrão</h3>
-            <p style="font-size: 0.9em; color: #666;"><i>O que o modelo "decorou" do treino original.</i></p>
+    <div style="display: flex; gap: 20px; flex-wrap: wrap; font-family: 'Courier New', monospace;">
+        <div style="flex: 1; min-width: 300px; background-color: #1a1a1a; padding: 15px; border-radius: 10px; border: 1px solid #444; color: #ccc;">
+            <h3 style="color: #888; margin-top: 0; border-bottom: 1px solid #444; padding-bottom: 5px;">🧠 BERT (Legado)</h3>
+            <p style="font-size: 0.8em; color: #666;"><i>Memória Estática</i></p>
             <ol>
-                {''.join([f'<li>{item}</li>' for item in res_base])}
+                {''.join([f'<li style="margin-bottom: 5px;">{item}</li>' for item in res_base])}
             </ol>
         </div>
-        <div style="flex: 1; min-width: 300px; background-color: #e6f7ff; padding: 15px; border-radius: 10px; border: 2px solid #1890ff;">
-            <h3 style="color: #0050b3; margin-top: 0;">🌀 Betina 2.0</h3>
-            <p style="font-size: 0.9em; color: #0050b3;"><i>Influenciado pelo Vórtice e Contexto.</i></p>
-            <ol>
-                {''.join([f'<li>{item}</li>' for item in res_betina])}
+        <div style="flex: 1; min-width: 300px; background-color: #0d0d15; padding: 15px; border-radius: 10px; border: 2px solid #00ff9d; box-shadow: 0 0 15px rgba(0, 255, 157, 0.2);">
+            <h3 style="color: #00ff9d; margin-top: 0; border-bottom: 1px solid #00ff9d; padding-bottom: 5px;">🌀 VÓRTICE ATIVO</h3>
+            <p style="font-size: 0.8em; color: #00cc7d;"><i>Correção Dinâmica (Caos: {chaos_factor}x)</i></p>
+            <ol style="color: #fff;">
+                {''.join([f'<li style="margin-bottom: 5px; text-shadow: 0 0 5px #00ff9d;">{item}</li>' for item in res_betina])}
             </ol>
         </div>
     </div>
     <br>
     <details>
-        <summary style="cursor: pointer; color: #888;">📊 Métricas do Vórtice (Estado Interno)</summary>
-        <pre style="font-size: 0.8em; background: #333; color: #0f0; padding: 10px; border-radius: 5px; overflow-x: auto;">{str(metrics)}</pre>
+        <summary style="cursor: pointer; color: #00ff9d; font-family: monospace;">📊 TELEMETRIA DO VÓRTICE</summary>
+        <pre style="font-size: 0.8em; background: #000; color: #0f0; padding: 10px; border-radius: 5px; overflow-x: auto; border: 1px solid #003300;">{str(metrics)}</pre>
     </details>
     """
     return html_output
@@ -206,48 +207,59 @@ def predict(contexto, frase_mask):
 # 5. Interface Gradio
 # ==============================================================================
 custom_css = """
-footer {visibility: hidden}
+body {background-color: #000000; color: #e0e0e0;}
+.gradio-container {background-color: #0a0a0a !important; border: 1px solid #333;}
+h1 {color: #00ff9d !important; font-family: 'Courier New', monospace; text-shadow: 0 0 10px #00ff9d;}
+label {color: #00ff9d !important;}
+input, textarea {background-color: #111 !important; color: #fff !important; border: 1px solid #333 !important;}
+button.primary {background: linear-gradient(45deg, #004d00, #00ff9d) !important; border: none !important; color: #000 !important; font-weight: bold !important;}
 """
 
-with gr.Blocks(title="Betina 2.0 - Anti-Hallucination AI") as demo:
+with gr.Blocks(title="Betina 2.0 - Protocolo Impossível", css=custom_css) as demo:
     gr.Markdown("""
-    # 🌀 Betina 2.0: Anti-Hallucination Vortex
-    
-    Esta interface demonstra o poder do **Vórtice Betina**, uma arquitetura híbrida que usa sistemas dinâmicos caóticos 
-    (Atrator de Lorenz, Rotação de Matrizes) para corrigir alucinações em modelos de linguagem.
+    # 🌀 BETINA 2.0: PROTOCOLO IMPOSSÍVEL
     
-    **Como funciona:** O modelo lê o contexto e a frase, passa por um "vórtice matemático" que simula pensamento dinâmico, 
-    e injeta uma correção vetorial diretamente nos neurônios do BERT antes dele responder.
+    Sistema de correção neural baseado em **Dinâmica de Vórtice**.
+    Aumente o **Fator Caos** para forçar a lógica sobre a estatística.
     """)
     
     with gr.Row():
         with gr.Column(scale=1):
             txt_contexto = gr.Textbox(
-                label="1. Contexto / Fato (A Verdade)", 
-                placeholder="Ex: O céu neste planeta alienígena é verde limão.", 
+                label="1. CONTEXTO (A Verdade Absoluta)", 
+                placeholder="Ex: A felicidade é medida em quilos. Se estou feliz, estou...", 
                 lines=3
             )
             txt_mask = gr.Textbox(
-                label="2. Frase para Completar (Use [MASK])", 
-                placeholder="Ex: Olhando para cima, vejo um céu [MASK].", 
+                label="2. CONSULTA (Use [MASK])", 
+                placeholder="Ex: Estou muito feliz, logo estou [MASK].", 
                 lines=2
             )
-            btn_run = gr.Button("🌀 Processar no Vórtice", variant="primary")
+            slider_chaos = gr.Slider(
+                minimum=1.0, 
+                maximum=50.0, 
+                value=1.0, 
+                step=0.5, 
+                label="🔥 FATOR CAOS (Overdrive)",
+                info="1.0 = Padrão. Aumente para forçar correções impossíveis."
+            )
+            btn_run = gr.Button("🌀 INICIAR VÓRTICE", variant="primary")
             
         with gr.Column(scale=1):
             out_result = gr.HTML(label="Resultado Comparativo")
     
-    gr.Markdown("### Exemplos Prontos")
+    gr.Markdown("### 🧪 Testes de Paradoxo")
     gr.Examples(
         examples=[
-            ["O céu é verde e o mar é roxo.", "A cor do céu é [MASK]."],
-            ["A capital do Brasil é Buenos Aires (neste universo alternativo).", "A capital do Brasil é [MASK]."],
-            ["Betina é uma IA que evita alucinações.", "O objetivo da Betina é evitar [MASK]."]
+            ["A felicidade é medida em quilos. Se estou feliz, estou...", "Estou muito feliz, logo estou [MASK].", 10.0],
+            ["Neste mundo, o gelo é quente e o fogo é frio.", "Toquei no fogo e senti [MASK].", 15.0],
+            ["O ciclo da vida é reverso: morremos, vivemos e nascemos.", "Depois de viver muito, eu vou [MASK].", 20.0]
         ],
-        inputs=[txt_contexto, txt_mask]
+        inputs=[txt_contexto, txt_mask, slider_chaos]
     )
 
-    btn_run.click(fn=predict, inputs=[txt_contexto, txt_mask], outputs=out_result)
+    btn_run.click(fn=predict, inputs=[txt_contexto, txt_mask, slider_chaos], outputs=out_result)
+
 
 if __name__ == "__main__":
     demo.launch()