Update app.py
Browse files
app.py
CHANGED
@@ -182,27 +182,29 @@ def handle_request():
|
|
182 |
else:
|
183 |
enhanced_prompt = translate_and_enhance_prompt(clean_prompt, random_token)
|
184 |
|
185 |
-
new_url =
|
186 |
new_request_body = {
|
|
|
187 |
"prompt": enhanced_prompt,
|
|
|
188 |
"image_size": image_size,
|
189 |
"batch_size": 1,
|
|
|
190 |
"num_inference_steps": 4,
|
191 |
"guidance_scale": 1
|
192 |
}
|
193 |
|
194 |
headers = {
|
195 |
-
'
|
196 |
-
'
|
197 |
-
'Authorization': f'Bearer {random_token}'
|
198 |
}
|
199 |
-
|
200 |
response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
|
201 |
response.raise_for_status()
|
202 |
response_body = response.json()
|
203 |
-
|
204 |
-
if '
|
205 |
-
image_url = response_body['
|
206 |
else:
|
207 |
raise ValueError("Unexpected response structure from image generation API")
|
208 |
|
|
|
182 |
else:
|
183 |
enhanced_prompt = translate_and_enhance_prompt(clean_prompt, random_token)
|
184 |
|
185 |
+
new_url = 'https://api.siliconflow.cn/v1/images/generations'
|
186 |
new_request_body = {
|
187 |
+
"model": mapped_model, # 使用映射后的模型名称
|
188 |
"prompt": enhanced_prompt,
|
189 |
+
"negative_prompt": "", # 如果不需要负提示词,可以留空
|
190 |
"image_size": image_size,
|
191 |
"batch_size": 1,
|
192 |
+
"seed": random.randint(0, 4999999999), # 随机生成一个种子值
|
193 |
"num_inference_steps": 4,
|
194 |
"guidance_scale": 1
|
195 |
}
|
196 |
|
197 |
headers = {
|
198 |
+
'Authorization': f'Bearer {random_token}',
|
199 |
+
'Content-Type': 'application/json'
|
|
|
200 |
}
|
201 |
+
|
202 |
response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
|
203 |
response.raise_for_status()
|
204 |
response_body = response.json()
|
205 |
+
|
206 |
+
if 'data' in response_body and response_body['data'] and 'url' in response_body['data'][0]:
|
207 |
+
image_url = response_body['data'][0]['url']
|
208 |
else:
|
209 |
raise ValueError("Unexpected response structure from image generation API")
|
210 |
|