Spaces:
Runtime error
Runtime error
syurein commited on
Commit ·
d63a0d3
1
Parent(s): b3d6d35
二つ分にした
Browse files
app.py
CHANGED
|
@@ -992,11 +992,8 @@ async def classify_image_llm(file: UploadFile = File(...)):
|
|
| 992 |
return {"danger":danger_level}
|
| 993 |
|
| 994 |
|
| 995 |
-
|
| 996 |
-
|
| 997 |
-
|
| 998 |
@app.post("/analyze")
|
| 999 |
-
async def
|
| 1000 |
scene: str = Form(...),
|
| 1001 |
mode: str = Form(...),
|
| 1002 |
value: str = Form(...),
|
|
@@ -1036,6 +1033,32 @@ async def create_mask_sum2(image: UploadFile = File(...),
|
|
| 1036 |
return FileResponse(output_path)
|
| 1037 |
|
| 1038 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1039 |
#三浦と相談
|
| 1040 |
|
| 1041 |
@app.post("/create-mask-and-inpaint-sum-llm-auto")
|
|
|
|
| 992 |
return {"danger":danger_level}
|
| 993 |
|
| 994 |
|
|
|
|
|
|
|
|
|
|
| 995 |
@app.post("/analyze")
|
| 996 |
+
async def create_mask_sum3(image: UploadFile = File(...),
|
| 997 |
scene: str = Form(...),
|
| 998 |
mode: str = Form(...),
|
| 999 |
value: str = Form(...),
|
|
|
|
| 1033 |
return FileResponse(output_path)
|
| 1034 |
|
| 1035 |
|
| 1036 |
+
@app.post("/create-mask-and-inpaint-sum-llm-simple")
|
| 1037 |
+
async def create_mask_sum2(image: UploadFile = File(...), risk_level: int = Form(...),
|
| 1038 |
+
x1: float = Form(...),
|
| 1039 |
+
y1: float = Form(...),
|
| 1040 |
+
x2: float = Form(...),
|
| 1041 |
+
y2: float = Form(...),):
|
| 1042 |
+
default_x = 0.001
|
| 1043 |
+
default_y = 0.001
|
| 1044 |
+
|
| 1045 |
+
|
| 1046 |
+
point1 = [default_x if math.isnan(x1) else x1, default_y if math.isnan(y1) else y1]
|
| 1047 |
+
|
| 1048 |
+
point2 = [default_x if math.isnan(x2) else x2, default_y if math.isnan(y2) else y2]
|
| 1049 |
+
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
| 1050 |
+
# 一意な識別子を生成
|
| 1051 |
+
unique_id = uuid.uuid4().hex
|
| 1052 |
+
input_path = save_image(image.file, f"./input_{timestamp}_{unique_id}.jpg")
|
| 1053 |
+
mask_path = llm_to_process_image_simple(risk_level, input_path, point1, point2,thresholds=thresholds)
|
| 1054 |
+
output_path = f"./output_simple_lama_{timestamp}_{unique_id}.jpg"
|
| 1055 |
+
print('point1,point2',point1,point2)#消去したくない範囲のこと
|
| 1056 |
+
# OpenCVでインペイント
|
| 1057 |
+
inpaint_image_with_mask1(input_path, mask_path, output_path)
|
| 1058 |
+
|
| 1059 |
+
return FileResponse(output_path)
|
| 1060 |
+
|
| 1061 |
+
|
| 1062 |
#三浦と相談
|
| 1063 |
|
| 1064 |
@app.post("/create-mask-and-inpaint-sum-llm-auto")
|