Correção do Processamento do Merge

app.py

@@ -6,7 +6,7 @@ import json
 import time
 import os
 import uuid
-import threading # Necessário para o processamento em paralelo
+import threading
 
 # Importações do LangChain
 from langchain.prompts import PromptTemplate
@@ -52,11 +52,9 @@ def process():
             mock_text = form_data.get('mock_text', 'Este é um texto de simulação.')
             mock_html = markdown2.markdown(mock_text, extras=["fenced-code-blocks", "tables"])
             yield f"data: {json.dumps({'progress': 100, 'message': 'Simulação concluída!', 'partial_result': {'id': 'grok-output', 'content': mock_html}, 'done': True, 'mode': 'atomic' if processing_mode == 'atomic' else 'hierarchical'})}\n\n"
-            # Simula o preenchimento de todas as caixas no modo teste
             if processing_mode == 'atomic':
                 yield f"data: {json.dumps({'partial_result': {'id': 'sonnet-output', 'content': mock_html}})}\n\n"
                 yield f"data: {json.dumps({'partial_result': {'id': 'gemini-output', 'content': mock_html}})}\n\n"
-
         else:
             if not solicitacao_usuario:
                 yield f"data: {json.dumps({'error': 'Solicitação não fornecida.'})}\n\n"
@@ -67,48 +65,31 @@ def process():
                 rag_context = get_relevant_context(file_paths, solicitacao_usuario)
                 
                 if processing_mode == 'atomic':
-                    # --- LÓGICA ATÔMICA (PARALELA) CORRIGIDA ---
                     results = {}
                     threads = []
-
                     def run_chain(chain, inputs, key):
-                        try:
-                            results[key] = chain.invoke(inputs)['text']
-                        except Exception as e:
-                            results[key] = f"Erro ao processar {key}: {e}"
+                        try: results[key] = chain.invoke(inputs)['text']
+                        except Exception as e: results[key] = f"Erro ao processar {key}: {e}"
 
                     models = {'grok': grok_llm, 'sonnet': claude_llm, 'gemini': gemini_llm}
                     prompt = PromptTemplate(template=PROMPT_ATOMICO_INICIAL, input_variables=["solicitacao_usuario", "rag_context"])
-                    
                     yield f"data: {json.dumps({'progress': 15, 'message': 'Iniciando processamento paralelo...'})}\n\n"
-
-                    # 1. Inicia todas as threads
                     for name, llm in models.items():
                         chain = LLMChain(llm=llm, prompt=prompt)
                         thread = threading.Thread(target=run_chain, args=(chain, {"solicitacao_usuario": solicitacao_usuario, "rag_context": rag_context}, name))
                         threads.append(thread)
                         thread.start()
-
-                    # 2. Aguarda a conclusão de todas as threads
                     for thread in threads:
                         thread.join()
-                    
                     yield f"data: {json.dumps({'progress': 80, 'message': 'Todos os modelos responderam. Formatando saída...'})}\n\n"
-
-                    # 3. Envia todos os resultados de uma vez, agora que estão prontos
                     grok_html = markdown2.markdown(results.get('grok', 'Falha ao obter resposta.'), extras=["fenced-code-blocks", "tables"])
                     yield f"data: {json.dumps({'partial_result': {'id': 'grok-output', 'content': grok_html}})}\n\n"
-                    
                     sonnet_html = markdown2.markdown(results.get('sonnet', 'Falha ao obter resposta.'), extras=["fenced-code-blocks", "tables"])
                     yield f"data: {json.dumps({'partial_result': {'id': 'sonnet-output', 'content': sonnet_html}})}\n\n"
-
                     gemini_html = markdown2.markdown(results.get('gemini', 'Falha ao obter resposta.'), extras=["fenced-code-blocks", "tables"])
                     yield f"data: {json.dumps({'partial_result': {'id': 'gemini-output', 'content': gemini_html}})}\n\n"
