Upload app.py

app.py

@@ -36,15 +36,15 @@ app.logger.setLevel(logging.INFO)
 MODEL_MAPPING = {
     "flux.1-schnell": {
         "provider": "black-forest-labs",
-        "mapping": "black-forest-labs/FLUX.1-schnell"
+        "mapping": "Pro/black-forest-labs/FLUX.1-schnell"
     },
-    "sd-turbo": {
-        "provider": "stabilityai",
-        "mapping": "stabilityai/sd-turbo"
+    "flux.1-dev": {
+        "provider": "black-forest-labs",
+        "mapping": "black-forest-labs/FLUX.1-dev"
     },
-    "sdxl-turbo": {
+    "stable-diffusion-3-5-large": {
         "provider": "stabilityai",
-        "mapping": "stabilityai/sdxl-turbo"
+        "mapping": "stabilityai/stable-diffusion-3-5-large"
     },
     "stable-diffusion-2-1": {
         "provider": "stabilityai",
@@ -182,27 +182,29 @@ def handle_request():
         else:
             enhanced_prompt = translate_and_enhance_prompt(clean_prompt, random_token)
         
-        new_url = f'https://api.siliconflow.cn/v1/{mapped_model}/text-to-image'
+        new_url = 'https://api.siliconflow.cn/v1/images/generations'
         new_request_body = {
+            "model": mapped_model,  # 使用映射后的模型名称
             "prompt": enhanced_prompt,
+            "negative_prompt": "",  # 如果不需要负提示词，可以留空
             "image_size": image_size,
             "batch_size": 1,
-            "num_inference_steps": 4,
-            "guidance_scale": 1
+            "seed": random.randint(0, 4999999999),  # 随机生成一个种子值
+            "num_inference_steps": 20,
+            "guidance_scale": 7.5
         }
         
         headers = {
-            'accept': 'application/json',
-            'content-type': 'application/json',
-            'Authorization': f'Bearer {random_token}'
+            'Authorization': f'Bearer {random_token}',
+            'Content-Type': 'application/json'
         }
-        
+
         response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
         response.raise_for_status()
         response_body = response.json()
-        
-        if 'images' in response_body and response_body['images'] and 'url' in response_body['images'][0]:
-            image_url = response_body['images'][0]['url']
+
+        if 'data' in response_body and response_body['data'] and 'url' in response_body['data'][0]:
+            image_url = response_body['data'][0]['url']
         else:
             raise ValueError("Unexpected response structure from image generation API")