revi13 commited on
Commit
22dcdf9
·
verified ·
1 Parent(s): 84a2196

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -6
app.py CHANGED
@@ -6,6 +6,10 @@ from PIL import Image
6
 
7
  app = FastAPI()
8
 
 
 
 
 
9
  @app.post("/generate")
10
  async def generate_image(request: Request):
11
  try:
@@ -13,11 +17,11 @@ async def generate_image(request: Request):
13
  base64_image = body.get("image")
14
  prompt = body.get("prompt", "")
15
 
16
- # Base64文字列 → PIL画像
17
  image_data = base64.b64decode(base64_image.split(",")[-1])
18
  image = Image.open(io.BytesIO(image_data)).convert("RGB")
19
 
20
- # 再エンコードしてBase64で返す
21
  buffered = io.BytesIO()
22
  image.save(buffered, format="PNG")
23
  encoded_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
@@ -27,8 +31,10 @@ async def generate_image(request: Request):
27
  "echo_prompt": prompt,
28
  "image_base64": f"data:image/png;base64,{encoded_img}"
29
  })
 
30
  except Exception as e:
31
- return JSONResponse(status_code=500, content={
32
- "success": False,
33
- "error": str(e)
34
- })
 
 
6
 
7
  app = FastAPI()
8
 
9
+ @app.get("/")
10
+ def root():
11
+ return {"status": "ok"}
12
+
13
  @app.post("/generate")
14
  async def generate_image(request: Request):
15
  try:
 
17
  base64_image = body.get("image")
18
  prompt = body.get("prompt", "")
19
 
20
+ # base64 → PIL Image
21
  image_data = base64.b64decode(base64_image.split(",")[-1])
22
  image = Image.open(io.BytesIO(image_data)).convert("RGB")
23
 
24
+ # 再エンコード
25
  buffered = io.BytesIO()
26
  image.save(buffered, format="PNG")
27
  encoded_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
31
  "echo_prompt": prompt,
32
  "image_base64": f"data:image/png;base64,{encoded_img}"
33
  })
34
+
35
  except Exception as e:
36
+ # ログ出力を追加
37
+ print("=== ERROR OCCURRED ===")
38
+ traceback.print_exc()
39
+ print("======================")
40
+ return JSONResponse(status_code=500, content={"success": False, "error": str(e)})