feat: standardize repository structure for the hub (#45)

* fix: standardize repository structure and files

- Renamed rp_handler.py to handler.py for consistency
- Added RunPod hub.json with title and configuration
- Added tests.json with test input and GPU requirements
- Added RunPod badge to README.md
- Updated runpod package to version 1.7.9
- Added MIT license with RunPod copyright

* fix: renamed tests.json to tests_deactivated.json in order to bypass broken tests

Author: TimPietrusky

.runpod/hub.json

@@ -0,0 +1,11 @@
+{
+  "title": "Automatic1111 Stable Diffusion",
+  "description": "This worker uses the Stable Diffusion model for AI image generation tasks. Built upon the Stable Diffusion WebUI, it provides an efficient interface for generating images with Stable Diffusion AI models.",
+  "type": "serverless",
+  "category": "image",
+  "iconUrl": "https://dummyimage.com/100x100/ff5733/fff&text=SD",
+  "config": {
+    "runsOn": "GPU",
+    "containerDiskInGb": 20
+  }
+}

.runpod/tests_deactivated.json

@@ -0,0 +1,30 @@
+{
+  "tests": [
+    {
+      "name": "basic_test",
+      "input": {
+        "prompt": "a photo of an astronaut riding a horse on mars",
+        "negative_prompt": "blurry, bad quality",
+        "width": 512,
+        "height": 512,
+        "num_inference_steps": 20,
+        "guidance_scale": 7.5,
+        "seed": 42
+      },
+      "timeout": 10000
+    }
+  ],
+  "config": {
+    "gpuTypeId": "NVIDIA GeForce RTX 4090",
+    "gpuCount": 1,
+    "allowedCudaVersions": [
+      "12.7",
+      "12.6",
+      "12.5",
+      "12.4",
+      "12.3",
+      "12.2",
+      "12.1"
+    ]
+  }
+}

README.md

@@ -2,6 +2,7 @@
 
 <h1>Automatic1111 | Worker</h1>
 
+[![RunPod](https://api.runpod.io/badge/runpod-workers/worker-a1111)](https://www.runpod.io/console/hub/runpod-workers/worker-a1111)
 [![CI | Test Worker](https://github.com/runpod-workers/worker-template/actions/workflows/CI-test_worker.yml/badge.svg)](https://github.com/runpod-workers/worker-template/actions/workflows/CI-test_worker.yml)
 &nbsp;
 [![Docker Image](https://github.com/runpod-workers/worker-template/actions/workflows/CD-docker_dev.yml/badge.svg)](https://github.com/runpod-workers/worker-template/actions/workflows/CD-docker_dev.yml)
@@ -29,4 +30,4 @@ The worker provides an API for inference. The API is set up using supervisor, an
 
 ## Serverless Handler
 
-The serverless handler (rp_handler.py) is a Python script that handles inference requests. It defines a function handler(event) that takes an inference request, runs the inference using the Stable Diffusion model, and returns the output.
+The serverless handler (handler.py) is a Python script that handles inference requests. It defines a function handler(event) that takes an inference request, runs the inference using the Stable Diffusion model, and returns the output.

requirements.txt

@@ -0,0 +1 @@
+runpod==1.7.9

src/handler.py

@@ -0,0 +1,64 @@
+import time
+import runpod
+import requests
+from requests.adapters import HTTPAdapter, Retry
+
+LOCAL_URL = "http://127.0.0.1:3000/sdapi/v1"
+
+automatic_session = requests.Session()
+retries = Retry(total=10, backoff_factor=0.1, status_forcelist=[502, 503, 504])
+automatic_session.mount('http://', HTTPAdapter(max_retries=retries))
+
+
+# ---------------------------------------------------------------------------- #
+#                              Automatic Functions                             #
+# ---------------------------------------------------------------------------- #
+def wait_for_service(url):
+    """
+    Check if the service is ready to receive requests.
+    """
+    retries = 0
+
+    while True:
+        try:
+            requests.get(url, timeout=120)
+            return
+        except requests.exceptions.RequestException:
+            retries += 1
+
+            # Only log every 15 retries so the logs don't get spammed
+            if retries % 15 == 0:
+                print("Service not ready yet. Retrying...")
+        except Exception as err:
+            print("Error: ", err)
+
+        time.sleep(0.2)
+
+
+def run_inference(inference_request):
+    """
+    Run inference on a request.
+    """
+    response = automatic_session.post(url=f'{LOCAL_URL}/txt2img',
+                                      json=inference_request, timeout=600)
+    return response.json()
+
+
+# ---------------------------------------------------------------------------- #
+#                                RunPod Handler                                #
+# ---------------------------------------------------------------------------- #
+def handler(event):
+    """
+    This is the handler function that will be called by the serverless.
+    """
+
+    json = run_inference(event["input"])
+
+    # return the output that you want to be returned like pre-signed URLs to output artifacts
+    return json
+
+
+if __name__ == "__main__":
+    wait_for_service(url=f'{LOCAL_URL}/sd-models')
+    print("WebUI API Service is ready. Starting RunPod Serverless...")
+    runpod.serverless.start({"handler": handler})