Feat: Dockerized the solution

Author: Ronnie-Leon76

Dockerfile

@@ -0,0 +1,38 @@
+# Stage 1: Build
+FROM --platform=linux/amd64 python:3.11.8-slim as build
+WORKDIR /app
+
+# Install build dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install build dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+
+COPY . .
+
+RUN chmod u+x setup.sh
+CMD ["./setup.sh"]
+
+# Install CPU-optimized PyTorch
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+
+# Install other requirements
+RUN pip install --no-cache-dir -r requirements.txt
+
+
+# Stage 2: Runtime
+FROM --platform=linux/amd64 python:3.11.8-slim
+WORKDIR /app
+
+# Copy only the necessary files from the build stage
+COPY --from=build /usr/local/lib/python3.11/site-packages /usr/local/lib/python3.11/site-packages
+COPY --from=build /app /app
+
+
+# Set entrypoint
+ENTRYPOINT ["uvicorn", "main:app", "--reload"]

main.py

@@ -199,3 +199,6 @@ async def get_powerbackup_quotation() -> Dict[str, Any]:
         return quotation
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to load quotation: {e}")
+    
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8080)

requirements.txt

@@ -111,8 +111,6 @@ tenacity==8.5.0
 tiktoken==0.7.0
 timm==1.0.8
 tokenizers==0.19.1
-torch==2.4.0
-torchvision==0.19.0
 tqdm==4.66.5
 transformers==4.44.1
 typing-inspect==0.9.0

setup.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+# Install libgl1
+sudo apt-get update && sudo apt-get install libgl1
+
+# Install tesseract-ocr
+sudo apt install tesseract-ocr -y
+
+# Install libtesseract-dev
+sudo apt install libtesseract-dev -y
+
+# Install poppler-utils
+sudo apt-get install poppler-utils -y