docker improvements

.gitignore

@@ -3,4 +3,7 @@ agents/__pycache__/*
 data_ingestion/__pycache__/*
 faiss_index_store
 orchestrator/__pycache__
-
+supervisord.log
+*.log
+Dockerfile.*
+docker-compose.yaml

Dockerfile

@@ -0,0 +1,48 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# Install system dependencies (including ffmpeg for audio)
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        build-essential \
+        gcc \
+        ffmpeg \
+        supervisor && \ # Install supervisor system package
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy all application code
+# Ensure these paths correctly copy all your necessary modules
+COPY ./agents /app/agents
+COPY ./data_ingestion /app/data_ingestion
+COPY ./orchestrator /app/orchestrator
+COPY ./streamlit /app/streamlit
+COPY ./example_portfolio.json /app/example_portfolio.json # For orchestrator and streamlit
+
+COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf
+
+
+ENV GEMINI_MODEL_NAME="gemini-1.5-flash-latest"
+ENV WHISPER_MODEL_SIZE="small"
+ENV FAISS_INDEX_PATH="/app/faiss_index_store" 
+
+
+ENV AGENT_API_URL="http://localhost:8001"
+ENV AGENT_SCRAPING_URL="http://localhost:8002"
+ENV AGENT_RETRIEVER_URL="http://localhost:8003"
+ENV AGENT_ANALYSIS_URL="http://localhost:8004"
+ENV AGENT_LANGUAGE_URL="http://localhost:8005"
+ENV AGENT_VOICE_URL="http://localhost:8006"
+ENV ORCHESTRATOR_URL="http://localhost:8000" 
+
+RUN mkdir -p /app/faiss_index_store && \
+    chown -R nobody:nogroup /app/faiss_index_store || true # Example, adjust user/group as needed
+
+
+EXPOSE 8501
+
+CMD ["/usr/bin/supervisord", "-c", "/etc/supervisor/conf.d/supervisord.conf"]

Dockerfile.fastapi

@@ -1,14 +0,0 @@
-FROM python:3.10
-
-WORKDIR /app
-
-COPY requirements.txt .
-
-RUN apt-get update && apt-get install -y --no-install-recommends build-essential gcc && \
-    pip install --no-cache-dir -r requirements.txt 
-
-COPY . .
-
-EXPOSE 8000
-
-

Dockerfile.streamlit

@@ -1,15 +0,0 @@
-FROM python:3.10-slim
-
-WORKDIR /app
-
-COPY requirements.txt .
-
-RUN apt-get update && apt-get install -y --no-install-recommends build-essential gcc ffmpeg && \
-    pip install --no-cache-dir -r requirements.txt
-
-COPY streamlit ./streamlit
-COPY example_portfolio.json .
-
-EXPOSE 8501
-
-CMD ["streamlit", "run", "streamlit/app.py", "--server.port=8501", "--server.address=0.0.0.0"]

README.md

@@ -1,15 +1,14 @@
 ---
-title: AI Financial Assistant
-emoji: 📈
-colorFrom: blue
-colorTo: green
+title: AI Financial Assistant (Single Container)
+emoji: ⚙️
+colorFrom: orange
+colorTo: red
 sdk: docker
-app_port: 8501 
+app_port: 8501
 pinned: false
-
 ---
 
-# AI Financial Assistant - Morning Market Brief
+# AI Financial Assistant - Morning Market Brief (Single Container)
 
 This application provides a voice-interactive morning market brief. It uses several AI agents for:
 - Speech-to-Text (STT)
@@ -21,6 +20,8 @@ This application provides a voice-interactive morning market brief. It uses seve
 - Language Generation (LLM)
 - Text-to-Speech (TTS)
 
+All services run within a single Docker container managed by Supervisord.
+
 ## How to Use
 1.  The application will start automatically once the Space is built.
 2.  Access the public URL provided by Hugging Face Spaces.
@@ -30,10 +31,10 @@ This application provides a voice-interactive morning market brief. It uses seve
 ## Environment Variables (Secrets)
 The following secrets **must be set in your Hugging Face Space settings** for the application to function correctly:
 
--   `FMP_API_KEY`: Your FinancialModelingPrep API key.
--   `ALPHAVANTAGE_API_KEY`: Your Alpha Vantage API key.
--   `GOOGLE_API_KEY`: Your Google API key for Gemini.
+-   `FMP_API_KEY`
+-   `ALPHAVANTAGE_API_KEY`
+-   `GOOGLE_API_KEY`
 -   `GEMINI_MODEL_NAME` (Optional): Defaults to `gemini-1.5-flash-latest` if not set.
 -   `WHISPER_MODEL_SIZE` (Optional): Defaults to `small` if not set.
 
-The `FAISS_INDEX_PATH` is configured internally to use `/app/faiss_index_store` and leverages a Docker named volume `faiss_index_volume` for persistence of the FAISS index during the Space's operational lifecycle.
+The `FAISS_INDEX_PATH` is configured internally to `/app/faiss_index_store`. Note that in this single-container setup on HF Spaces, this path might be ephemeral unless specific persistent storage is configured for the Space.

docker-compose.yaml

@@ -1,138 +0,0 @@
-version: '3.8'
-
-services:
-  api_agent:
-    build:
-      context: .
-      dockerfile: Dockerfile.fastapi
-    command: uvicorn agents.api_agent:app --host 0.0.0.0 --port 8001 
-    volumes:
-      - .:/app 
-    ports:
-      - "8001:8001"
-    environment:
-      - FMP_API_KEY=${FMP_API_KEY}
-      - ALPHAVANTAGE_API_KEY=${ALPHAVANTAGE_API_KEY}
-    networks:
-      - agent_network
-
-  scraping_agent:
-    build:
-      context: .
-      dockerfile: Dockerfile.fastapi
-    command: uvicorn agents.scraping_agent:app --host 0.0.0.0 --port 8002 
-    volumes:
-      - .:/app
-    ports:
-      - "8002:8002"
-    environment:
-      - FMP_API_KEY=${FMP_API_KEY}
-    networks:
-      - agent_network
-
-  retriever_agent:
-    build:
-      context: .
-      dockerfile: Dockerfile.fastapi
-    command: uvicorn agents.retriever_agent:app --host 0.0.0.0 --port 8003 
-    volumes:
-      - .:/app
-      - faiss_index_volume:/app/faiss_index_store 
-    ports:
-      - "8003:8003"
-    environment:
-      - FAISS_INDEX_PATH=/app/faiss_index_store 
-    networks:
-      - agent_network
-
-  analysis_agent:
-    build:
-      context: .
-      dockerfile: Dockerfile.fastapi
-    command: uvicorn agents.analysis_agent:app --host 0.0.0.0 --port 8004 
-    volumes:
-      - .:/app
-    ports:
-      - "8004:8004"
-    networks:
-      - agent_network
-
-  language_agent:
-    build:
-      context: .
-      dockerfile: Dockerfile.fastapi
-    command: uvicorn agents.language_agent:app --host 0.0.0.0 --port 8005 
-    volumes:
-      - .:/app
-    ports:
-      - "8005:8005"
-    environment:
-      - GOOGLE_API_KEY=${GOOGLE_API_KEY} 
-      - GEMINI_MODEL_NAME=${GEMINI_MODEL_NAME:-gemini-1.5-flash-latest}
-    networks:
-      - agent_network
-
-  voice_agent:
-    build:
-      context: .
-      dockerfile: Dockerfile.fastapi
-    command: uvicorn agents.voice_agent:app --host 0.0.0.0 --port 8006 
-    volumes:
-      - .:/app
-    ports:
-      - "8006:8006"
-    environment:
-      - WHISPER_MODEL_SIZE=${WHISPER_MODEL_SIZE:-small}
-    networks:
-      - agent_network
-
-  orchestrator:
-    build:
-      context: .
-      dockerfile: Dockerfile.fastapi
-    command: uvicorn orchestrator.main:app --host 0.0.0.0 --port 8000 
-    volumes:
-      - .:/app 
-      - ./example_portfolio.json:/app/example_portfolio.json 
-    ports:
-      - "8000:8000" 
-    environment:
-      - AGENT_API_URL=http://api_agent:8001
-      - AGENT_SCRAPING_URL=http://scraping_agent:8002
-      - AGENT_RETRIEVER_URL=http://retriever_agent:8003
-      - AGENT_ANALYSIS_URL=http://analysis_agent:8004
-      - AGENT_LANGUAGE_URL=http://language_agent:8005
-      - AGENT_VOICE_URL=http://voice_agent:8006
-    depends_on: 
-      - api_agent
-      - scraping_agent
-      - retriever_agent
-      - analysis_agent
-      - language_agent
-      - voice_agent
-    networks:
-      - agent_network
-
-  streamlit_app:
-    build:
-      context: .
-      dockerfile: Dockerfile.streamlit
-    command: streamlit run streamlit/app.py --server.port=8501 --server.address=0.0.0.0 --browser.gatherUsageStats=false
-    volumes:
-      - ./streamlit:/app/streamlit 
-      - ./example_portfolio.json:/app/example_portfolio.json 
-    ports:
-      - "8501:8501" 
-    environment:
-      - ORCHESTRATOR_URL=http://orchestrator:8000 
-    depends_on:
-      - orchestrator
-    networks:
-      - agent_network
-
-volumes:
-  faiss_index_volume: 
-
-networks:
-  agent_network: 
-    driver: bridge

requirements.txt

@@ -15,4 +15,5 @@ faster-whisper
 python-multipart 
 langgraph
 streamlit 
-streamlit-mic-recorder
+streamlit-mic-recorder
+supervisor

supervisord.conf

@@ -0,0 +1,82 @@
+[supervisord]
+nodaemon=true ; run supervisord in the foreground
+
+[program:api_agent]
+command=uvicorn agents.api_agent:app --host 0.0.0.0 --port 8001
+directory=/app
+autostart=true
+autorestart=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0
+
+[program:scraping_agent]
+command=uvicorn agents.scraping_agent:app --host 0.0.0.0 --port 8002
+directory=/app
+autostart=true
+autorestart=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0
+
+[program:retriever_agent]
+command=uvicorn agents.retriever_agent:app --host 0.0.0.0 --port 8003
+directory=/app
+autostart=true
+autorestart=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0
+
+[program:analysis_agent]
+command=uvicorn agents.analysis_agent:app --host 0.0.0.0 --port 8004
+directory=/app
+autostart=true
+autorestart=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0
+
+[program:language_agent]
+command=uvicorn agents.language_agent:app --host 0.0.0.0 --port 8005
+directory=/app
+autostart=true
+autorestart=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0
+
+[program:voice_agent]
+command=uvicorn agents.voice_agent:app --host 0.0.0.0 --port 8006
+directory=/app
+autostart=true
+autorestart=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0
+
+[program:orchestrator]
+command=uvicorn orchestrator.main:app --host 0.0.0.0 --port 8000
+directory=/app
+autostart=true
+autorestart=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0
+
+[program:streamlit_app]
+command=streamlit run streamlit/app.py --server.port=8501 --server.address=0.0.0.0 --browser.gatherUsageStats=false
+directory=/app
+autostart=true
+autorestart=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0