Spaces:
Runtime error
Runtime error
flask_request
Browse files
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
from flask import Flask, request, render_template, send_file, jsonify, send_from_directory
|
2 |
from flask_socketio import SocketIO, join_room, leave_room, close_room, rooms, disconnect
|
3 |
from flask_cors import CORS
|
4 |
from flask_limiter import Limiter
|
@@ -155,10 +155,10 @@ def submit_task():
|
|
155 |
return jsonify({'error': 'Maximum number of concurrent tasks reached'}), 429
|
156 |
|
157 |
task_id = str(uuid.uuid4())
|
158 |
-
file =
|
159 |
-
mode =
|
160 |
-
weight1 = float(
|
161 |
-
weight2 = float(
|
162 |
|
163 |
# ファイルタイプの制限
|
164 |
allowed_extensions = {'png', 'jpg', 'jpeg', 'gif'}
|
@@ -235,9 +235,9 @@ def root():
|
|
235 |
# process_refined のエンドポイント
|
236 |
@app.route('/process_refined', methods=['POST'])
|
237 |
def process_refined():
|
238 |
-
file =
|
239 |
-
weight1 = float(
|
240 |
-
weight2 = float(
|
241 |
|
242 |
image = ensure_rgb(Image.open(file.stream))
|
243 |
sotai_image, sketch_image = process_image_as_base64(image, "refine", weight1, weight2)
|
@@ -249,7 +249,7 @@ def process_refined():
|
|
249 |
|
250 |
@app.route('/process_original', methods=['POST'])
|
251 |
def process_original():
|
252 |
-
file =
|
253 |
|
254 |
image = ensure_rgb(Image.open(file.stream))
|
255 |
sotai_image, sketch_image = process_image_as_base64(image, "original")
|
@@ -261,7 +261,7 @@ def process_original():
|
|
261 |
|
262 |
@app.route('/process_sketch', methods=['POST'])
|
263 |
def process_sketch():
|
264 |
-
file =
|
265 |
|
266 |
image = ensure_rgb(Image.open(file.stream))
|
267 |
sotai_image, sketch_image = process_image_as_base64(image, "sketch")
|
|
|
1 |
+
from flask import Flask, request as flask_request, render_template, send_file, jsonify, send_from_directory
|
2 |
from flask_socketio import SocketIO, join_room, leave_room, close_room, rooms, disconnect
|
3 |
from flask_cors import CORS
|
4 |
from flask_limiter import Limiter
|
|
|
155 |
return jsonify({'error': 'Maximum number of concurrent tasks reached'}), 429
|
156 |
|
157 |
task_id = str(uuid.uuid4())
|
158 |
+
file = flask_request.files['file']
|
159 |
+
mode = flask_request.form.get('mode', 'refine')
|
160 |
+
weight1 = float(flask_request.form.get('weight1', 0.4))
|
161 |
+
weight2 = float(flask_request.form.get('weight2', 0.3))
|
162 |
|
163 |
# ファイルタイプの制限
|
164 |
allowed_extensions = {'png', 'jpg', 'jpeg', 'gif'}
|
|
|
235 |
# process_refined のエンドポイント
|
236 |
@app.route('/process_refined', methods=['POST'])
|
237 |
def process_refined():
|
238 |
+
file = flask_request.files['file']
|
239 |
+
weight1 = float(flask_request.form.get('weight1', 0.4))
|
240 |
+
weight2 = float(flask_request.form.get('weight2', 0.3))
|
241 |
|
242 |
image = ensure_rgb(Image.open(file.stream))
|
243 |
sotai_image, sketch_image = process_image_as_base64(image, "refine", weight1, weight2)
|
|
|
249 |
|
250 |
@app.route('/process_original', methods=['POST'])
|
251 |
def process_original():
|
252 |
+
file = flask_request.files['file']
|
253 |
|
254 |
image = ensure_rgb(Image.open(file.stream))
|
255 |
sotai_image, sketch_image = process_image_as_base64(image, "original")
|
|
|
261 |
|
262 |
@app.route('/process_sketch', methods=['POST'])
|
263 |
def process_sketch():
|
264 |
+
file = flask_request.files['file']
|
265 |
|
266 |
image = ensure_rgb(Image.open(file.stream))
|
267 |
sotai_image, sketch_image = process_image_as_base64(image, "sketch")
|