api_server.py 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. from fastapi import FastAPI, Request
  2. import uvicorn
  3. from MsgSend import send_message
  4. import json
  5. app = FastAPI()
  6. @app.post("/record")
  7. async def record_endpoint(request: Request):
  8. """
  9. 接收POST请求并返回ok
  10. """
  11. try:
  12. data = await request.json()
  13. content = json.loads(data.get("content", ""))
  14. # 如果需要发送消息,取消下面一行的注释
  15. send_message(content.get("text"))
  16. print(f"Received data: {data.get('content', '')}")
  17. return {"message": "ok"}
  18. except Exception as e:
  19. print(f"Error processing request: {e}")
  20. return {"message": "error", "detail": str(e)}
  21. @app.post("/text")
  22. async def message_endpoint(request: Request):
  23. """
  24. 接收POST请求并返回ok
  25. """
  26. try:
  27. data = await request.json()
  28. content = data.get("content", "")
  29. # 如果需要发送消息,取消下面一行的注释
  30. send_message(content)
  31. print(f"Received data: {data.get('content', '')}")
  32. return {"message": "ok"}
  33. except Exception as e:
  34. print(f"Error processing request: {e}")
  35. if __name__ == "__main__":
  36. uvicorn.run(app, host="0.0.0.0", port=9017)