|
from flask import Flask, request, jsonify |
|
import requests |
|
app = Flask(__name__) |
|
@app.route('/api/v1/image', methods=['POST']) |
|
def generate_image(): |
|
data = request.get_json() |
|
if 'APIKEY' not in data or 'prompt' not in data or 'userId' not in data: |
|
return jsonify({'error': 'Missing required parameters'}), 400 |
|
headers = {'Authorization': data['APIKEY']} |
|
post_data = { |
|
'model': data['ii'], |
|
'prompt': data['prompt'], |
|
'userId': data['userId'] |
|
} |
|
print(data['ii']) |
|
response = requests.post('https://api.ddosxd.ru/v1/image', headers=headers, json=post_data) |
|
print(response.json()) |
|
return response.json() |
|
if __name__ == '__main__': |
|
app.run(host="0.0.0.0", port=7860, use_reloader=False) |
|
|