Docker configs

.github/workflows/huggingFace.yml

@@ -0,0 +1,19 @@
+name: Sync to Hugging Face hub
+on:
+  push:
+    branches: [main]
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+          lfs: true
+      - name: Push to HF Space
+        env:
+          HF_API_KEY: ${{ secrets.HF_API_KEY }}
+        run: |
+          git push https://HF_USERNAME:${HF_API_KEY}@huggingface.co/spaces/SCGR/Promptaid_Vision main

README.md

@@ -1,8 +1,45 @@
-# PromptAid-Vision
-UCL IXN summer project - IFRC
+---
+title: PromptAid Vision
+emoji: 🚀
+colorFrom: blue
+colorTo: red
+sdk: docker
+app_port: 8080
+---
 
-System architecture
-React => Python => postgreSQL => docker => fly.io/hugging face
+# PromptAid Vision
 
-VLMs
-GPT-4O, Gemini, Claude, hugging face inference api (llava, Blip)
+AI-powered crisis map analysis platform using Vision Language Models.
+
+## Features
+
+- **Crisis map/drone imagery analysis**
+- **Explore & export database**
+- **Analytics dashboard**
+
+## Tech Stack
+
+- **Backend**: FastAPI, Python 3.11
+- **AI Models**: OpenAI GPT-4V, Google Gemini, Hugging Face models
+- **Database**: PostgreSQL
+- **Storage**: S3-compatible storage
+- **Frontend**: React with TypeScript
+
+## Environment Variables
+
+- `DATABASE_URL`: PostgreSQL connection string
+- `S3_ENDPOINT`: S3-compatible storage endpoint
+- `S3_ACCESS_KEY`: Storage access key
+- `S3_SECRET_KEY`: Storage secret key
+- `S3_BUCKET`: Storage bucket name
+- `OPENAI_API_KEY`: OpenAI API key
+- `GOOGLE_API_KEY`: Google API key
+- `HF_API_KEY`: Hugging Face API key
+
+## API Endpoints
+
+- `/api/images` - Image management
+- `/api/captions` - AI caption generation
+- `/api/metadata` - Metadata operations
+- `/api/models` - Available AI models
+- `/api/contribute` - Contribution system

deploy.sh

