Update: Finalize Docker configuration and paths

Dockerfile

@@ -22,13 +22,15 @@ RUN apt-get update && \
     libglib2.0-0 \
     git \
     && apt-get clean && \
-    rm -rf /var/lib/apt/lists/* && \
-    rm -rf /tmp/* && \
-    mkdir -p /tmp && \
-    chmod 777 /tmp
+    rm -rf /var/lib/apt/lists/*
 
-# Create cache directories with proper permissions
-RUN mkdir -p /home/user/.cache && \
+# Create all necessary directories with proper permissions
+RUN mkdir -p /tmp/uploads && \
+    mkdir -p /tmp/results && \
+    mkdir -p /tmp/models/parts/weights/weights && \
+    chmod -R 777 /tmp && \
+    chown -R user:user /tmp && \
+    mkdir -p /home/user/.cache && \
     mkdir -p /home/user/huggingface && \
     mkdir -p /home/user/ultralytics && \
     chown -R user:user /home/user/.cache && \
@@ -38,19 +40,10 @@ RUN mkdir -p /home/user/.cache && \
 # Copy requirements first
 COPY requirements.txt .
 
-# Install dependencies with improved error handling
+# Install dependencies
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir setuptools wheel && \
-    pip install --no-cache-dir -r requirements.txt || (echo "pip install failed" && exit 1) && \
-    rm -rf ~/.cache/pip && \
-    rm -rf /tmp/* && \
-    mkdir -p /tmp && \
-    chmod 777 /tmp
-
-# Create necessary directories with proper permissions
-RUN mkdir -p /tmp/models/parts/weights/weights && \
-    chmod -R 777 /tmp && \
-    chown -R user:user /tmp/models
+    pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY --chown=user:user . .

app/main.py

@@ -47,20 +47,32 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Mount static files
-app.mount("/static", StaticFiles(directory="app/static"), name="static")
-templates = Jinja2Templates(directory="app/templates")
+# Get absolute paths for static and template directories
+APP_DIR = Path(__file__).resolve().parent
+STATIC_DIR = APP_DIR / "static"
+TEMPLATES_DIR = APP_DIR / "templates"
+
+# Mount static files using absolute path
+app.mount("/static", StaticFiles(directory=str(STATIC_DIR)), name="static")
+templates = Jinja2Templates(directory=str(TEMPLATES_DIR))
 
 # Include routers
 app.include_router(inference.router)
 
-# Create required directories
-UPLOAD_DIR = Path("tmp/uploads")
-RESULTS_DIR = Path("tmp/results")
+# Create required directories with absolute paths
+UPLOAD_DIR = Path("/tmp/uploads")
+RESULTS_DIR = Path("/tmp/results")
 
 def create_dirs():
+    # Ensure directories exist with proper permissions
     for dir_path in [UPLOAD_DIR, RESULTS_DIR]:
-        dir_path.mkdir(parents=True, exist_ok=True)
+        try:
+            dir_path.mkdir(parents=True, exist_ok=True)
+            # Set permissions to 777 to ensure write access
+            os.chmod(str(dir_path), 0o777)
+        except Exception as e:
+            print(f"Error creating directory {dir_path}: {e}")
+            raise
 
 # Add to startup
 @app.on_event("startup")
@@ -68,4 +80,4 @@ async def startup_event():
     create_dirs()
 
 if __name__ == "__main__":
-    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)
+    uvicorn.run("main:app", host="0.0.0.0", port=7860)