From d9711e2a6ab2613dcb1f3003e61f33239980122e Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Tue, 18 Jul 2023 17:04:57 -0400
Subject: [PATCH] clean nvidia

---
 infra/Docker/Dockerfile.gpu | 39 +++++++++++++++++++++++++------------
 1 file changed, 27 insertions(+), 12 deletions(-)

diff --git a/infra/Docker/Dockerfile.gpu b/infra/Docker/Dockerfile.gpu
index 33e9c430..dacc4211 100644
--- a/infra/Docker/Dockerfile.gpu
+++ b/infra/Docker/Dockerfile.gpu
@@ -1,7 +1,27 @@
+# Use an official Python runtime as a parent image
 FROM nvidia/cuda:11.7.0-runtime-ubuntu20.04
 
-# Set working directory
-WORKDIR /app/
+# Set environment variables
+ENV EVAL_PORT=8000 \
+    MODEL_NAME=gpt-4 \
+    CELERY_BROKER_URL=redis://localhost:6379 \
+    SERVER=http://localhost:${EVAL_PORT} \
+    USE_GPU=True \
+    PLAYGROUND_DIR=playground \
+    LOG_LEVEL=INFO \
+    BOT_NAME=Orca \
+    # You will need to set these environment variables to your actual keys in production
+    OPENAI_API_KEY=your_openai_api_key \
+    WINEDB_HOST=your_winedb_host \
+    WINEDB_PASSWORD=your_winedb_password \
+    BING_SEARCH_URL=your_bing_search_url \
+    BING_SUBSCRIPTION_KEY=your_bing_subscription_key \
+    SERPAPI_API_KEY=your_serpapi_api_key \
+    REDIS_HOST=your_redis_host \
+    REDIS_PORT=your_redis_port
+
+# Set work directory
+WORKDIR /usr/src/app
 
 # Install system dependencies
 ENV DEBIAN_FRONTEND=noninteractive
@@ -11,23 +31,18 @@ RUN apt-get update && \
     apt-get install -y python3.10 python3-pip curl && \
     apt-get install -y nodejs npm
 
-# Set environment variables
-ENV PATH "/root/.local/bin:$PATH"
-ENV PORT 8001
-
-# Copy project files
-COPY requirements.txt .
+# Add requirements file
+COPY requirements.txt ./
 
-# Install project dependencies
+# Install any needed packages specified in requirements.txt
 RUN python3.10 -m pip install --upgrade pip && \
     python3.10 -m pip install --no-cache-dir -r requirements.txt
 
-# Copy application files
+# Bundle app source
 COPY . .
 
-
 # Expose port
-# EXPOSE ${EVAL_PORT}
+EXPOSE ${EVAL_PORT}
 
 # Run the application
 CMD ["uvicorn", "api.app:app", "--host", "0.0.0.0", "--port", "8000"]
\ No newline at end of file