@@ -0,0 +1,55 @@
+#!/bin/bash
+
+echo "🚀 PromptAid Vision Deployment Script"
+echo "====================================="
+
+echo "📦 Building backend Docker image..."
+cd py_backend
+docker build -t promptaid-vision-backend .
+
+if [ $? -eq 0 ]; then
+    echo "✅ Backend image built successfully!"
+else
+    echo "❌ Backend image build failed!"
+    exit 1
+fi
+
+cd ..
+
+echo "🧪 Testing production setup locally..."
+docker-compose -f docker-compose.prod.yml up -d
+
+echo "⏳ Waiting for services to start..."
+sleep 10
+
+echo "🔍 Checking backend health..."
+response=$(curl -s -o /dev/null -w "%{http_code}" http://localhost:8080/api/models)
+
+if [ $response -eq 200 ]; then
+    echo "✅ Backend is healthy and responding!"
+else
+    echo "❌ Backend health check failed (HTTP $response)"
+    echo "📋 Stopping services..."
+    docker-compose -f docker-compose.prod.yml down
+    exit 1
+fi
+
+echo "📋 Stopping test services..."
+docker-compose -f docker-compose.prod.yml down
+
+echo ""
+echo "🎉 Deployment preparation completed!"
+echo ""
+echo "📋 Next steps:"
+echo "1. Set up cloud database (PostgreSQL)"
+echo "2. Set up cloud storage (S3-compatible)"
+echo "3. Configure environment variables in Hugging Face Spaces"
+echo "4. Push your code to Hugging Face Spaces"
+echo ""
+echo "🔧 To test locally again, run:"
+echo "   docker-compose -f docker-compose.prod.yml up"
+echo ""
+echo "🌍 To deploy to Hugging Face Spaces:"
+echo "   git add ."
+echo "   git commit -m 'Prepare for Hugging Face Spaces deployment'"
+echo "   git push"

docker-compose.prod.yml

@@ -0,0 +1,48 @@
+version: "3.8"
+
+services:
+  backend:
+    build: ./py_backend
+    ports:
+      - "8080:8080"
+    environment:
+      - DATABASE_URL=postgresql://promptaid:promptaid@postgres:5432/promptaid
+      - S3_ENDPOINT=http://minio:9000
+      - S3_ACCESS_KEY=promptaid
+      - S3_SECRET_KEY=promptaid
+      - S3_BUCKET=promptaid
+      - ENVIRONMENT=production
+      - BASE_URL=http://localhost:8080
+    depends_on:
+      - postgres
+      - minio
+    restart: unless-stopped
+
+  postgres:
+    image: postgres:16
+    restart: always
+    environment:
+      POSTGRES_USER: promptaid
+      POSTGRES_PASSWORD: promptaid
+      POSTGRES_DB: promptaid
+    ports:
+      - "5433:5432"
+    volumes:
+      - pgdata:/var/lib/postgresql/data
+
+  minio:
+    image: minio/minio:latest
+    restart: always
+    command: server /data --console-address ":9001"
+    environment:
+      MINIO_ROOT_USER: promptaid
+      MINIO_ROOT_PASSWORD: promptaid
+    ports:
+      - "9000:9000"
+      - "9001:9001"
+    volumes:
+      - minio_data:/data
+
+volumes:
+  pgdata:
+  minio_data:

frontend/src/pages/UploadPage/UploadPage.tsx

@@ -480,14 +480,15 @@ export default function UploadPage() {
 
   return (
     <PageContainer>
-      <Container
-        heading="Upload Your Image"
-        headingLevel={2}
-        withHeaderBorder
-        withInternalPadding
-        className="max-w-7xl mx-auto"
-      >
-        <div className={styles.uploadContainer} data-step={step}>
+      {step !== 3 && (
+        <Container
+          heading="Upload Your Image"
+          headingLevel={2}
+          withHeaderBorder
+          withInternalPadding
+          className="max-w-7xl mx-auto"
+        >
+          <div className={styles.uploadContainer} data-step={step}>
           {/* Drop-zone */}
           {step === 1 && !searchParams.get('step') && (
             <div className="space-y-6">
@@ -829,22 +830,6 @@ export default function UploadPage() {
           </div>
         )}
 
-        {/* Success page */}
-        {step === 3 && (
-          <div className={styles.successContainer}>
-            <Heading level={2} className={styles.successHeading}>Saved!</Heading>
-            <p className={styles.successText}>Your caption has been successfully saved.</p>
-            <div className={styles.successButton}>
-              <Button
-                name="upload-another"
-                onClick={resetToStep1}
-              >
-                Upload Another
-              </Button>
-            </div>
-          </div>
-        )}
-
         {/* Full Size Image Modal */}
         {isFullSizeModalOpen && (
           <div className={styles.fullSizeModalOverlay} onClick={() => setIsFullSizeModalOpen(false)}>
@@ -870,7 +855,24 @@ export default function UploadPage() {
         )}
 
         </div>
-      </Container>
+        </Container>
+      )}
+
+      {/* Success page - outside the upload container */}
+      {step === 3 && (
+        <div className={styles.successContainer}>
+          <Heading level={2} className={styles.successHeading}>Saved!</Heading>
+          <p className={styles.successText}>Your caption has been successfully saved.</p>
+          <div className={styles.successButton}>
+            <Button
+              name="upload-another"
+              onClick={resetToStep1}
+            >
+              Upload Another
+            </Button>
+          </div>
+        </div>
+      )}
     </PageContainer>
   );
 }

py_backend/.dockerignore

@@ -0,0 +1,27 @@
+__pycache__
+*.pyc
+*.pyo
+*.pyd
+.Python
+env
+pip-log.txt
+pip-delete-this-directory.txt
+.tox
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.log
+.git
+.mypy_cache
+.pytest_cache
+.hypothesis
+.env
+.venv
+venv/
+tests/
+*.md
+.gitignore
+README.md

py_backend/Dockerfile

@@ -0,0 +1,19 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    gcc \
+    postgresql-client \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+RUN mkdir -p /app/logs
+
+EXPOSE 8080
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8080"]

py_backend/app/config.py

@@ -1,17 +1,19 @@
 from pydantic_settings import BaseSettings
 
 class Settings(BaseSettings):
-    DATABASE_URL: str
-    S3_ENDPOINT: str
-    S3_ACCESS_KEY: str
-    S3_SECRET_KEY: str
-    S3_BUCKET: str
-    
+    DATABASE_URL: str = "postgresql://user:pass@host:5432/dbname"
+    S3_ENDPOINT: str = "https://your-s3-endpoint.com"
+    S3_ACCESS_KEY: str = ""
+    S3_SECRET_KEY: str = ""
+    S3_BUCKET: str = ""
     OPENAI_API_KEY: str = ""
     ANTHROPIC_API_KEY: str = ""
     GOOGLE_API_KEY: str = ""
     HF_API_KEY: str = ""
-
+    SPACE_ID: str = ""
+    ENVIRONMENT: str = "development"
+    BASE_URL: str = "http://localhost:8080"
+    
     class Config:
         env_file = ".env"
         env_file_encoding = "utf-8-sig"

py_backend/app/main.py

@@ -4,15 +4,17 @@ from app.routers import upload, caption, metadata, models
 from app.config import settings
 from app.routers.images import router as images_router
 
-
-
 app = FastAPI(title="PromptAid Vision")
 
 app.add_middleware(
     CORSMiddleware,
     allow_origins=[
         "http://localhost:3000",
-        "http://localhost:5173"
+        "http://localhost:5173",
+        "https://huggingface.co",
+        "https://*.hf.space",
+        "https://*.hf.space",
+        "*"
     ],
     allow_credentials=True,
     allow_methods=["*"],
@@ -27,6 +29,8 @@ app.include_router(images_router,  prefix="/api/contribute", tags=["contribute"]
 
 print("🚀 PromptAid Vision API server ready")
 print("📊 Available endpoints: /api/images, /api/captions, /api/metadata, /api/models")
+print(f"🌍 Environment: {settings.ENVIRONMENT}")
+print("🔗 CORS enabled for Hugging Face Spaces")
 
 
 

py_backend/app/routers/caption.py

@@ -3,12 +3,12 @@ from sqlalchemy.orm import Session
 from typing import List
 from .. import crud, database, schemas, storage
 from ..services.vlm_service import vlm_manager
+from ..config import settings
 
 from ..services.stub_vlm_service import StubVLMService
 from ..services.gpt4v_service import GPT4VService
 from ..services.gemini_service import GeminiService
 from ..services.huggingface_service import LLaVAService, BLIP2Service, InstructBLIPService
-from ..config import settings
 
 stub_service = StubVLMService()
 vlm_manager.register_service(stub_service)
@@ -129,6 +129,9 @@ async def create_caption(
     except Exception:
         url = f"/api/images/{c.image_id}/file"
     
+    if url and url.startswith('/'):
+        url = f"{settings.BASE_URL}{url}"
+    
     img_dict = convert_image_to_dict(c, url)
     return schemas.ImageOut(**img_dict)
 
@@ -152,6 +155,10 @@ def get_caption(
     except Exception:
         url = f"/api/images/{caption.image_id}/file"
     
+
+    if url and url.startswith('/'):
+        url = f"{settings.BASE_URL}{url}"
+    
     img_dict = convert_image_to_dict(caption, url)
     return schemas.ImageOut(**img_dict)
 
@@ -176,6 +183,9 @@ def get_captions_by_image(
         except Exception:
             url = f"/api/images/{caption.image_id}/file"
         
+        if url and url.startswith('/'):
+            url = f"{settings.BASE_URL}{url}"
+        
         img_dict = convert_image_to_dict(caption, url)
         result.append(schemas.ImageOut(**img_dict))
     
@@ -232,6 +242,9 @@ def update_caption(
     except Exception:
         url = f"/api/images/{caption.image_id}/file"
     
+    if url and url.startswith('/'):
+        url = f"{settings.BASE_URL}{url}"
+    
     img_dict = convert_image_to_dict(caption, url)
     return schemas.ImageOut(**img_dict)
 

py_backend/app/routers/upload.py

@@ -3,6 +3,7 @@ from pydantic import BaseModel
 import io
 from sqlalchemy.orm import Session
 from .. import crud, schemas, storage, database
+from ..config import settings
 from typing import List
 import boto3
 import time
@@ -35,6 +36,9 @@ def convert_image_to_dict(img, image_url):
             print(f"Warning: Error processing countries for image {img.image_id}: {e}")
             countries_list = []
     
+    if image_url and image_url.startswith('/'):
+        image_url = f"{settings.BASE_URL}{image_url}"
+    
     img_dict = {
         "image_id": img.image_id,
         "file_key": img.file_key,

py_backend/requirements.txt

@@ -19,3 +19,4 @@ httpx
 requests
 pycountry>=22.3.5
 pycountry-convert>=0.7.2
+python-multipart