Update main.py

main.py

@@ -5,6 +5,8 @@ from urllib.parse import quote
 import asyncio
 from io import BytesIO
 
+url = os.environ.get("url")
+
 app = FastAPI()
 
 def create_job(prompt, model, sampler, seed, neg):
@@ -16,7 +18,7 @@ def create_job(prompt, model, sampler, seed, neg):
         seed = '-1'
     if neg is None:
         neg = "(long list of negative prompts removed for brevity)"
-    url = 'https://api.prodia.com/generate'
+    url = f'https://api.{url}.com/generate'
     params = {
         'new': 'true',
         'prompt': quote(prompt),
@@ -36,7 +38,7 @@ def create_job(prompt, model, sampler, seed, neg):
 @app.get("/generate_image")
 async def generate_image(prompt: str, model: str = None, sampler: str = None, seed: str = None, neg: str = None):
     job_id = create_job(prompt, model, sampler, seed, neg)
-    url = f'https://api.prodia.com/job/{job_id}'
+    url = f'https://api.{url}.com/job/{job_id}'
     headers = {'accept': '*/*'}
     
     while True:
@@ -44,7 +46,7 @@ async def generate_image(prompt: str, model: str = None, sampler: str = None, se
         response.raise_for_status()
         job_response = response.json()
         if job_response['status'] == 'succeeded':
-            image_url = f'https://images.prodia.xyz/{job_id}.png'
+            image_url = f'https://images.{url}.xyz/{job_id}.png'
             image_response = requests.get(image_url)
             image_response.raise_for_status()
             image = BytesIO(image_response.content)