Update app.py

app.py

@@ -344,38 +344,30 @@ fetch('${apiUrl}', {
 
 @app.route('/api/restore', methods=['POST'])
 def api_restore():
-    if 'file' not in request.files:
+    data = request.get_json()
+    if not data or 'file' not in data:
         return jsonify({'error': 'No file uploaded'}), 400
-    
-    file = request.files['file']
-    version = request.form.get('version', 'v1.4')
-    scale = float(request.form.get('scale', 2))
-    weight = float(request.form.get('weight', 0.5)) if version == 'CodeFormer' else None
-    
-    if file.filename == '':
-        return jsonify({'error': 'No selected file'}), 400
-    
-    try:
-        # Save uploaded file to temp location
-        temp_dir = tempfile.mkdtemp()
-        input_path = os.path.join(temp_dir, file.filename)
-        file.save(input_path)
-        
-        # Process image
-        output_path = process_image(input_path, version, scale, weight)
-        
-        # Return the processed image
-        return send_file(output_path, mimetype='image/jpeg')
-    
-    except Exception as e:
-        return jsonify({'error': str(e)}), 500
-    
-    finally:
-        # Clean up temp files
-        if 'input_path' in locals() and os.path.exists(input_path):
-            os.remove(input_path)
-        if 'temp_dir' in locals() and os.path.exists(temp_dir):
-            os.rmdir(temp_dir)
+
+    import base64
+    from io import BytesIO
+    from PIL import Image
+
+    # Base64データをデコード
+    file_data = data['file'].split(",")[1]
+    img_bytes = base64.b64decode(file_data)
+    temp_dir = tempfile.mkdtemp()
+    input_path = os.path.join(temp_dir, "input.jpg")
+    with open(input_path, "wb") as f:
+        f.write(img_bytes)
+
+    version = data.get('version', 'v1.4')
+    scale = float(data.get('scale', 2))
+    weight = float(data.get('weight', 0.5)) if version == 'CodeFormer' else None
+
+    # 画像処理
+    output_path = process_image(input_path, version, scale, weight)
+    return send_file(output_path, mimetype="image/jpeg")
+
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)