Added list_models, list_samplers and wait, fixed Org strings

app.py

@@ -16,18 +16,34 @@ class RenderNet:
     def __init__(self, api_key, base=None):
         self.base = base or "https://api.prodia.com/v1"
         self.headers = {
-            "X-Prodia-Key": api_key
+            "X-RenderNet-Key": api_key
         }
     
     def generate(self, params):
         response = self._post(f"{self.base}/sd/generate", params)
         return response.json()
     
+    def get_job(self, job_id):
+        response = self._get(f"{self.base}/job/{job_id}")
+        return response.json()
+
+    def wait(self, job):
+        job_result = job
+
+        while job_result['status'] not in ['succeeded', 'failed']:
+            time.sleep(0.25)
+            job_result = self.get_job(job['job'])
+
+        return job_result
 
     def list_models(self):
         response = self._get(f"{self.base}/sd/models")
         return response.json()
 
+    def list_samplers(self):
+        response = self._get(f"{self.base}/sd/samplers")
+        return response.json()
+
     def _post(self, url, params):
         headers = {
             **self.headers,
@@ -36,7 +52,7 @@ class RenderNet:
         response = requests.post(url, headers=headers, data=json.dumps(params))
 
         if response.status_code != 200:
-            raise Exception(f"Bad Prodia Response: {response.status_code}")
+            raise Exception(f"Bad RenderNet Response: {response.status_code}")
 
         return response
 
@@ -44,7 +60,7 @@ class RenderNet:
         response = requests.get(url, headers=self.headers)
 
         if response.status_code != 200:
-            raise Exception(f"Bad Prodia Response: {response.status_code}")
+            raise Exception(f"Bad RenderNet Response: {response.status_code}")
 
         return response
 
@@ -131,8 +147,8 @@ def send_to_txt2img(image):
         return result
 
 
-prodia_client = Prodia(api_key=os.getenv("PRODIA_API_KEY"))
-model_list = prodia_client.list_models()
+rendernet_client = RenderNet(api_key=os.getenv("RENDERNET_API_KEY"))
+model_list = rendernet_client.list_models()
 model_names = {}
 
 for model_name in model_list:
@@ -140,7 +156,7 @@ for model_name in model_list:
     model_names[name_without_ext] = model_name
 
 def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
-    result = prodia_client.generate({
+    result = rendernet_client.generate({
         "prompt": prompt,
         "negative_prompt": negative_prompt,
         "model": model,
@@ -152,7 +168,7 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
         "seed": seed
     })
 
-    job = prodia_client.wait(result)
+    job = rendernet_client.wait(result)
 
     return job["imageUrl"]
 
@@ -166,10 +182,10 @@ css = """
 with gr.Blocks(css=css) as demo:
     with gr.Row():
         with gr.Column(scale=6):
-            model = gr.Dropdown(interactive=True,value="absolutereality_v181.safetensors [3d9d4d2b]", show_label=True, label="Stable Diffusion Checkpoint", choices=prodia_client.list_models())
+            model = gr.Dropdown(interactive=True,value="absolutereality_v181.safetensors [3d9d4d2b]", show_label=True, label="Stable Diffusion Checkpoint", choices=rendernet_client.list_models())
   
         with gr.Column(scale=1):
-            gr.Markdown(elem_id="powered-by-prodia", value="AUTOMATIC1111 Stable Diffusion Web UI.<br>Powered by [Prodia](https://prodia.com).<br>For more features and faster generation times check out our [API Docs](https://docs.prodia.com/reference/getting-started-guide).")
+            gr.Markdown(elem_id="powered-by-rendernet", value="AUTOMATIC1111 Stable Diffusion Web UI.<br>Powered by [RenderNet](https://rendernet.ai).<br>For more features and faster generation times check out our [API Docs](https://rendernet.ai/).")
 
 
     with gr.Tabs() as tabs: