Upload 2 files

Dockerfile

@@ -1,26 +1,23 @@
-# Dockerfile
 FROM python:3.10-slim
 
+# Set working directory in the container
 WORKDIR /app
-# tells the system to treat /app as the "home" directory, so Streamlit will create /app/.streamlit instead of /.streamlit.
-# /app is already writable as root, so no permission issues.
-ENV HOME=/app
 
-# (Optional) system deps if you need them; safe to remove if not needed
-RUN apt-get update && apt-get install -y --no-install-recommends curl && rm -rf /var/lib/apt/lists/*
+# Copy requirements and install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Install Python deps
-COPY requirements.txt /app/
-RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
+# Create non-root user for HF Spaces and make /app writable
+RUN adduser --disabled-password --gecos '' --uid 1000 --gid 1000 user && chown -R user:user /app
 
-# Copy code
-COPY . /app
+# Copy application files
+COPY . .
 
-# Do NOT set STREAMLIT_SERVER_PORT here.
-# Spaces will inject $PORT at runtime.
+# Expose Streamlit port
+EXPOSE 8501
 
+# Health check for Streamlit
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-# Start Streamlit on the port Spaces provides, and bind to 0.0.0.0
-# CMD ["bash", "-lc", "streamlit run /app/frontend.py --server.port $PORT --server.address 0.0.0.0"]
-# CMD ["sh", "-c", "streamlit run /app/frontend.py --server.port $PORT --server.address 0.0.0.0"]
-CMD ["streamlit", "run", "/app/frontend.py", "--server.port", "8501", "--server.address", "0.0.0.0"]
+# Run Streamlit as the non-root user
+CMD ["streamlit", "run", "frontend.py", "--server.port=8501", "--server.address=0.0.0.0"]

frontend.py

@@ -59,7 +59,8 @@ with st.sidebar:
         
         # Show available tools
         st.markdown("### 🛠️ Available Tools")
-        for tool in st.session_state.available_tools:
+        available_tools = st.session_state.get("available_tools", [])
+        for tool in available_tools:
             st.markdown(f"• `{tool}`")
     else:
         st.info("⏳ Agent Initializing...")
@@ -200,10 +201,9 @@ if prompt := st.chat_input("Type your message here..."):
         message_placeholder.markdown(response)
         st.session_state.messages.append({"role": "assistant", "content": response})
 
-# Initialize agent on first load
-# Optional: manual connect so page renders fast
+# Auto-initialize agent on first load
 if not st.session_state.agent_initialized:
-    if st.sidebar.button("🔌 Connect agent"):
+    with st.spinner("🔧 Initializing agent..."):
         asyncio.run(initialize_agent())
 
 # Footer