update ui

app.py

@@ -1,15 +1,18 @@
-import os
 import streamlit as st
 from crewai import Crew, Process, Agent, Task
 from langchain_core.callbacks import BaseCallbackHandler
 from langchain_openai import ChatOpenAI
 from typing import Any, Dict
 
+## TODO
+## use an example, and clarify which agent is speaking in the UI
+
+
 # documentation
 # https://www.linkedin.com/pulse/building-simple-user-interfaces-crewai-streamlit-robyn-le-sueur-azvdf/
 
 # Initialize the OpenAI model for use with agents
-openai = ChatOpenAI(model="gpt-3.5-turbo", temperature=0.5)
+openai = ChatOpenAI(model="gpt-3.5-turbo", temperature=0.1)
 
 class CustomHandler(BaseCallbackHandler):
     """A custom handler for logging interactions within the process chain."""
@@ -20,8 +23,9 @@ class CustomHandler(BaseCallbackHandler):
 
     def on_chain_start(self, serialized: Dict[str, Any], outputs: Dict[str, Any], **kwargs: Any) -> None:
         """Log the start of a chain with user input."""
-        st.session_state.messages.append({"role": "assistant", "content": outputs['input']})
-        st.chat_message("assistant").write(outputs['input'])
+        # comment out verbose chain input
+        #st.session_state.messages.append({"role": "assistant", "content": outputs['input']})
+        #st.chat_message("assistant").write(outputs['input'])
         
     def on_agent_action(self, serialized: Dict[str, Any], inputs: Dict[str, Any], **kwargs: Any) -> None:
         """""Log the action taken by an agent during a chain run."""
@@ -30,8 +34,9 @@ class CustomHandler(BaseCallbackHandler):
         
     def on_chain_end(self, outputs: Dict[str, Any], **kwargs: Any) -> None:
         """Log the end of a chain with the output generated by an agent."""
-        st.session_state.messages.append({"role": self.agent_name, "content": outputs['output']})
-        st.chat_message(self.agent_name).write(outputs['output'])
+        message = f"**{self.agent_name}:**\n\n{outputs['output']}"
+        st.session_state.messages.append({"role": self.agent_name, "content": message})
+        st.chat_message(self.agent_name).write(message)
 
 # Define agents with their specific roles and goals
 project_manager = Agent(
@@ -52,7 +57,7 @@ coder = Agent(
                  You produce functional, feature complete code.''',
     goal='Develop high-quality, well-structured Python code.',
     llm=openai,
-    callbacks=[CustomHandler("Coder")],
+    callbacks=[CustomHandler("Python Coder")],
 )
 
 # Streamlit UI setup
@@ -97,6 +102,6 @@ if prompt := st.chat_input():
     final = project_crew.kickoff()
 
     # Display the final result
-    result = f"## Here is the Final Result \n\n {final}"
+    result = f"### Crew Manager's Final Result: \n\n {final}"
     st.session_state.messages.append({"role": "assistant", "content": result})
     st.chat_message("assistant").write(result)