Sun Dec 3 12:35:00 CST 2023 inscode

.inscode

 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"

main.py

-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