Merge branch 'master'

.dockerignore

@@ -0,0 +1,34 @@
+# Include any files or directories that you don't want to be copied to your
+# container here (e.g., local build artifacts, temporary files, etc.).
+#
+# For more help, visit the .dockerignore file reference guide at
+# https://docs.docker.com/go/build-context-dockerignore/
+
+**/.DS_Store
+**/__pycache__
+**/.venv
+**/.classpath
+**/.dockerignore
+**/.env
+**/.git
+**/.gitignore
+**/.project
+**/.settings
+**/.toolstarget
+**/.vs
+**/.vscode
+**/*.*proj.user
+**/*.dbmdl
+**/*.jfm
+**/bin
+**/charts
+**/docker-compose*
+**/compose.y*ml
+**/Dockerfile*
+**/node_modules
+**/npm-debug.log
+**/obj
+**/secrets.dev.yaml
+**/values.dev.yaml
+LICENSE
+README.md

.gitignore

@@ -0,0 +1,10 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv

.python-version

@@ -0,0 +1 @@
+3.11

Dockerfile

@@ -0,0 +1,45 @@
+# syntax=docker/dockerfile:1
+
+ARG PYTHON_VERSION=3.11
+FROM python:${PYTHON_VERSION}-slim as base
+
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+
+# Install uv
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
+
+WORKDIR /app
+
+# Create a proper non-root user WITH home
+ARG UID=10001
+RUN adduser \
+    --disabled-password \
+    --gecos "" \
+    --uid "${UID}" \
+    appuser
+
+# Set correct HOME and cache
+ENV HOME=/home/appuser
+ENV UV_CACHE_DIR=/home/appuser/.cache/uv
+
+# Copy dependency files
+COPY pyproject.toml uv.lock ./
+
+# Install dependencies
+RUN --mount=type=cache,target=/home/appuser/.cache/uv \
+    uv sync --frozen --no-install-project
+
+# Copy source
+COPY . .
+
+# Install project
+RUN --mount=type=cache,target=/home/appuser/.cache/uv \
+    uv sync --frozen
+
+# Switch user
+USER appuser
+
+EXPOSE 7860
+
+CMD ["uv", "run", "python", "main.py"]

README.md

@@ -1,10 +1,9 @@
 ---
-title: Image Upscaler
-emoji: 🏢
-colorFrom: blue
-colorTo: yellow
+title: AI Image Upscaler
+emoji: 🔍
+colorFrom: purple
+colorTo: blue
 sdk: docker
+app_port: 7860
 pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

main.py

@@ -0,0 +1,49 @@
+import gradio as gr
+import numpy as np
+from PIL import Image
+from realesrgan import RealESRGANer
+from basicsr.archs.rrdbnet_arch import RRDBNet
+
+# Load model once at startup (not on every call)
+def load_model(scale=4):
+    model = RRDBNet(
+        num_in_ch=3, num_out_ch=3,
+        num_feat=64, num_block=23, num_grow_ch=32, scale=scale
+    )
+    upsampler = RealESRGANer(
+        scale=scale,
+        model_path=f"https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x{scale}plus.pth",
+        model=model,
+        half=False  # set True if GPU available
+    )
+    return upsampler
+
+upsampler_cache = {}
+
+def upscale(image: Image.Image, scale: int):
+    if scale not in upsampler_cache:
+        upsampler_cache[scale] = load_model(scale)
+    
+    upsampler = upsampler_cache[scale]
+    img_array = np.array(image)
+    output, _ = upsampler.enhance(img_array, outscale=scale)
+    return Image.fromarray(output)
+
+# Gradio UI
+with gr.Blocks(title="AI Image Upscaler") as demo:
+    gr.Markdown("## 🔍 AI Image Upscaler\nPowered by Real-ESRGAN")
+    
+    with gr.Row():
+        with gr.Column():
+            input_img = gr.Image(type="pil", label="Input Image")
+            scale_choice = gr.Radio(
+                choices=[2, 4, 8], value=4, label="Upscale Factor"
+            )
+            btn = gr.Button("Upscale", variant="primary")
+        
+        with gr.Column():
+            output_img = gr.Image(type="pil", label="Upscaled Output")
+    
+    btn.click(fn=upscale, inputs=[input_img, scale_choice], outputs=output_img)
+
+demo.launch()

pyproject.toml

@@ -0,0 +1,15 @@
+[project]
+name = "image-upscaler"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "basicsr>=1.4.2",
+    "gradio>=6.11.0",
+    "numpy<2",
+    "pillow>=12.2.0",
+    "realesrgan>=0.3.0",
+    "torch==2.1.2",
+    "torchvision==0.16.2",
+]