Update Dockerfile to optimize application setup and streamline dependencies

ai.py

@@ -109,4 +109,5 @@ def generate_cube(user_prompt: str) -> CubeAI:
     )
     chain = prompt | model
     response = chain.invoke({"user_prompt": user_prompt})
+    print(response.model_dump(mode='json'))
     return response

main.py

@@ -11,6 +11,7 @@ import cv2
 import numpy as np
 from PIL import Image
 from fastapi import FastAPI, HTTPException, UploadFile, File
+from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 from starlette.middleware.cors import CORSMiddleware
 
@@ -26,6 +27,8 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
 
 class LUTRequest(BaseModel):
     file_id: str
@@ -390,7 +393,7 @@ async def transform_lut(request: LUTTransformRequest):
                     status_code=500, detail="Failed to save new cube file"
                 )
 
-            sample_image_path = "sample.jpg"
+            sample_image_path = "static/sample.jpg"
             if not os.path.exists(sample_image_path):
                 raise HTTPException(status_code=404, detail="Sample image not found")
 
@@ -418,4 +421,4 @@ async def transform_lut(request: LUTTransformRequest):
 
 @app.get("/health")
 async def health_check():
-    return {"status": "healthy", "sample_image_exists": os.path.exists("sample.jpg")}
+    return {"status": "healthy", "sample_image_exists": os.path.exists("static/sample.jpg")}