feat: first shot at dockerizing app.py

.dockerignore

@@ -0,0 +1,16 @@
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+.Python
+.env
+.env.*
+*.sqlite
+*.db
+output/
+.git/
+.gitignore
+.dockerignore
+Dockerfile
+*.md
+!README.md

Dockerfile

@@ -0,0 +1,51 @@
+# Compile stage: install build dependencies
+FROM python:3.12-slim AS builder
+
+# Set working directory
+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 Python dependencies
+COPY requirements.txt .
+RUN python -m pip install --upgrade pip && \
+    python -m pip install --no-cache-dir wheel && \
+    pip wheel --no-cache-dir --no-deps --wheel-dir /app/wheels -r requirements.txt
+
+# Final stage: copy only necessary files
+FROM python:3.12-slim
+
+WORKDIR /app
+
+# Copy wheels from builder stage
+COPY --from=builder /app/wheels /wheels
+
+# Install Python packages from wheels
+RUN pip install --no-cache /wheels/*
+
+# Copy application files
+COPY app.py flux.py stablediffusion.py ./
+COPY pyproject.toml README.md ./
+
+# Create directories for temporary files
+RUN mkdir -p /tmp/flux /tmp/flux2 && \
+    chmod 777 /tmp/flux /tmp/flux2
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONPATH=/app
+
+# Copy and set permissions for entrypoint script
+COPY entrypoint.sh .
+RUN chmod +x entrypoint.sh
+
+# Expose the port that Gradio will run on
+EXPOSE 7860
+
+# Command to run the application
+ENTRYPOINT ["./entrypoint.sh"]

app.py

@@ -236,6 +236,8 @@ with gr.Blocks() as demo:
 demo.launch(
     mcp_server=True,
     share=True,
-    ssr_mode=False)
+    ssr_mode=False,
+    server_name="0.0.0.0",  # Listen on all interfaces
+    server_port=7860)  # Use the port we exposed in Dockerfile
 
 # EOF

docker-compose.yaml

@@ -0,0 +1,19 @@
+version: '3.8'
+
+services:
+  simple-calculator:
+    build: .
+    ports:
+      - "7860:7860"
+    environment:
+      - PYTHONUNBUFFERED=1
+      # Add your Modal credentials here or use .env file
+      - MODAL_TOKEN_ID=${MODAL_TOKEN_ID:-}
+      - MODAL_TOKEN_SECRET=${MODAL_TOKEN_SECRET:-}
+    volumes:
+      # Mount the Modal config directory to persist credentials
+      - modal-config:/root/.modal
+    restart: unless-stopped
+
+volumes:
+  modal-config: # Named volume for Modal configuration

entrypoint.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+# Create Modal config directory
+mkdir -p ~/.modal
+
+# Write Modal token configuration if provided
+if [ ! -z "$MODAL_TOKEN_ID" ] && [ ! -z "$MODAL_TOKEN_SECRET" ]; then
+    echo "{\"token_id\":\"$MODAL_TOKEN_ID\",\"token_secret\":\"$MODAL_TOKEN_SECRET\"}" > ~/.modal/config.json
+fi
+
+# Start the application
+exec python app.py

env.example

@@ -0,0 +1,3 @@
+# Copy this file to .env and fill in your Modal credentials
+MODAL_TOKEN_ID=your_modal_token_id_here
+MODAL_TOKEN_SECRET=your_modal_token_secret_here

flux.py

@@ -2,13 +2,27 @@
 #
 # See https://modal.com/docs/examples/flux
 
-# Setting up the image and dependencies
-
+import os
+import sys
 import time
 from io import BytesIO
 from pathlib import Path
 import modal
-import modal.running_app
+# import modal.running_app
+
+# Check for Modal credentials
+MODAL_TOKEN_ID = os.getenv('MODAL_TOKEN_ID')
+MODAL_TOKEN_SECRET = os.getenv('MODAL_TOKEN_SECRET')
+
+if not MODAL_TOKEN_ID or not MODAL_TOKEN_SECRET:
+    print("WARNING: Modal credentials not found. Image generation will return placeholder images.", file=sys.stderr)
+    MODAL_AVAILABLE = False
+else:
+    try:
+        MODAL_AVAILABLE = True
+    except ImportError:
+        print("WARNING: Modal package not available. Image generation will return placeholder images.", file=sys.stderr)
+        MODAL_AVAILABLE = False
 
 # We’ll make use of the full CUDA toolkit in this example, so we’ll build our container image
 # off of the nvidia/cuda base.
@@ -284,6 +298,10 @@ def generate_image2(prompt: str):
     """
 
     print("DEBUG: generate_image2 called with prompt:", prompt)
+    
+    if not MODAL_AVAILABLE:
+        print("DEBUG: Modal not available, returning placeholder image")
+        return generate_image1(prompt)
 
     if prompt is None or prompt.strip() == "A portrait of a handsome software developer":
         print("DEBUG: Returning hardcoded image URL for default prompt")