Merge pull request #3 from rtuszik/dev

docs: update README and Docker configuration

README.md

@@ -1,21 +1,18 @@
 # Flux Replicate GUI
-
 A simple web interface for running Flux models using the Replicate API. Use it to generate images with custom LoRAs and fine-tuned Flux models.
 
 ## What it does
-
 - Runs Flux models via Replicate API
 - Lets you use custom LoRAs and fine-tuned models
 - Allows disabling the Safety Checker
 - Saves your settings
 - Shows generated images in a gallery
 
 ## Setup
-
 1. Clone the repo:
    ```
-   git clone https://github.com/yourusername/flux-replicate-gui.git
-   cd flux-replicate-gui
+   git clone https://github.com/rtuszik/replicate-flux-lora.git
+   cd replicate-flux-lora
    ```
 
 2. Create and activate a virtual environment:
@@ -29,14 +26,17 @@ A simple web interface for running Flux models using the Replicate API. Use it t
    pip install -r requirements.txt
    ```
 
-4. Set your Replicate API key:
-   ```
-   export REPLICATE_API_TOKEN=your_api_key_here
-   ```
-   On Windows, use `set REPLICATE_API_TOKEN=your_api_key_here`
+4. Set up your environment variables:
+   - Copy the example.env file to create your own .env file:
+     ```
+     cp example.env .env
+     ```
+   - Edit the .env file and replace 'your_api_key_here' with your actual Replicate API key:
+     ```
+     REPLICATE_API_TOKEN=your_api_key_here
+     ```
 
 ## Run it
-
 1. Make sure your virtual environment is activated
 
 2. Start the app:
@@ -47,26 +47,27 @@ A simple web interface for running Flux models using the Replicate API. Use it t
 3. Open `http://localhost:8080` in your browser
 
 4. Choose a model, set your options, enter a prompt, and generate images
-## Docker
 
+## Docker
 **Docker is currently experimental.**
 
-To run the app in a Docker container, follow these steps:
-
-1. Make sure you have Docker installed on your system.
-2. Build the Docker image:
-   ```
-   docker build -t flux-replicate-gui .
-   ```
-   Run the Docker container:
-   ```
-   docker run -p 8080:8080 flux-replicate-gui
-   ```
-   Open `http://localhost:8080` in your browser
-
-
+### Docker Compose
+```yaml
+services:
+  replicate-flux-lora:
+    image: ghcr.io/rtuszik/replicate-flux-lora
+    container_name: replicate-flux-lora
+    environment:
+      - REPLICATE_API_TOKEN=${REPLICATE_API_TOKEN}
+    ports:
+      - "8080:8080"
+    volumes:
+      - ${HOST_OUTPUT_DIR}:/app/output
+    restart: unless-stopped
+```
+Replace `${REPLICATE_API_TOKEN}` with your actual Replicate API key.
+Replace `${HOST_OUTPUT_DIR}` with the directory on your host machine where you want to save generated images.
 
 ## Need help?
-
 Check out Replicate's guide on fine-tuning Flux:
-https://replicate.com/blog/fine-tune-flux
+https://replicate.com/blog/fine-tune-flux

docker-compose.yml

@@ -1,8 +1,8 @@
 
 services:
-  flux-replicate-gui:
-    image: ghcr.io/rtuszik/flux-dev-lora-python:main
-    container_name: flux-replicate-gui
+  replicate-flux-lora:
+    image: ghcr.io/rtuszik/replicate-flux-lora
+    container_name: replicate-flux-lora
     environment:
       - REPLICATE_API_TOKEN=${REPLICATE_API_TOKEN}
     ports: