MiklX commited on
Commit
bb78d6c
1 Parent(s): 39555fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -12
app.py CHANGED
@@ -1,16 +1,18 @@
1
  from flask import Flask, request, jsonify
2
- import easyocr
3
- from PIL import Image
4
  import requests
5
- from io import BytesIO
6
  app = Flask(__name__)
7
- @app.route('/read_image', methods=['POST'])
8
- def read_image():
9
- url = request.json['url']
10
- response = requests.get(url)
11
- image = Image.open(BytesIO(response.content))
12
- reader = easyocr.Reader(['ru'], gpu=True)
13
- text = reader.readtext(image)
14
- return jsonify({'text': text})
 
 
 
 
 
15
  if __name__ == '__main__':
16
- app.run(host="0.0.0.0", port=7860, use_reloader=False)
 
1
  from flask import Flask, request, jsonify
 
 
2
  import requests
 
3
  app = Flask(__name__)
4
+ @app.route('/api/v1/image', methods=['POST'])
5
+ def generate_image():
6
+ data = request.get_json()
7
+ if 'APIKEY' not in data or 'prompt' not in data or 'userId' not in data:
8
+ return jsonify({'error': 'Missing required parameters'}), 400
9
+ headers = {'Authorization': data['APIKEY']}
10
+ post_data = {
11
+ 'model': 'dalle3',
12
+ 'prompt': data['prompt'],
13
+ 'userId': data['userId']
14
+ }
15
+ response = requests.post('https://api.ddosxd.ru/v1/image', headers=headers, json=post_data)
16
+ return response.json()
17
  if __name__ == '__main__':
18
+ app.run(debug=True)