Add: diffsketcher Dockerfile with original implementation

Dockerfile

@@ -9,22 +9,35 @@ RUN apt-get update && apt-get install -y \
     git \
     libcairo2-dev \
     pkg-config \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
+# Clone the repository
+RUN git clone https://github.com/ximinng/DiffSketcher.git /code/diffsketcher
+
 # Install PyTorch and torchvision
 RUN pip install torch==2.0.0 torchvision==0.15.1 --extra-index-url https://download.pytorch.org/whl/cpu
 
-# Install cairosvg and other dependencies
-RUN pip install cairosvg cairocffi cssselect2 defusedxml tinycss2
+# Install dependencies
+WORKDIR /code/diffsketcher
+RUN pip install -r requirements.txt
+
+# Install diffvg
+RUN git clone https://github.com/BachiLi/diffvg.git /code/diffvg && \
+    cd /code/diffvg && \
+    git submodule update --init --recursive && \
+    python setup.py install
 
-# Install FastAPI and other dependencies
-RUN pip install fastapi uvicorn pydantic pillow numpy requests
+# Install additional dependencies
+RUN pip install cairosvg cairocffi cssselect2 defusedxml tinycss2 fastapi uvicorn pydantic
 
-# Copy the model files
-COPY . /code/
+# Copy the handler and API
+COPY handler.py /code/
+COPY api.py /code/
 
 # Set environment variables
-ENV PYTHONUNBUFFERED=1
+ENV PYTHONPATH=/code:/code/diffsketcher:/code/diffvg
+ENV MODEL_DIR=/code/diffsketcher
 
 # Run the API server
-CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "8000"]