diff --git a/.inscode b/.inscode index 7d17616b58c3588c83483b4d2f306c1a69c1904f..3a001eb88671424594301483eb8665b81f6f4826 100644 --- a/.inscode +++ b/.inscode @@ -1,4 +1,5 @@ run = "pip install -r requirements.txt;python main.py" +language = "python" [packager] AUTO_PIP = true @@ -9,4 +10,7 @@ PATH = "${VIRTUAL_ENV}/bin:${PATH}" PYTHONPATH = "$PYTHONHOME/lib/python3.10:${VIRTUAL_ENV}/lib/python3.10/site-packages" REPLIT_POETRY_PYPI_REPOSITORY = "http://mirrors.csdn.net.cn/repository/csdn-pypi-mirrors/simple" MPLBACKEND = "TkAgg" -POETRY_CACHE_DIR = "/root/${PROJECT_DIR}/.cache/pypoetry" \ No newline at end of file +POETRY_CACHE_DIR = "/root/${PROJECT_DIR}/.cache/pypoetry" + +[debugger] +program = "main.py" diff --git a/main.py b/main.py index 4c0c135f61696bcf42c375ca5ab62aa5b105afc8..e6958fdf33d010702873cb12b0cc1aae28804f05 100644 --- a/main.py +++ b/main.py @@ -1 +1,56 @@ -print('欢迎来到 InsCode') \ No newline at end of file +import requests +from flask import Flask, request, jsonify +from flask_cors import CORS +import random + +app = Flask(__name__) +CORS(app, resources={r"/*": {"origins":"*"}}, send_wildcard=True) + +API_URL = "https://w0s4i2j810q2f49b.aistudio-hub.baidu.com/image/generations" +headers = { + # 请前往 https://aistudio.baidu.com/index/accessToken 查看 访问令牌 + "Authorization": "token 80e314d558efc67b7787fa24483a323cdde9b733", + "Content-Type": "application/json" +} + + +def query(payload): + print(payload) + response = requests.post(API_URL, headers=headers, json=payload) + return response.json() + + +@app.route('/') +def hello(): + return 'hello!' + +@app.route('/test') +def test(): + return query({ + 'seed': random.randint(100000, 999999999), + 'n': 1, + 'prompt': '生成一只卡通皮卡丘', + 'size': '768x768', + 'steps':25 + }) + + +@app.route('/image/generations', methods=['POST']) +def generation(): + data = request.json + n = data.get('n') + prompt = data.get('prompt') + size = data.get('size') + steps = data.get('steps') + + return query({ + 'seed': random.randint(100000, 999999999), + 'n': n, + 'negative_prompt': '皮卡丘', + 'prompt': prompt, + 'size': size, + 'steps':steps + }) + +if __name__ == '__main__': + app.run(host='0.0.0.0') \ No newline at end of file