tuna2134 commited on
Commit
4f9c946
Β·
1 Parent(s): 854f362

fix: fastapi

Browse files
Files changed (1) hide show
  1. main.py +14 -13
main.py CHANGED
@@ -1,6 +1,7 @@
1
  from PIL import Image, ImageDraw, ImageFont, ImageEnhance
2
  from pilmoji import Pilmoji
3
- from flask import Flask, request, send_file
 
4
  import textwrap
5
  import requests
6
  import warnings
@@ -113,20 +114,20 @@ def make(name, tag, id, content, icon):
113
  return file
114
 
115
 
116
- app = Flask(__name__)
117
 
118
 
119
- @app.route("/", methods=["GET"])
120
- def main():
121
- res = make(
122
- request.args.get("name") or "名焑し",
123
- request.args.get("tag") or "0000",
124
- request.args.get("id") or "0000000000000000000",
125
- request.args.get("content") or "γ“γ‚Œγ―γƒ†γ‚Ήγƒˆγ§γ™",
126
- request.args.get(
127
- "icon") or "https://cdn.discordapp.com/embed/avatars/0.png"
128
- )
129
- return send_file(res, mimetype="image/png")
130
 
131
 
132
  if __name__ == "__main__":
 
1
  from PIL import Image, ImageDraw, ImageFont, ImageEnhance
2
  from pilmoji import Pilmoji
3
+ from fastapi import FastAPI
4
+ from fastapi.responses import Response
5
  import textwrap
6
  import requests
7
  import warnings
 
114
  return file
115
 
116
 
117
+ app = FastAPI()
118
 
119
 
120
+ @app.get("/", responses={
121
+ 200: {
122
+ "content": {"image/png": {}}
123
+ }
124
+ })
125
+ def main(
126
+ name: str = "名焑し", tag: str = "0000", id: str = "0000000000000000000",
127
+ content: str = "γ“γ‚Œγ―γƒ†γ‚Ήγƒˆγ§γ™", icon: str = "https://cdn.discordapp.com/embed/avatars/0.png"
128
+ ):
129
+ res = make(name, tag, id, content, icon)
130
+ return Response(content=res, media_type="image/png")
131
 
132
 
133
  if __name__ == "__main__":