-                    
                     yield f"data: {json.dumps({'progress': 100, 'message': 'Processamento Atômico concluído!', 'done': True, 'mode': 'atomic'})}\n\n"
-
                 else:
-                    # --- LÓGICA HIERÁRQUICA (SEQUENCIAL) ---
                     yield f"data: {json.dumps({'progress': 15, 'message': 'O GROK está processando sua solicitação com os arquivos...'})}\n\n"
                     prompt_grok = PromptTemplate(template=PROMPT_HIERARQUICO_GROK, input_variables=["solicitacao_usuario", "rag_context"])
                     chain_grok = LLMChain(llm=grok_llm, prompt=prompt_grok)
@@ -131,39 +112,47 @@ def process():
                     if not resposta_gemini or not resposta_gemini.strip(): raise ValueError("Falha no serviço Gemini: Sem resposta.")
                     gemini_html = markdown2.markdown(resposta_gemini, extras=["fenced-code-blocks", "tables"])
                     yield f"data: {json.dumps({'progress': 100, 'message': 'Processamento concluído!', 'partial_result': {'id': 'gemini-output', 'content': gemini_html}, 'done': True, 'mode': 'hierarchical'})}\n\n"
-
             except Exception as e:
                 print(f"Ocorreu um erro durante o processamento: {e}")
                 yield f"data: {json.dumps({'error': f'Ocorreu um erro inesperado na aplicação: {e}'})}\n\n"
 
     return Response(generate_stream(mode, form_data, temp_file_paths), mimetype='text/event-stream')
 
-# --- ROTA PARA O MERGE ---
+# --- ROTA DE MERGE ATUALIZADA PARA STREAMING ---
 @app.route('/merge', methods=['POST'])
 def merge():
     data = request.get_json()
-    try:
-        prompt_merge = PromptTemplate(template=PROMPT_ATOMICO_MERGE, input_variables=["solicitacao_usuario", "texto_para_analise_grok", "texto_para_analise_sonnet", "texto_para_analise_gemini"])
-        
-        claude_with_max_tokens = claude_llm.bind(max_tokens=12000)
-        chain_merge = LLMChain(llm=claude_with_max_tokens, prompt=prompt_merge)
-
-        resposta_merge = chain_merge.invoke({
-            "solicitacao_usuario": data.get('solicitacao_usuario'),
-            "texto_para_analise_grok": data.get('grok_text'),
-            "texto_para_analise_sonnet": data.get('sonnet_text'),
-            "texto_para_analise_gemini": data.get('gemini_text')
-        })['text']
-        
-        if not resposta_merge or not resposta_merge.strip():
-            raise ValueError("Falha no serviço de Merge (Claude Sonnet): Sem resposta.")
-        
-        merge_html = markdown2.markdown(resposta_merge, extras=["fenced-code-blocks", "tables"])
-        return jsonify({"success": True, "content": merge_html})
-
-    except Exception as e:
-        print(f"Erro no processo de merge: {e}")
-        return jsonify({"success": False, "error": str(e)}), 500
+    
+    def generate_merge_stream():
+        try:
+            yield f"data: {json.dumps({'progress': 0, 'message': 'Iniciando o processo de merge...'})}\n\n"
+            
+            prompt_merge = PromptTemplate(template=PROMPT_ATOMICO_MERGE, input_variables=["solicitacao_usuario", "texto_para_analise_grok", "texto_para_analise_sonnet", "texto_para_analise_gemini"])
+            
+            claude_with_max_tokens = claude_llm.bind(max_tokens=12000)
+            chain_merge = LLMChain(llm=claude_with_max_tokens, prompt=prompt_merge)
+
+            yield f"data: {json.dumps({'progress': 50, 'message': 'Enviando textos para o Claude Sonnet para consolidação...'})}\n\n"
+
+            resposta_merge = chain_merge.invoke({
+                "solicitacao_usuario": data.get('solicitacao_usuario'),
+                "texto_para_analise_grok": data.get('grok_text'),
+                "texto_para_analise_sonnet": data.get('sonnet_text'),
+                "texto_para_analise_gemini": data.get('gemini_text')
+            })['text']
+            
+            if not resposta_merge or not resposta_merge.strip():
+                raise ValueError("Falha no serviço de Merge (Claude Sonnet): Sem resposta.")
+            
+            merge_html = markdown2.markdown(resposta_merge, extras=["fenced-code-blocks", "tables"])
+            
+            yield f"data: {json.dumps({'progress': 100, 'message': 'Merge concluído!', 'final_result': {'content': merge_html}, 'done': True})}\n\n"
+
+        except Exception as e:
+            print(f"Erro no processo de merge: {e}")
+            yield f"data: {json.dumps({'error': str(e)})}\n\n"
+            
+    return Response(generate_merge_stream(), mimetype='text/event-stream')
 
 if __name__ == '__main__':
     app.run(debug=True)

