Vela commited on
Commit
411f074
·
1 Parent(s): f0cdbe5

modified docker file

Browse files
Dockerfile CHANGED
@@ -2,22 +2,34 @@ FROM python:3.11
2
 
3
  WORKDIR /app
4
 
5
- COPY . /app
 
6
 
7
- RUN pip install --upgrade pip
8
- RUN pip install -r requirements.txt
 
9
 
10
- EXPOSE 7860
 
11
 
12
- # Ensure logs directory exists
13
  RUN mkdir -p /app/logs && chmod -R 777 /app/logs
14
 
15
- # Install tmux to run multiple processes
16
- RUN apt-get update && apt-get install -y tmux
 
 
 
17
 
18
  # Set Python path
19
  ENV PYTHONPATH="/app/src"
20
 
21
- # Start FastAPI & Streamlit together using tmux
 
 
 
 
 
22
 
23
- CMD ["sh", "-c", "fastapi dev src/backend/main.py --host 0.0.0.0 --port 8000 & sleep 5 && streamlit run src/frontend/home.py --server.port 7860 --server.address 0.0.0.0"]
 
 
2
 
3
  WORKDIR /app
4
 
5
+ # Copy requirements first for better caching
6
+ COPY requirements.txt /app/
7
 
8
+ # Install dependencies and upgrade pip
9
+ RUN pip install --no-cache-dir --upgrade pip \
10
+ && pip install --no-cache-dir -r requirements.txt
11
 
12
+ # Copy remaining application code
13
+ COPY . /app
14
 
15
+ # Create logs directory with proper permissions
16
  RUN mkdir -p /app/logs && chmod -R 777 /app/logs
17
 
18
+ # Install additional dependencies
19
+ RUN apt-get update && apt-get install -y tmux curl
20
+
21
+ # Ensure the Hugging Face cache is set correctly
22
+ ENV TRANSFORMERS_CACHE="/app/.cache/huggingface"
23
 
24
  # Set Python path
25
  ENV PYTHONPATH="/app/src"
26
 
27
+ # Add a script to manage process control
28
+ COPY wait-for-it.sh /usr/local/bin/wait-for-it
29
+ RUN chmod +x /usr/local/bin/wait-for-it
30
+
31
+ # Expose ports for both FastAPI (8000) and Streamlit (7860)
32
+ EXPOSE 8000 7860
33
 
34
+ # Combined startup with better control
35
+ CMD ["sh", "-c", "tmux new-session -d -s backend 'uvicorn src.backend.main:app --host 0.0.0.0 --port 8000' && /usr/local/bin/wait-for-it localhost:8000 --timeout=30 && streamlit run src/frontend/home.py --server.port 7860 --server.address 0.0.0.0"]
chroma.log DELETED
@@ -1,20 +0,0 @@
1
- WARNING: [10-03-2025 20:07:32] chroma_server_nofile is not supported on Windows. chroma_server_nofile will not be set.
2
- WARNING: [10-03-2025 20:07:51] chroma_server_nofile is not supported on Windows. chroma_server_nofile will not be set.
3
- INFO: [10-03-2025 20:07:52] Anonymized telemetry enabled. See https://docs.trychroma.com/telemetry for more information.
4
- DEBUG: [10-03-2025 20:07:52] Starting component System
5
- DEBUG: [10-03-2025 20:07:52] Starting component OpenTelemetryClient
6
- DEBUG: [10-03-2025 20:07:52] Starting component SqliteDB
7
- DEBUG: [10-03-2025 20:07:52] Starting component SimpleQuotaEnforcer
8
- DEBUG: [10-03-2025 20:07:52] Starting component Posthog
9
- DEBUG: [10-03-2025 20:07:52] Starting component SimpleRateLimitEnforcer
10
- DEBUG: [10-03-2025 20:07:52] Starting component LocalSegmentManager
11
- DEBUG: [10-03-2025 20:07:52] Starting component LocalExecutor
12
- DEBUG: [10-03-2025 20:07:52] Starting component SegmentAPI
13
- DEBUG: [10-03-2025 20:07:52] Starting component SimpleAsyncRateLimitEnforcer
14
- INFO: [10-03-2025 20:07:52] Started server process [17592]
15
- INFO: [10-03-2025 20:07:52] Waiting for application startup.
16
- INFO: [10-03-2025 20:07:52] Application startup complete.
17
- INFO: [10-03-2025 20:07:52] Uvicorn running on http://localhost:8000 (Press CTRL+C to quit)
18
- INFO: [10-03-2025 20:16:05] ::1:57148 - "GET /api/v2/auth/identity HTTP/1.1" 200
19
- INFO: [10-03-2025 20:16:05] ::1:57149 - "GET /api/v2/tenants/default_tenant HTTP/1.1" 200
20
- INFO: [10-03-2025 20:16:05] ::1:57149 - "GET /api/v2/tenants/default_tenant/databases/default_database HTTP/1.1" 200
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
requirements.txt CHANGED
@@ -8,6 +8,4 @@ torch
8
  torchvision
9
  torchaudio
10
  sentence_transformers
11
- groq
12
- chromadb
13
- uvicorn
 
8
  torchvision
9
  torchaudio
10
  sentence_transformers
11
+ groq
 
 
src/backend/models/__pycache__/embedding_model.cpython-313.pyc CHANGED
Binary files a/src/backend/models/__pycache__/embedding_model.cpython-313.pyc and b/src/backend/models/__pycache__/embedding_model.cpython-313.pyc differ
 
src/backend/models/embedding_model.py CHANGED
@@ -1,11 +1,9 @@
1
  from sentence_transformers import SentenceTransformer
2
  from utils import logger
3
- from fastapi import UploadFile
4
- from PIL import Image
5
 
6
  logger = logger.get_logger()
7
 
8
- model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
9
 
10
  def get_text_embedding(search_query: str):
11
  try:
 
1
  from sentence_transformers import SentenceTransformer
2
  from utils import logger
 
 
3
 
4
  logger = logger.get_logger()
5
 
6
+ model = SentenceTransformer("all-MiniLM-L6-v2")
7
 
8
  def get_text_embedding(search_query: str):
9
  try: