Upload 4 files

docker-compose.yml

@@ -1,18 +1,43 @@
 version: '3.8'
 
 services:
-  app:
-    build:
-      context: .
-      dockerfile: Dockerfile
-    container_name: ai-assistant
+  bloom-ai:
+    build: .
+    container_name: bloom-ai-assistant
+    volumes:
+      - ./app:/app
+      - model-cache:/app/cache
     ports:
-      - "7860:7860"
+      - "8000:8000"
     environment:
-      - TELEGRAM_TOKEN=7437859619:AAGeGG3ZkLM0OVaw-Exx1uMRE55JtBCZZCY  # Replace with your actual token
-      - CHAT_ID=-1002228627548                                         # Replace with your actual chat ID
-      - MODEL_NAME=lifeweb-ai/shiraz                                   # Model name
-      - PORT=7860                                                     # Application port
+      - HF_HOME=/app/cache
+      - TRANSFORMERS_CACHE=/app/cache
+      - TORCH_HOME=/app/cache/torch
+      - PYTHONUNBUFFERED=1
+      - PYTHONDONTWRITEBYTECODE=1
+    restart: unless-stopped
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:8000/"]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 40s
+
+  prometheus:
+    image: prom/prometheus:latest
+    container_name: bloom-ai-prometheus
     volumes:
-      - ./static:/app/static
+      - ./prometheus.yml:/etc/prometheus/prometheus.yml
+      - prometheus-data:/prometheus
+    ports:
+      - "9090:9090"
+    command:
+      - '--config.file=/etc/prometheus/prometheus.yml'
+      - '--storage.tsdb.path=/prometheus'
+      - '--web.console.libraries=/usr/share/prometheus/console_libraries'
+      - '--web.console.templates=/usr/share/prometheus/consoles'
     restart: unless-stopped
+
+volumes:
+  model-cache:
+  prometheus-data:

dockerfile

@@ -1,31 +1,37 @@
 FROM python:3.10-slim
 
-WORKDIR /app
+# Set environment variables
+ENV PYTHONUNBUFFERED=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    HF_HOME=/app/cache \
+    TRANSFORMERS_CACHE=/app/cache \
+    TORCH_HOME=/app/cache/torch
 
 # Install system dependencies
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends \
+RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
+    curl \
+    git \
+    libpq-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Create cache directory with proper permissions
-RUN mkdir -p /app/.cache/huggingface && \
-    chmod -R 777 /app/.cache
-
-# Set environment variable
-ENV HF_HOME=/app/.cache/huggingface
+# Set working directory
+WORKDIR /app
 
-# Copy requirements first
+# Install Python dependencies first
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu && \
+    pip install --no-cache-dir -r requirements.txt
 
-# Create necessary directories
-RUN mkdir -p /app/static && \
-    chmod -R 755 /app/static
+# Copy project files
+COPY . .
 
-# Copy application code
-COPY app/ .
+# Create necessary directories
+RUN mkdir -p /app/cache /app/logs /app/static /app/templates && \
+    chmod -R 777 /app
 
-EXPOSE 7860
+# Expose port
+EXPOSE 8000
 
-CMD ["python", "app.py"]
+# Start command
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]

prometheus-config.json

@@ -0,0 +1,9 @@
+global:
+  scrape_interval: 15s
+  evaluation_interval: 15s
+
+scrape_configs:
+  - job_name: 'bloom-ai'
+    static_configs:
+      - targets: ['bloom-ai:8000']
+    metrics_path: '/metrics'

requirements.txt

@@ -4,20 +4,14 @@ uvicorn[standard]>=0.24.0
 python-multipart>=0.0.6
 jinja2>=3.1.2
 
-# Telegram and Communication
-python-telegram-bot>=20.7
-httpx[http2,socks]>=0.25.2
-websockets>=12.0
-discord.py>=2.3.2
-
 # AI and Machine Learning
-torch
-transformers
-sentence-transformers
+torch==2.1.1+cpu
+transformers==4.35.2
+sentence-transformers==2.2.2
 accelerate>=0.26.0
-protobuf
-sentencepiece
-hazm
+hazm==0.7.0
+protobuf==4.25.1
+sentencepiece==0.1.99
 
 # Async and Network
 aiofiles>=23.2.1
@@ -40,28 +34,12 @@ plotly>=5.18.0
 pillow>=10.1.0
 bs4>=0.0.1
 
-# Database and Authentication
-sqlalchemy>=2.0.23
-python-jose[cryptography]>=3.3.0
-passlib[bcrypt]>=1.7.4
-social-auth-core>=4.5.0
-
-# Task Scheduling and Management
-apscheduler>=3.10.4
-eventlet>=0.33.3
-
-# UI and Visualization
-dash>=2.14.1
-gradio>=4.7.1
-flask-socketio>=5.3.6
-
 # Environment and Configuration
 python-dotenv>=1.0.0
-
-# Optional but Recommended
 pydantic>=2.5.2
-python-jose>=3.3.0
-bcrypt>=4.0.1
-cryptography>=41.0.5
-cachetools
-psutil
+
+# HTTP Client
+httpx[http2,socks]>=0.25.2
+
+# Cache and Storage
+cachetools>=5.3.2