templates/index.html

@@ -3,19 +3,23 @@
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Sistema Multi-Agente IA</title>
+    <title>Sistema Multi-Agente de IA</title>
     <link rel="stylesheet" href="{{ url_for('static', filename='style.css') }}">
 </head>
 <body>
 
+    <!-- Loader Overlay com Barra de Progresso -->
     <div id="loader-overlay" style="display: none;">
         <div class="loader-content">
             <div class="loader-spinner"></div>
             <p id="loader-message">Processando sua solicitação...</p>
-            <div class="progress-bar-container"><div id="progress-bar" class="progress-bar"></div></div>
+            <div class="progress-bar-container">
+                <div id="progress-bar" class="progress-bar"></div>
+            </div>
         </div>
     </div>
 
+    <!-- Novo Botão Flutuante de Merge -->
     <button id="merge-btn" class="floating-merge-btn" style="display: none;">Processar Merge</button>
 
     <div class="container">
@@ -25,6 +29,7 @@
                 <p id="flow-description">GROK ➔ Claude Sonnet ➔ Gemini</p>
             </div>
             <div class="controls-container">
+                <!-- Novo Seletor de Modo de Processamento -->
                 <div class="mode-toggle" title="A versão 'Hierárquica' gerará um único no texto que passará por revisão em duas instâncias. Na versão 'Atômica', serão gerados 3 textos, um em cada modelo de IA; e depois um 4º texto será gerado fazendo um texto final consolidado dessas 3 versões.">
                     <span>Hierárquico</span>
                     <label class="switch">
@@ -44,6 +49,7 @@
         
         <div id="error-box-container"></div>
 
+        <!-- Formulários -->
         <div id="real-form-container">
             <form id="request-form-real">
                 <label for="solicitacao_usuario">Digite sua solicitação (ou arraste arquivos aqui):</label>
@@ -60,12 +66,14 @@
             </form>
         </div>
 
+        <!-- Resultados -->
         <div id="results-container" class="results-container" style="display: none;">
             <div class="result-column"><div class="column-header"><h2>GROK</h2><button class="copy-btn" onclick="copyToClipboard('grok-output')">Copiar</button></div><div class="output-box" id="grok-output"></div></div>
             <div class="result-column"><div class="column-header"><h2>Claude Sonnet</h2><button class="copy-btn" onclick="copyToClipboard('sonnet-output')">Copiar</button></div><div class="output-box" id="sonnet-output"></div></div>
             <div class="result-column"><div class="column-header"><h2>Gemini</h2><button class="copy-btn" onclick="copyToClipboard('gemini-output')">Copiar</button></div><div class="output-box" id="gemini-output"></div></div>
         </div>
 
+        <!-- Novo Container para Resultado Final do Merge -->
         <div id="final-result-container" style="display: none;">
             <h2>Texto Final</h2>
             <div class="output-box" id="final-output"></div>
@@ -73,7 +81,7 @@
     </div>
 
     <script>
-        // --- Variáveis Globais ---
+        // --- Variáveis Globais e Lógica de UI ---
         const processingModeSwitch = document.getElementById('processing-mode-switch');
         const modeSwitch = document.getElementById('mode-switch');
         const realContainer = document.getElementById('real-form-container');
@@ -89,30 +97,23 @@
         const finalResultContainer = document.getElementById('final-result-container');
         const finalOutput = document.getElementById('final-output');
         let attachedFiles = [];
-        let originalUserQuery = ""; // Salvar a query original
+        let originalUserQuery = "";
 
-        // --- Lógica de UI ---
         modeSwitch.addEventListener('change', function() {
             realContainer.style.display = this.checked ? 'none' : 'block';
             mockContainer.style.display = this.checked ? 'block' : 'none';
         });
-
         processingModeSwitch.addEventListener('change', function() {
-            const description = document.getElementById('flow-description');
-            description.textContent = this.checked ? "GROK | Claude Sonnet | Gemini (Paralelo)" : "GROK ➔ Claude Sonnet ➔ Gemini";
+            document.getElementById('flow-description').textContent = this.checked ? "GROK | Claude Sonnet | Gemini (Paralelo)" : "GROK ➔ Claude Sonnet ➔ Gemini";
         });
         
-        // --- Lógica de Upload (sem alterações) ---
+        // --- Lógica de Upload ---
         ['dragenter', 'dragover', 'dragleave', 'drop'].forEach(eventName => {
             textarea.addEventListener(eventName, preventDefaults, false);
             document.body.addEventListener(eventName, preventDefaults, false);
         });
-        ['dragenter', 'dragover'].forEach(eventName => {
-            textarea.addEventListener(eventName, () => textarea.classList.add('drag-over'), false);
-        });
-        ['dragleave', 'drop'].forEach(eventName => {
-            textarea.addEventListener(eventName, () => textarea.classList.remove('drag-over'), false);
-        });
+        ['dragenter', 'dragover'].forEach(eventName => textarea.addEventListener(eventName, () => textarea.classList.add('drag-over'), false));
+        ['dragleave', 'drop'].forEach(eventName => textarea.addEventListener(eventName, () => textarea.classList.remove('drag-over'), false));
         textarea.addEventListener('drop', handleDrop, false);
         function preventDefaults(e) { e.preventDefault(); e.stopPropagation(); }
         function handleDrop(e) { handleFiles(e.dataTransfer.files); }
@@ -146,22 +147,16 @@
 
         async function handleFormSubmit(event) {
             event.preventDefault();
-            
-            // Limpa o estado da UI
             errorContainer.innerHTML = '';
             resultsContainer.style.display = 'none';
             finalResultContainer.style.display = 'none';
             mergeBtn.style.display = 'none';
             document.querySelectorAll('.output-box').forEach(box => box.innerHTML = '');
-
-            // Inicia o loader
             loaderMessage.textContent = 'Iniciando conexão...';
             progressBar.style.width = '0%';
             loader.style.display = 'flex';
-
             const formData = new FormData();
             formData.append('processing_mode', processingModeSwitch.checked ? 'atomic' : 'hierarchical');
-            
             if (modeSwitch.checked) {
                 formData.append('mode', 'test');
                 formData.append('mock_text', document.getElementById('mock_text').value);
@@ -172,9 +167,6 @@
                 formData.append('solicitacao', originalUserQuery);
                 attachedFiles.forEach(file => { formData.append('files', file); });
             }
-
-            // A lógica de streaming via fetch continua a mesma
-            // ... (código do fetch e do reader, sem alterações)
             try {
                 const response = await fetch('/process', { method: 'POST', body: formData });
                 if (!response.ok || !response.body) throw new Error(`Erro na resposta do servidor: ${response.statusText}`);
@@ -191,7 +183,7 @@
                             if (jsonData.trim()) {
                                 try {
                                     const data = JSON.parse(jsonData);
-                                    processStreamData(data);
+                                    processStreamData(data, false);
                                 } catch (e) { console.error("Erro ao parsear JSON:", jsonData); }
                             }
                         }
@@ -204,39 +196,12 @@
             }
         }
 
