File size: 774 Bytes
00d1ad6 be5364e bb78d6c 8c3d303 bb78d6c 8c3d303 bb78d6c 8c3d303 bb78d6c 8c3d303 9d04ddf bb78d6c be5364e 361d32f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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['model'],
'prompt': data['prompt'],
'userId': data['userId']
}
print(data['model'])
response = requests.post('https://api.example.com/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)
|