Update app.py

app.py

@@ -10,9 +10,10 @@ from io import BytesIO
 
 class Prodia:
     def __init__(self, api_key, base=None):
-        self.base = base or "https://api.prodia.com/v1"
+        self.base = base or "https://api.prodia.com/v2"
         self.headers = {
             "X-Prodia-Key": api_key
+            "Authorization": f"Bearer {os.getenv("API_KEY")}"
         }
 
     def generate(self, params):
@@ -40,6 +41,12 @@ class Prodia:
         response = self._get(f"{self.base}/sdxl/samplers")
         return response.json()
 
+
+    def generate_v2(self, config):
+        response = self._post("https://inference.prodia.com/v2/job", {"type": "v2.job.sdxl.txt2img", "config": config})
+        return Image.open(BytesIO(r.content)).convert("RGBA")
+        
+
     def _post(self, url, params):
         headers = {
             **self.headers,
@@ -60,6 +67,8 @@ class Prodia:
 
         return response
 
+    
+
 
 def image_to_base64(image_path):
     # Open the image with PIL
@@ -78,16 +87,24 @@ def image_to_base64(image_path):
 prodia_client = Prodia(api_key=os.getenv("PRODIA_API_KEY"))
 
 def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
-    result = prodia_client.generate({
+
+    config_without_model_and_sampler = {
         "prompt": prompt,
         "negative_prompt": negative_prompt,
-        "model": model,
         "steps": steps,
-        "sampler": sampler,
         "cfg_scale": cfg_scale,
         "width": width,
         "height": height,
         "seed": seed
+    }
+
+    if model == "sd_xl_base_1.0.safetensors [be9edd61]":
+        return prodia_client.generate_v2(config_without_model_and_sampler)
+    
+    result = prodia_client.generate({
+        **config_without_model_and_sampler
+        "model": model,
+        "sampler": sampler
     })
 
     job = prodia_client.wait(result)