-        function processStreamData(data) {
-            if (data.error) {
-                showError(data.error);
-                loader.style.display = 'none';
-                return;
-            }
-            
-            loaderMessage.textContent = data.message;
-            progressBar.style.width = data.progress + '%';
-
-            if (data.partial_result) {
-                resultsContainer.style.display = 'flex';
-                const targetBox = document.getElementById(data.partial_result.id);
-                if (targetBox) targetBox.innerHTML = data.partial_result.content;
-            }
-
-            if (data.done) {
-                setTimeout(() => {
-                    loader.style.display = 'none';
-                    // Mostra o botão de merge apenas se o modo for Atômico
-                    if (data.mode === 'atomic') {
-                        mergeBtn.style.display = 'block';
-                    }
-                }, 1000);
-            }
-        }
-
-        // --- Lógica do Botão de Merge ---
+        // --- Lógica do Botão de Merge com Streaming ---
         mergeBtn.addEventListener('click', async function() {
             loaderMessage.textContent = 'Processando o merge dos textos...';
             progressBar.style.width = '0%';
             loader.style.display = 'flex';
-            this.style.display = 'none'; // Esconde o botão
+            this.style.display = 'none';
 
             const payload = {
                 solicitacao_usuario: originalUserQuery,
@@ -251,21 +216,63 @@
                     headers: { 'Content-Type': 'application/json' },
                     body: JSON.stringify(payload)
                 });
-                const result = await response.json();
+                if (!response.ok || !response.body) throw new Error(`Erro na resposta do servidor: ${response.statusText}`);
                 
-                if (result.success) {
-                    finalOutput.innerHTML = result.content;
-                    finalResultContainer.style.display = 'block';
-                } else {
-                    showError(result.error || "Ocorreu um erro desconhecido no merge.");
+                const reader = response.body.getReader();
+                const decoder = new TextDecoder();
+                while (true) {
+                    const { done, value } = await reader.read();
+                    if (done) break;
+                    const chunk = decoder.decode(value, { stream: true });
+                    const lines = chunk.split('\n\n');
+                    lines.forEach(line => {
+                        if (line.startsWith('data: ')) {
+                            const jsonData = line.substring(6);
+                            if (jsonData.trim()) {
+                                try {
+                                    const data = JSON.parse(jsonData);
+                                    processStreamData(data, true); // true = é merge
+                                } catch (e) { console.error("Erro ao parsear JSON do merge:", jsonData); }
+                            }
+                        }
+                    });
                 }
             } catch (error) {
                 showError("A conexão falhou ao tentar processar o merge.");
-            } finally {
                 loader.style.display = 'none';
             }
         });
 
+        // --- Função de Processamento de Stream Unificada ---
+        function processStreamData(data, isMerge) {
+            if (data.error) {
+                showError(data.error);
+                loader.style.display = 'none';
+                return;
+            }
+            
+            loaderMessage.textContent = data.message;
+            progressBar.style.width = data.progress + '%';
+
+            if (isMerge && data.final_result) {
+                finalOutput.innerHTML = data.final_result.content;
+                finalResultContainer.style.display = 'block';
+            } else if (data.partial_result) {
+                resultsContainer.style.display = 'flex';
+                const targetBox = document.getElementById(data.partial_result.id);
+                if (targetBox) targetBox.innerHTML = data.partial_result.content;
+            }
+
+            if (data.done) {
+                setTimeout(() => {
+                    loader.style.display = 'none';
+                    if (data.mode === 'atomic' && !isMerge) {
+                        mergeBtn.style.display = 'block';
+                    }
+                }, 1000);
+            }
+        }
+
         function showError(message) {
             errorContainer.innerHTML = `<div class="error-box"><strong>Erro:</strong> ${message}<span class="close-btn-error" onclick="this.parentElement.style.display='none';" title="Fechar">&times;</span></div>`;
         }