Update main.py

main.py

@@ -59,13 +59,25 @@ def inference(img_path, Style, if_face=None):
         print('global exception', error)
         return None, None
 
+
+
+
+
+
+
 @app.post("/inference/")
-async def inference_api(request: InferenceRequest):
-    img_path = request.img_path
-    Style = request.Style
-    if_face = request.if_face
-    output, save_path = inference(img_path, Style, if_face)
-    return {"output": output, "save_path": save_path}
+async def inference_api(file: UploadFile = File(...), Style: str = Form(...), if_face: str = Form(...)):
+    try:
+        contents = await file.read()
+        img_path = f"input.{file.filename}"
+        with open(img_path, "wb") as f:
+            f.write(contents)
+
+        output, save_path = inference(img_path, Style, if_face)
+        
+        return FileResponse(save_path)
+    except Exception as e:
+        return {"error": str(e)}
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")