-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
99 lines (80 loc) · 3.47 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
from flask import Flask, render_template, request, send_file, redirect, url_for
import base64
from PIL import Image
import numpy as np
import io
import zipfile
import os
app = Flask(__name__)
def files_to_image(files, existing_image=None):
encoded_files = []
# If an existing image is provided, extract its data first
if existing_image:
existing_files = image_to_files(existing_image.read())
for file_name, file_data in existing_files:
encoded_data = base64.b64encode(file_data).decode('utf-8')
encoded_files.append(
f"{file_name}||{len(encoded_data)}||{encoded_data}")
# Add new files
for file in files:
file_data = file.read()
encoded_data = base64.b64encode(file_data).decode('utf-8')
file_name = file.filename
encoded_files.append(
f"{file_name}||{len(encoded_data)}||{encoded_data}")
combined_encoded_data = '<<>>'.join(encoded_files)
data_len = len(combined_encoded_data)
image_size = int(np.ceil(np.sqrt(data_len / 3)))
image_array = np.zeros((image_size, image_size, 3), dtype=np.uint8)
encoded_bytes = combined_encoded_data.encode('utf-8')
flat_image_array = image_array.flatten()
for i in range(min(len(flat_image_array), len(encoded_bytes))):
flat_image_array[i] = encoded_bytes[i]
image_array = flat_image_array.reshape((image_size, image_size, 3))
image = Image.fromarray(image_array)
img_byte_arr = io.BytesIO()
image.save(img_byte_arr, format='PNG')
img_byte_arr = img_byte_arr.getvalue()
return img_byte_arr
def image_to_files(image_data):
image = Image.open(io.BytesIO(image_data))
image_array = np.array(image)
flat_image_array = image_array.flatten()
encoded_bytes = bytes([byte for byte in flat_image_array if byte != 0])
combined_encoded_data = encoded_bytes.decode('utf-8')
encoded_files = combined_encoded_data.split('<<>>')
extracted_files = []
for encoded_file in encoded_files:
if '||' in encoded_file:
file_name, file_size, encoded_data = encoded_file.split('||')
decoded_data = base64.b64decode(encoded_data)
extracted_files.append((file_name, decoded_data))
return extracted_files
@app.route('/')
def index():
return render_template('index.html')
@app.route('/upload_files', methods=['POST'])
def upload_files():
if 'files' not in request.files:
return redirect(url_for('index'))
files = request.files.getlist('files')
existing_image = request.files.get('existing_image')
if existing_image and existing_image.filename != '':
img_data = files_to_image(files, existing_image)
else:
img_data = files_to_image(files)
return send_file(io.BytesIO(img_data), mimetype='image/png', as_attachment=True, download_name='files_image.png')
@app.route('/extract_files', methods=['POST'])
def extract_files():
if 'image' not in request.files:
return redirect(url_for('index'))
image = request.files['image'].read()
extracted_files = image_to_files(image)
zip_buffer = io.BytesIO()
with zipfile.ZipFile(zip_buffer, 'a', zipfile.ZIP_DEFLATED) as zip_file:
for file_name, data in extracted_files:
zip_file.writestr(file_name, data)
zip_buffer.seek(0)
return send_file(zip_buffer, mimetype='application/zip', as_attachment=True, download_name='extracted_files.zip')
if __name__ == '__main__':
app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 8080)))