Update app.py

app.py

@@ -125,10 +125,23 @@ def main():
             b64 = base64.b64encode(html.encode()).decode()
             st.markdown(f'<a href="data:file/html;base64,{b64}" download="ui.html">Download HTML</a>', unsafe_allow_html=True)
 
-            st.subheader("🧠 Agent Log")
-            log = "".join([f"---\n{m['role'].title()}:\n{m['content']}\n\n" for m in final_state["messages"]])
-            st.text_area("Agent Communication", value=log, height=300)
-
+            # Communication History
+            st.subheader("🧠 Agent Communication Log")
+            history_text = ""
+            for msg in final_state["messages"]:
+                role = msg["role"].replace("_", " ").title()
+                content = msg["content"]
+                history_text += f"---\n{role}:\n{content}\n\n"
+            st.text_area("Agent Dialogue", value=history_text, height=300)
+
+            # Download Chat Log
+            b64_hist = base64.b64encode(history_text.encode()).decode()
+            st.markdown(
+                f'<a href="data:file/txt;base64,{b64_hist}" download="agent_communication.txt" '
+                'style="padding: 0.4em 1em; background: #4CAF50; color: white; border-radius: 0.3em; text-decoration: none;">'
+                '📥 Download Communication Log</a>',
+                unsafe_allow_html=True
+            )
             st.subheader("📊 Performance")
             st.write(f"⏱️ Total Time: {total_time:.2f} seconds")
             st.write(f"🔁 Iterations: {final_state['iteration']}")