Update app.py

app.py

@@ -77,25 +77,35 @@ app = Flask(__name__)
 @app.route('/reconstruir', methods=['POST'])
 def reconstruir_imagem():
     try:
+
+        token = request.form.get('token')
         version = request.form.get('version',"v1.4")
         scale = int(request.form.get('scale',2))
         img_file = request.files['imagem']
 
-        temp_filename = 'temp.jpg'
-        img_file.save(temp_filename)
+        if token == "api_key_abcd":
+
+            temp_filename = 'temp.jpg'
+            img_file.save(temp_filename)
 
-        output, save_path = inference(temp_filename, version, scale)
+            output, save_path = inference(temp_filename, version, scale)
 
-        if output is not None:
-            # return send_file(save_path, mimetype='image/jpeg')
-             with open(save_path, 'rb') as image_file:
-                encoded_image = base64.b64encode(image_file.read()).decode('utf-8')
-                return jsonify({'image_base64': encoded_image})
+            if output is not None:
+                # return send_file(save_path, mimetype='image/jpeg')
+                 with open(save_path, 'rb') as image_file:
+                    encoded_image = base64.b64encode(image_file.read()).decode('utf-8')
+                    return jsonify({'status': 'success', 'message':'Imagem restaurada com sucesso', 'image_base64': encoded_image})
+            else:
+                return jsonify({'status': 'error', 'message':'Falha na reconstrução da imagem'})
+
+           
         else:
-            return jsonify({'error': 'Falha na reconstrução da imagem'})
 
+            return jsonify({'status': 'error', 'message': 'Token invalido'})
+
+       
     except Exception as e:
-        return jsonify({'error': str(e)})
+        return jsonify({'status': 'error', 'message': str(e)})
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=80)