Fix the deployment space problem

Dockerfile

@@ -1,121 +1,36 @@
 FROM python:3.9-slim
 
-# Create user to avoid permission issues on HuggingFace Spaces
+# Create user to avoid permission issues on Hugging Face Spaces
 RUN useradd -m -u 1000 user
+
 WORKDIR /app
 
-# --- deps first (stable cache) ---
+# === Install dependencies ===
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 RUN pip install fastapi uvicorn python-multipart requests
 
-# --- copy backend pieces explicitly (no UI yet) ---
-# Copy only what backend needs (adjust these paths to your repo)
+# === Copy backend (app + model + data) ===
 COPY app.py ./app.py
 COPY scripts/ ./scripts/
 COPY checkpoints/ ./checkpoints/
 COPY data/ ./data/
 
-# --- generate FastAPI app (as you do now) ---
-RUN mkdir -p api
-COPY app.py api/app_gradio.py
-RUN echo 'import os\n\
-os.environ["MODEL_CHECKPOINT"] = "checkpoints/best_checkpoint.pth"\n\
-os.environ["MULTI_TASK"] = "true"\n\
-\n\
-import sys\n\
-from pathlib import Path\n\
-sys.path.append(str(Path(__file__).parent.parent))\n\
-\n\
-from fastapi import FastAPI, File, UploadFile, HTTPException\n\
-from fastapi.responses import JSONResponse, FileResponse, Response, HTMLResponse\n\
-from fastapi.middleware.cors import CORSMiddleware\n\
-from PIL import Image\n\
-import io\n\
-import json\n\
-import math\n\
-from datetime import datetime\n\
-from scripts.inference import ModelInference\n\
-\n\
-app = FastAPI(title=\"Decade Classifier API\")\n\
-\n\
-app.add_middleware(\n\
-    CORSMiddleware,\n\
-    allow_origins=[\"*\"],\n\
-    allow_credentials=True,\n\
-    allow_methods=[\"*\"],\n\
-    allow_headers=[\"*\"],\n\
-)\n\
-\n\
-model_inference = ModelInference(\n\
-    checkpoint_path=\"checkpoints/best_checkpoint.pth\",\n\
-    multi_task=True\n\
-)\n\
-\n\
-@app.get(\"/\", response_class=HTMLResponse)\n\
-async def root():\n\
-    html = Path(\"ui/index.html\").read_text(\"utf-8\")\n\
-    return HTMLResponse(html, headers={\"Cache-Control\": \"no-cache, no-store, must-revalidate\"})\n\
-\n\
-@app.get(\"/health\")\n\
-async def health_check():\n\
-    return {\n\
-        \"status\": \"healthy\",\n\
-        \"model_loaded\": True,\n\
-        \"model_info\": {\"model_name\": \"resnet50\", \"multi_task\": True}\n\
-    }\n\
-\n\
-@app.get(\"/training_data\")\n\
-async def get_training_data():\n\
-    with open(\"data/splits/train.json\", \"r\") as f:\n\
-        data = json.load(f)\n\
-    return {\"status\": \"success\", \"data\": data}\n\
-\n\
-@app.get(\"/proxy_image\")\n\
-async def proxy_image(url: str):\n\
-    import requests\n\
-    try:\n\
-        if url.startswith(\"http://\"):\n\
-            https_url = url.replace(\"http://\", \"https://\")\n\
-            try:\n\
-                resp = requests.get(https_url, timeout=5)\n\
-                if resp.status_code == 200:\n\
-                    return Response(content=resp.content, media_type=\"image/jpeg\")\n\
-            except:\n\
-                pass\n\
-        resp = requests.get(url, timeout=5)\n\
-        return Response(content=resp.content, media_type=\"image/jpeg\")\n\
-    except:\n\
-        return Response(status_code=404)\n\
-\n\
-@app.post(\"/predict\")\n\
-async def predict(file: UploadFile = File(...)):\n\
-    contents = await file.read()\n\
-    image = Image.open(io.BytesIO(contents)).convert(\"RGB\")\n\
-    temp_path = Path(f\"temp_{datetime.now().timestamp()}.jpg\")\n\
-    image.save(temp_path)\n\
-    try:\n\
-        result = model_inference.predict_image(temp_path)\n\
-        return JSONResponse({\n\
-            \"status\": \"success\",\n\
-            \"predictions\": result,\n\
-            \"timestamp\": datetime.now().isoformat(),\n\
-            \"model_info\": {\"model_name\": \"resnet50\", \"multi_task\": True}\n\
-        })\n\
-    finally:\n\
-        if temp_path.exists():\n\
-            temp_path.unlink()\n\
-' > api/app.py
+# === Copy API code ===
+COPY api/app.py api/app.py
 
-# --- copy UI LAST so any UI change invalidates cache ---
+# === Copy UI last (so UI updates bust cache) ===
 RUN mkdir -p ui
 COPY ui/ ./ui/
 
-# OPTIONAL: tiny cache-buster for the UI layer
+# Optional cache-buster so HF rebuilds UI layer
 RUN echo "UI_BUILD_TS=$(date +%s)" > /app/ui/.buildtag
 
-# Permissions & run
+# === Permissions & user ===
 RUN chown -R user:user /app
 USER user
+
 EXPOSE 7860
+
+# === Start FastAPI ===
 CMD ["uvicorn", "api.app:app", "--host", "0.0.0.0", "--port", "7860"]