Initial commit

Author: anajmi07

cx-agent-frontend/Dockerfile

@@ -0,0 +1,48 @@
+# Multi-stage build for Streamlit frontend
+FROM python:3.11-slim as builder
+
+# Install uv
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /bin/uv
+
+# Set working directory
+WORKDIR /app
+
+# Copy dependency files
+COPY pyproject.toml ./
+
+# Install dependencies
+RUN uv sync --frozen --no-dev
+
+# Production stage
+FROM python:3.11-slim as production
+
+# Install uv
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /bin/uv
+
+# Create non-root user
+RUN groupadd -r appuser && useradd -r -g appuser appuser
+
+# Set working directory
+WORKDIR /app
+
+# Copy virtual environment from builder
+COPY --from=builder /app/.venv /app/.venv
+
+# Copy application code
+COPY src/ ./src/
+
+# Change ownership to non-root user
+RUN chown -R appuser:appuser /app
+
+# Switch to non-root user
+USER appuser
+
+# Set environment variables
+ENV PATH="/app/.venv/bin:$PATH"
+ENV PYTHONPATH="/app/src"
+
+# Expose port
+EXPOSE 8501
+
+# Run application
+CMD ["streamlit", "run", "src/app.py", "--server.port=8501", "--server.address=0.0.0.0"]

cx-agent-frontend/pyproject.toml

@@ -1 +1,32 @@
+[project]
+name = "cx-agent-frontend"
+version = "0.1.0"
+description = "Clean Streamlit Frontend for CX Agent"
+requires-python = ">=3.11"
+dependencies = [
+    "streamlit>=1.28.0",
+    "requests>=2.31.0",
+    "pydantic>=2.5.0",
+    "boto3>=1.34.0",
+]
 
+[project.optional-dependencies]
+dev = [
+    "ruff>=0.1.0",
+    "mypy>=1.7.0",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src"]
+
+[tool.ruff]
+target-version = "py311"
+line-length = 88
+
+[tool.mypy]
+python_version = "3.11"
+strict = true

cx-agent-frontend/src/__init__.py

@@ -0,0 +1 @@
+# Frontend package

cx-agent-frontend/src/app.py

@@ -0,0 +1,146 @@
+"""Main Streamlit application."""
+
+import uuid
+from datetime import datetime
+
+import streamlit as st
+
+from components.chat import render_message, render_sidebar
+from components.config import render_agentcore_config
+from models.message import Message
+from services.conversation_client import ConversationClient
+from services.agentcore_client import AgentCoreClient
+
+
+def init_session_state():
+    """Initialize session state variables."""
+    if "conversation_id" not in st.session_state:
+        st.session_state.conversation_id = None
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+    if "user_id" not in st.session_state:
+        st.session_state.user_id = f"user_{str(uuid.uuid4())[:8]}"
+    if "agent_runtime_arn" not in st.session_state:
+        st.session_state.agent_runtime_arn = ""
+    if "region" not in st.session_state:
+        st.session_state.region = "us-east-1"
+    if "use_agentcore" not in st.session_state:
+        st.session_state.use_agentcore = False
+    if "auth_token" not in st.session_state:
+        st.session_state.auth_token = ""
+
+
+def main():
+    """Main application."""
+    st.set_page_config(
+        page_title="CX Agent Chat",
+        page_icon="🤖",
+        layout="wide",
+        initial_sidebar_state="expanded",
+    )
+
+    # Custom CSS for better design
+    st.markdown("""
+    <style>
+    .stButton > button {
+        border-radius: 20px;
+        border: 1px solid #e0e0e0;
+        transition: all 0.3s ease;
+    }
+    .stButton > button:hover {
+        transform: translateY(-2px);
+        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+    }
+    .feedback-section {
+        background-color: #f8f9fa;
+        padding: 10px;
+        border-radius: 10px;
+        margin: 10px 0;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+    st.title("🤖 CX Agent Chat")
+    if st.session_state.get('use_agentcore', False):
+        st.caption("Powered by AWS Bedrock AgentCore Runtime")
+    else:
+        st.caption("Powered by LiteLLM Gateway hosted on AWS")
+
+    init_session_state()
+    
+    # Render configuration
+    config_valid = render_agentcore_config()
+    model = render_sidebar()
+
+    # Initialize client based on configuration
+    if st.session_state.use_agentcore and config_valid:
+        auth_token = st.session_state.get('auth_token', '')
+        client = AgentCoreClient(
+            agent_runtime_arn=st.session_state.agent_runtime_arn,
+            region=st.session_state.region,
+            auth_token=auth_token
+        )
+        if auth_token:
+            st.info("🚀 Connected to AgentCore Runtime")
+        else:
+            st.warning("⚠️ AgentCore configured but no auth token provided")
+    else:
+        client = ConversationClient()
+        if st.session_state.use_agentcore:
+            st.warning("⚠️ AgentCore configuration invalid, using local backend")
+        else:
+            st.info("🔧 Using local backend")
+
+    # Display chat messages
+    for message in st.session_state.messages:
+        render_message(message, client)
+
+    # Chat input
+    if prompt := st.chat_input("Type your message..."):
+        # Create conversation if needed
+        if not st.session_state.conversation_id:
+            with st.spinner("Creating conversation..."):
+                conversation_id = client.create_conversation(st.session_state.user_id)
+                if not conversation_id:
+                    st.stop()
+                st.session_state.conversation_id = conversation_id
+
+        # Add user message
+        user_message = Message(role="user", content=prompt, timestamp=datetime.now())
+        st.session_state.messages.append(user_message)
+        render_message(user_message)
+
+        # Send message and get response
+        with st.spinner("Thinking..."):
+            response = client.send_message(
+                st.session_state.conversation_id, prompt, model
+            )
+
+            if response:
+                # Debug: Show what we received
+                st.write("Debug - API Response:", response)
+                
+                # Add assistant message
+                metadata = {
+                    "model": model,
+                    "status": response.get("status", "success"),
+                }
+                
+                # Add tools_used to metadata if available
+                if "tools_used" in response and response["tools_used"]:
+                    metadata["tools_used"] = ",".join(response["tools_used"])
+                
+                assistant_message = Message(
+                    role="assistant",
+                    content=response["response"],
+                    timestamp=datetime.now(),
+                    metadata=metadata,
+                )
+                st.session_state.messages.append(assistant_message)
+                render_message(assistant_message, client)
+
+        st.rerun()
+
+
+if __name__ == "__main__":
+    main()

cx-agent-frontend/src/components/__init__.py

@@ -0,0 +1 @@
+# Components package

cx-agent-frontend/src/components/chat.py

@@ -0,0 +1,101 @@
+"""Chat components."""
+
+import streamlit as st
+
+from models.message import Message
+from services.conversation_client import ConversationClient
+
+
+def render_message(message: Message, client: ConversationClient = None):
+    """Render a single message."""
+    with st.chat_message(message.role):
+        st.write(message.content)
+
+        # Add feedback buttons for assistant messages
+        if message.role == "assistant" and client:
+            message_id = str(message.metadata.get("message_id", hash(message.content)))
+            
+            # Check if feedback already given
+            feedback_key = f"feedback_given_{message_id}"
+            if feedback_key not in st.session_state:
+                st.session_state[feedback_key] = False
+            
+            if not st.session_state[feedback_key]:
+                st.markdown("---")
+                st.markdown("**Was this helpful?**")
+                
+                col1, col2, col3 = st.columns([2, 2, 6])
+                
+                with col1:
+                    if st.button("👍 Helpful", key=f"up_{message_id}", use_container_width=True):
+                        if client.submit_feedback(message_id, st.session_state.get("conversation_id", "default"), 1.0, ""):
+                            st.session_state[feedback_key] = True
+                            st.success("✓ Thank you for your feedback!")
+                            st.rerun()
+                
+                with col2:
+                    if st.button("👎 Not helpful", key=f"down_{message_id}", use_container_width=True):
+                        with st.expander("📝 Tell us more (optional)", expanded=True):
+                            feedback_text = st.text_area(
+                                "How can we improve?",
+                                key=f"text_{message_id}",
+                                placeholder="Your feedback helps us improve...",
+                                height=80
+                            )
+                            if st.button("📤 Submit Feedback", key=f"submit_{message_id}", type="primary"):
+                                if client.submit_feedback(message_id, st.session_state.get("conversation_id", "default"), 0.0, feedback_text):
+                                    st.session_state[feedback_key] = True
+                                    st.success("✓ Thank you for your feedback!")
+                                    st.rerun()
+            else:
+                st.markdown("<div style='color: #28a745; font-size: 0.9em;'>✓ Feedback submitted</div>", unsafe_allow_html=True)
+
+        # Show tool calls if available
+        if message.metadata and "tools_used" in message.metadata:
+            tools_str = message.metadata["tools_used"]
+            if tools_str:
+                tools_used = [tool.strip() for tool in tools_str.split(",") if tool.strip()]
+                if tools_used:
+                    st.markdown("**🔧 Tools Used:**")
+                    for tool in tools_used:
+                        st.markdown(f"• `{tool}`")
+
+        # Show metadata if available
+        if message.metadata:
+            with st.expander("Message Details", expanded=False):
+                st.json(message.metadata)
+
+
+def render_sidebar():
+    """Render sidebar with configuration."""
+    with st.sidebar:
+        st.header("Configuration")
+
+        # Model selection
+        model = st.selectbox(
+            "Model",
+            options=[
+                "gpt-4o-mini",
+                "gpt-4o",
+                "bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0",
+                "bedrock/anthropic.claude-3-5-haiku-20241022-v1:0",
+            ],
+            index=0,
+        )
+
+        # User ID
+        user_id = st.text_input("User ID", value=st.session_state.user_id)
+        st.session_state.user_id = user_id
+
+        # Conversation controls
+        st.header("Conversation")
+
+        if st.button("New Conversation", type="primary"):
+            st.session_state.conversation_id = None
+            st.session_state.messages = []
+            st.rerun()
+
+        if st.session_state.conversation_id:
+            st.success(f"Active: {st.session_state.conversation_id[:8]}...")
+
+        return model

cx-agent-frontend/src/components/config.py

@@ -0,0 +1,60 @@
+"""Configuration components for AgentCore."""
+
+import streamlit as st
+
+
+def render_agentcore_config():
+    """Render AgentCore configuration in sidebar."""
+    with st.sidebar:
+        st.header("⚙️ Configuration")
+        
+        # Backend selection
+        use_agentcore = st.checkbox(
+            "Use AWS AgentCore",
+            value=st.session_state.get("use_agentcore", False),
+            help="Toggle between local backend and AWS AgentCore"
+        )
+        st.session_state.use_agentcore = use_agentcore
+        
+        if use_agentcore:
+            st.subheader("AgentCore Settings")
+            
+            # Agent Runtime ARN
+            agent_runtime_arn = st.text_input(
+                "Agent Runtime ARN",
+                value=st.session_state.get("agent_runtime_arn", ""),
+                placeholder="arn:aws:bedrock-agent-runtime:region:account:agent/agent-id",
+                help="AWS Bedrock Agent Runtime ARN"
+            )
+            st.session_state.agent_runtime_arn = agent_runtime_arn
+            
+            # Region
+            region = st.selectbox(
+                "AWS Region",
+                ["us-east-1", "us-west-2", "eu-west-1", "ap-southeast-1"],
+                index=0 if st.session_state.get("region", "us-east-1") == "us-east-1" else 
+                      ["us-east-1", "us-west-2", "eu-west-1", "ap-southeast-1"].index(st.session_state.get("region", "us-east-1"))
+            )
+            st.session_state.region = region
+            
+            # Auth Token
+            auth_token = st.text_input(
+                "JWT Auth Token",
+                value=st.session_state.get("auth_token", ""),
+                type="password",
+                placeholder="Enter JWT token from Cognito",
+                help="JWT token for authentication with AgentCore"
+            )
+            st.session_state.auth_token = auth_token
+            
+            # Validation
+            config_valid = bool(agent_runtime_arn.strip() and auth_token.strip())
+            if not agent_runtime_arn.strip():
+                st.error("⚠️ Agent Runtime ARN is required")
+            if not auth_token.strip():
+                st.error("⚠️ JWT Auth Token is required")
+            
+            return config_valid
+        else:
+            st.info("Using local backend")
+            return True