File size: 3,733 Bytes
06626f0 1130bd7 2d118fc a996c52 97b657f 1130bd7 97b657f 1130bd7 5fa3d68 1130bd7 e7f75f3 8bcf8dc 1130bd7 5fa3d68 1130bd7 184f2d6 8609612 97b657f 8609612 2d118fc 1130bd7 09d275f 1130bd7 8609612 1130bd7 8609612 184f2d6 1130bd7 184f2d6 1130bd7 8609612 1130bd7 06626f0 1130bd7 2d118fc bbc169d 1130bd7 5fa3d68 1130bd7 5fa3d68 1130bd7 8609612 1130bd7 2d118fc bbc169d 1130bd7 8609612 1130bd7 5fa3d68 1130bd7 |
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 100 |
import gradio as gr
import zipfile
import io
import re
import traceback
import os
import tempfile
def process_files(phase, objecten_file_data, templates_zip_file_data):
try:
# Read and process the 'objecten.txt' file
content = objecten_file_data.decode('utf-8')
t = content.strip().split('\n\n')
phases = ["iFAT", "(i)SAT"]
objs = {p: [] for p in phases}
if phase in phases:
objs = {phase: []}
else:
objs = {p: [] for p in phases}
regObject = r"\d{4}[a-zA-Z]{2}"
for g in t:
ls = [line.strip() for line in g.strip().split('\n') if line.strip()]
k = ls[0].strip()
if k in objs:
objs[k] = ls[1:]
else:
error_msg = f"Key [{k}] is not recognized."
return None, error_msg
# Extract object codes
for k in objs:
objs[k] = [re.search(regObject, o).group(0) for o in objs[k] if re.search(regObject, o)]
# Read template files from the uploaded ZIP file
templates_zip_data = templates_zip_file_data
with zipfile.ZipFile(io.BytesIO(templates_zip_data), 'r') as zip_ref:
template_files = {info.filename: zip_ref.read(info.filename) for info in zip_ref.infolist()}
# Create a temporary file for the output ZIP
temp_dir = tempfile.mkdtemp()
output_zip_path = os.path.join(temp_dir, 'output.zip')
# Create the ZIP file
with zipfile.ZipFile(output_zip_path, 'w') as zf:
for k in objs:
regPhase = ''
if k == '(i)SAT':
regPhase = r'sat'
elif k == 'iFAT':
regPhase = r'fat'
# Filter template files for this phase
phase_templates = []
for filename, file_data in template_files.items():
if re.search(regPhase, filename, re.IGNORECASE):
phase_templates.append((filename, file_data))
if not phase_templates:
error_msg = f"Phase {k} has no templates."
return None, error_msg
for o in objs[k]:
folder_path = f"{o}/"
for template_filename, file_content in phase_templates:
# Adjust filename if needed
if re.search(r"hut_\d{4}[a-zA-Z]{2}", template_filename, re.IGNORECASE):
adjusted_filename = template_filename[:4] + o + template_filename[10:]
else:
adjusted_filename = template_filename
file_path = folder_path + adjusted_filename
zf.writestr(file_path, file_content)
return output_zip_path, "" # Return the path of the ZIP file and an empty error log
except Exception as e:
# Capture the full traceback
error_msg = traceback.format_exc()
return None, error_msg
phase_options = ['iFAT', '(i)SAT', 'All']
interface = gr.Interface(
fn=process_files,
inputs=[
gr.Dropdown(choices=phase_options, label="Select Phase"),
gr.File(label="Upload 'objecten.txt' File", type='binary'),
gr.File(label="Upload Templates ZIP File", type='binary')
],
outputs=[
gr.File(label="Download ZIP File"), # This now receives a file path
gr.Textbox(label="Error Log") # Textbox to display the error log
],
title="Template Folder Generator",
description="Upload 'objecten.txt' and a ZIP file containing template files to generate folders and files."
)
interface.launch()
|