Update app.py

app.py

@@ -74,26 +74,22 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
     except Exception as e:
         return f"⚠️ Erro ao chamar o modelo: {e}"
 
-    # === 🔍 Extração robusta do texto do assistente ===
+    # === Extrair somente o texto do assistente ===
     try:
         data = outputs[0]["generated_text"]
-        # Procura o último bloco com role == "assistant"
         assistant_blocks = [b for b in data if b.get("role") == "assistant"]
         if assistant_blocks:
-            last_assistant = assistant_blocks[-1]
-            content = last_assistant.get("content")
-            if isinstance(content, list):
-                # O modelo retorna [{'type':'text','text':'resposta'}]
-                response = content[0].get("text", "").strip()
+            last = assistant_blocks[-1]
+            content = last.get("content")
+            if isinstance(content, list) and len(content) > 0:
+                text = content[0].get("text", "")
             else:
-                response = str(content).strip()
-        else:
-            response = str(data)
-    except Exception:
-        # fallback caso o formato varie
-        response = str(outputs)
+                text = str(content)
+            return text.strip()
+        return ""
+    except Exception as e:
+        return f"⚠️ Erro ao processar saída: {e}"
 
-    return response
 
 # -------------------------
 # Gradio UI