Add the initial docker file

docker/README.md

@@ -0,0 +1,14 @@
+# Overview
+In this folder, we create the different dockerfiles for using pykoi.
+
+1. `pykoi-cpu`: The base image for the cpu-based usage.
+2. `pykoi-cpu-custom`: When you run this docker image, try to modify the `app.py` and mount it when running the docker container.
+
+To run a docker container, we can use the following command:
+```bash
+docker run -dp 5000:5000 -v $(pwd)/app.py:/app/app.py \
+        --name alex_test \
+        weialexchen/pykoi-cpu:app
+```
+
+Note that we need to keep the exposed port as 5000 (default value) to make the server work. 

docker/pykoi-cpu-custom/Dockerfile

@@ -0,0 +1,13 @@
+FROM weialexchen/pykoi-cpu:base
+
+# set the working directory in the container
+WORKDIR /app
+
+# copy the content
+COPY . /app
+
+# Expose the port on which the app will be running 
+EXPOSE 5000
+
+# For dev
+CMD ["python", "app.py"]

docker/pykoi-cpu-custom/app.py

@@ -0,0 +1,34 @@
+"""Demo for the chatbot application using OpenAI endpoint."""
+import pykoi
+
+##########################################################
+# Creating an OpenAI model (requires an OpenAI API key) #
+##########################################################
+# enter openai api key here
+api_key = "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"
+
+# Creating an OpenAI model
+model = pykoi.ModelFactory.create_model(
+    model_source="openai",
+    api_key=api_key)
+
+#####################################
+# Creating a chatbot with the model #
+#####################################
+database = pykoi.QuestionAnswerDatabase(debug=True)
+chatbot = pykoi.Chatbot(model=model, feedback="vote")
+# chatbot = pykoi.Chatbot(model=model, feedback="rank")
+dashboard = pykoi.Dashboard(database=database)
+
+###########################################################
+# Starting the application and add chatbot as a component #
+###########################################################
+# Create the application
+# app = pykoi.Application(debug=False, share=True)
+app = pykoi.Application(
+    host='0.0.0.0',
+    debug=False, 
+    share=True)
+app.add_component(chatbot)
+app.add_component(dashboard)
+app.run()

docker/pykoi-cpu/Dockerfile

@@ -0,0 +1,26 @@
+FROM python:3.10
+LABEL maintainer="Pykio Team"
+LABEL repository="Pykoi"
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN apt update && \
+    apt install -y bash \
+                   build-essential \
+                   git \
+                   git-lfs \
+                   curl \
+                   ca-certificates \
+                   libsndfile1-dev
+
+# make sure to use venv
+RUN python -m venv /opt/venv
+ENV PATH="/opt/venv/bin:$PATH"
+
+# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
+# follow the instructions here: https://cloud.google.com/tpu/docs/run-in-container#train_a_jax_model_in_a_docker_container
+RUN python -m pip install --no-cache-dir --upgrade pip && \
+    python -m pip install --upgrade --no-cache-dir \
+    pykoi
+
+CMD ["/bin/bash"]

docker/pykoi-gpu/Dockerfile

@@ -0,0 +1,43 @@
+FROM python:3.10
+LABEL maintainer="Pykio Team"
+LABEL repository="Pykoi"
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN apt update && \
+    apt install -y bash \
+                   build-essential \
+                   git \
+                   git-lfs \
+                   curl \
+                   ca-certificates \
+                   libsndfile1-dev \
+                   libgl1
+
+# make sure to use venv
+RUN python3 -m venv /opt/venv
+ENV PATH="/opt/venv/bin:$PATH"
+
+# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
+RUN python3 -m pip install --no-cache-dir --upgrade pip && \
+    python3 -m pip install --no-cache-dir \
+        torch \
+        torchvision \
+        torchaudio && \
+    python3 -m pip install --no-cache-dir \
+        accelerate \
+        datasets \
+        hf-doc-builder \
+        huggingface-hub \
+        Jinja2 \
+        librosa \
+        numpy \
+        scipy \
+        tensorboard \
+        transformers \
+        omegaconf \
+        pytorch-lightning \
+        xformers \
+        pykoi
+
+CMD ["/bin/bash"]

example/chatbot/openai_model_demo.py

@@ -5,7 +5,7 @@
 # Creating an OpenAI model (requires an OpenAI API key) #
 ##########################################################
 # enter openai api key here
-api_key = ""
+api_key = "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"
 
 # Creating an OpenAI model
 model = pykoi.ModelFactory.create_model(
@@ -25,7 +25,9 @@
 ###########################################################
 # Create the application
 # app = pykoi.Application(debug=False, share=True)
-app = pykoi.Application(debug=False, share=False)
+app = pykoi.Application(
+    debug=False, 
+    share=True)
 app.add_component(chatbot)
 app.add_component(dashboard)
 app.run()

gradio_test.py

@@ -0,0 +1,8 @@
+import gradio as gr
+
+def greet(name):
+    return "Hello " + name + "!"
+
+demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+
+demo.launch()