UI updates

agent.py

@@ -209,7 +209,7 @@ def get_agent_config() -> OmegaConf:
     })
     return cfg
 
-def initialize_agent(_cfg, update_func=None):
+def initialize_agent(_cfg, agent_progress_callback=None):
     
     legal_assistant_instructions = """
     - You are a helpful legal assistant, with expertise in case law for the state of Alaska.
@@ -242,6 +242,6 @@ def initialize_agent(_cfg, update_func=None):
         tools=create_assistant_tools(_cfg),
         topic="Case law in Alaska",
         custom_instructions=legal_assistant_instructions,
-        update_func=update_func,
+        agent_progress_callback=agent_progress_callback,
     )
     return agent

requirements.txt

@@ -1,9 +1,9 @@
 omegaconf==2.3.0
 python-dotenv==1.0.1
-streamlit==1.32.2
+streamlit==1.39.0
 streamlit_pills==0.3.0
 streamlit-feedback==0.1.3
 uuid==1.30
 langdetect==1.0.9
 langcodes==3.4.0
-vectara-agentic==0.1.16
+vectara-agentic==0.1.18

st_app.py

@@ -1,5 +1,6 @@
 from PIL import Image
 import sys
+import re
 
 import streamlit as st
 from streamlit_pills import pills
@@ -12,9 +13,6 @@ from agent import initialize_agent, get_agent_config
 
 initial_prompt = "How can I help you today?"
 
-def toggle_logs():
-    st.session_state.show_logs = not st.session_state.show_logs
-
 def show_example_questions():        
     if len(st.session_state.example_messages) > 0 and st.session_state.first_turn:            
         selected_example = pills("Queries to Try:", st.session_state.example_messages, index=None)
@@ -24,23 +22,41 @@ def show_example_questions():
             return True
     return False
 
-def update_func(status_type: AgentStatusType, msg: str):
-    if status_type != AgentStatusType.AGENT_UPDATE:
-        output = f"{status_type.value} - {msg}"
-        st.session_state.log_messages.append(output)
+def agent_progress_callback(status_type: AgentStatusType, msg: str):
+    output = f'<span style="color:blue;">{status_type.value}</span>: {msg}'
+    st.session_state.log_messages.append(output)
+    if 'status' in st.session_state:
+        latest_message = ''
+        if status_type == AgentStatusType.TOOL_CALL:
+            match = re.search(r"'([^']*)'", msg)
+            tool_name = match.group(1) if match else "Unknown tool"
+            latest_message = f"Calling tool {tool_name}..."
+        elif status_type == AgentStatusType.TOOL_OUTPUT:
+            latest_message = "Analyzing tool output..."
+        else:
+            return
+        
+        st.session_state.status.update(label=latest_message)
+        max_log_msg_size = 200
+        with st.session_state.status:
+            for log_msg in st.session_state.log_messages:
+                st.markdown(log_msg[:max_log_msg_size]+'...', unsafe_allow_html=True)
+
+@st.dialog(title="Agent logs", width='large')
+def show_modal():
+    for log_msg in st.session_state.log_messages:
+        st.write(log_msg, unsafe_allow_html=True)
 
 async def launch_bot():
     def reset():
         st.session_state.messages = [{"role": "assistant", "content": initial_prompt, "avatar": "🦖"}]
-        st.session_state.thinking_message = "Agent at work..."
         st.session_state.log_messages = []
         st.session_state.prompt = None
         st.session_state.ex_prompt = None
         st.session_state.first_turn = True
-        st.session_state.logs_enabled = False
         st.session_state.show_logs = False
         if 'agent' not in st.session_state:
-            st.session_state.agent = initialize_agent(cfg, update_func=update_func)
+            st.session_state.agent = initialize_agent(cfg, agent_progress_callback=agent_progress_callback)
         else:
             st.session_state.agent.clear_memory()
 
@@ -67,15 +83,9 @@ async def launch_bot():
             if st.button('Start Over'):
                 reset()
                 st.rerun()
-        with bc2:  # Updated button for enabling/disabling logs
-            if st.session_state.logs_enabled:
-                if st.button('Disable Logs', key='disable_logs'):
-                    st.session_state.logs_enabled = False
-                    st.rerun()
-            else:
-                if st.button('Enable Logs', key='enable_logs'):
-                    st.session_state.logs_enabled = True
-                    st.rerun()
+        with bc2:
+            if st.button('Show Logs'):
+                show_modal()
 
         st.divider()
         st.markdown(
@@ -106,7 +116,7 @@ async def launch_bot():
         prompt = st.chat_input()
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt, "avatar": '🧑‍💻'})
-        st.session_state.prompt = prompt  # Save the prompt in session state
+        st.session_state.prompt = prompt
         st.session_state.log_messages = []
         st.session_state.show_logs = False
         with st.chat_message("user", avatar='🧑‍💻'):
@@ -117,9 +127,9 @@ async def launch_bot():
     # Generate a new response if last message is not from assistant
     if st.session_state.prompt:
         with st.chat_message("assistant", avatar='🤖'):
-            with st.spinner(st.session_state.thinking_message):
-                res = st.session_state.agent.chat(st.session_state.prompt)
-                res = escape_dollars_outside_latex(res)
+            st.session_state.status = st.status('Processing...', expanded=False)
+            res = st.session_state.agent.chat(st.session_state.prompt)
+            res = escape_dollars_outside_latex(res)
             message = {"role": "assistant", "content": res, "avatar": '🤖'}
             st.session_state.messages.append(message)
             st.markdown(res)
@@ -137,27 +147,14 @@ async def launch_bot():
 
     # Record user feedback
     if (st.session_state.messages[-1]["role"] == "assistant") & (st.session_state.messages[-1]["content"] != initial_prompt):
-        if st.session_state.show_logs and st.session_state.logs_enabled:  # Only show logs if enabled
-            streamlit_feedback(
-                feedback_type="thumbs", on_submit=thumbs_feedback, key=st.session_state.feedback_key,
-                kwargs={"user_query": st.session_state.messages[-2]["content"],
-                        "bot_response": st.session_state.messages[-1]["content"],
-                        "demo_name": cfg["demo_name"]}
+        if "feedback_key" not in st.session_state:
+            st.session_state.feedback_key = 0
+        streamlit_feedback(
+            feedback_type="thumbs", on_submit=thumbs_feedback, key=str(st.session_state.feedback_key),
+            kwargs={"user_query": st.session_state.messages[-2]["content"],
+                    "bot_response": st.session_state.messages[-1]["content"],
+                    "demo_name": cfg["demo_name"]}
             )
 
-    log_placeholder = st.empty()
-    with log_placeholder.container():
-        if st.session_state.logs_enabled:  # Show logs button only if log toggle is enabled
-            if st.session_state.show_logs:
-                st.button("Hide Logs", on_click=toggle_logs)
-                for msg in st.session_state.log_messages:
-                    if len(msg) > 100:  # Use text_area for longer messages
-                        st.text_area(label="Log", value=msg, height=100, disabled=True)
-                    else:
-                        st.text(msg)
-            else:
-                if len(st.session_state.log_messages) > 0:
-                    st.button("Show Logs", on_click=toggle_logs)
-
 
     sys.stdout.flush()