refactor: update application name and enhance header display in Streamlit app

This commit is contained in:
2025-03-26 12:27:00 +00:00
parent bd56cc839d
commit 49aebc12d5
4 changed files with 68 additions and 45 deletions

View File

@@ -85,6 +85,7 @@ def init_session_state():
base_url=base_url,
)
st.session_state.model_name = model_name
st.session_state.provider_name = provider_name # Store provider name
logger.info("LLMClient initialized successfully.")
except Exception as e:
@@ -182,9 +183,34 @@ def handle_user_input():
def main():
"""Main function to run the Streamlit app."""
st.title("MCP Chat App") # Updated title
try:
init_session_state()
# --- Display Enhanced Header ---
provider_name = st.session_state.get("provider_name", "Unknown Provider")
model_name = st.session_state.get("model_name", "Unknown Model")
mcp_manager = st.session_state.client.mcp_manager # Get the manager
server_count = 0
tool_count = 0
if mcp_manager and mcp_manager.initialized:
server_count = len(mcp_manager.servers)
try:
# Get tool count (might be slightly slow if many tools/servers)
tool_count = len(mcp_manager.list_all_tools())
except Exception as e:
logger.warning(f"Could not retrieve tool count for header: {e}")
tool_count = "N/A" # Display N/A if listing fails
# Display the new header format
st.markdown(f"# Say Hi to **{provider_name.capitalize()}**!")
st.write(f"MCP Servers: **{server_count}** | Tools: **{tool_count}**")
st.write(f"Model: **{model_name}**")
st.divider()
# -----------------------------
# Removed the previous caption display
display_chat_messages()
handle_user_input()
except Exception as e: