Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -230,45 +230,6 @@ def get_root_url(request: Request, route_path: str, root_path: str | None):
|
|
| 230 |
return root_path
|
| 231 |
gr.route_utils.get_root_url = get_root_url
|
| 232 |
|
| 233 |
-
@app.post("/magic_quill/generate_image")
|
| 234 |
-
async def generate_image(request: Request):
|
| 235 |
-
data = await request.json()
|
| 236 |
-
|
| 237 |
-
# 1. Extract headers from the FastAPI request object
|
| 238 |
-
forward_headers = get_zerogpu_headers(request.headers)
|
| 239 |
-
|
| 240 |
-
try:
|
| 241 |
-
# 2. Instantiate client with headers
|
| 242 |
-
client = Client(BACKEND_URL, headers=forward_headers)
|
| 243 |
-
|
| 244 |
-
res = await run_in_threadpool(
|
| 245 |
-
client.predict,
|
| 246 |
-
data["merged_image"],
|
| 247 |
-
data["total_mask"],
|
| 248 |
-
data["original_image"],
|
| 249 |
-
data["add_color_image"],
|
| 250 |
-
data["add_edge_mask"],
|
| 251 |
-
data["remove_edge_mask"],
|
| 252 |
-
data["fill_mask"],
|
| 253 |
-
data["add_prop_image"],
|
| 254 |
-
data["positive_prompt"],
|
| 255 |
-
data["negative_prompt"],
|
| 256 |
-
data["fine_edge"],
|
| 257 |
-
data["fix_perspective"],
|
| 258 |
-
data["grow_size"],
|
| 259 |
-
data["edge_strength"],
|
| 260 |
-
data["color_strength"],
|
| 261 |
-
data["local_strength"],
|
| 262 |
-
data["seed"],
|
| 263 |
-
data["steps"],
|
| 264 |
-
data["cfg"],
|
| 265 |
-
api_name="/generate"
|
| 266 |
-
)
|
| 267 |
-
return {'res': res}
|
| 268 |
-
except Exception as e:
|
| 269 |
-
print(f"Error in backend generation: {e}")
|
| 270 |
-
return {'error': str(e)}
|
| 271 |
-
|
| 272 |
@app.post("/magic_quill/process_background_img")
|
| 273 |
async def process_background_img(request: Request):
|
| 274 |
img = await request.json()
|
|
|
|
| 230 |
return root_path
|
| 231 |
gr.route_utils.get_root_url = get_root_url
|
| 232 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 233 |
@app.post("/magic_quill/process_background_img")
|
| 234 |
async def process_background_img(request: Request):
|
| 235 |
img = await request.json()
|