import streamlit as st import zipfile import tempfile import shutil import os import json from PIL import Image import numpy as np import cv2 from io import BytesIO import time import random st.set_page_config(page_title="MOCKJET – 100+ Mockups in Seconds", layout="wide") st.title("🚀 MOCKJET - 100+ Mockups in Seconds") st.write("Upload your base mockups and design images to generate product mockups in seconds!") # --- ファイル保存用ディレクトリ TMP_ROOT = tempfile.mkdtemp() BASES_DIR = os.path.join(TMP_ROOT, "bases") DESIGNS_DIR = os.path.join(TMP_ROOT, "designs") OUT_DIR = os.path.join(TMP_ROOT, "out") os.makedirs(BASES_DIR, exist_ok=True) os.makedirs(DESIGNS_DIR, exist_ok=True) os.makedirs(OUT_DIR, exist_ok=True) # --- Helper: 画像ファイル保存 def save_uploaded_files(files, save_dir): valid_ext = [".png", ".jpg", ".jpeg"] saved_paths = [] for uploaded_file in files: fname = uploaded_file.name ext = os.path.splitext(fname)[1].lower() if ext in valid_ext: out_path = os.path.join(save_dir, fname) with open(out_path, "wb") as f: f.write(uploaded_file.getbuffer()) saved_paths.append(out_path) elif ext == ".zip": # ZIP解凍 with zipfile.ZipFile(uploaded_file, "r") as zip_ref: for zipinfo in zip_ref.infolist(): ext2 = os.path.splitext(zipinfo.filename)[1].lower() if ext2 in valid_ext: zip_ref.extract(zipinfo, save_dir) saved_paths.append(os.path.join(save_dir, zipinfo.filename)) return saved_paths # --- ファイルアップロードUI st.header("Upload Mockup Bases (ZIP or PNG/JPG/JPEG files)") bases_files = st.file_uploader("Upload base files", type=["png", "jpg", "jpeg", "zip"], accept_multiple_files=True) st.header("Upload Design Images (.png, .jpg, .jpeg)") design_files = st.file_uploader("Upload design files", type=["png", "jpg", "jpeg"], accept_multiple_files=True) st.header("Upload templates.json (coordinates of print area)") templates_file = st.file_uploader("Upload templates.json", type=["json"]) # 例: サイドバーにヘッダー付きで置く st.sidebar.header("Utilities") # Download demokit.zip with open("demokit.zip", "rb") as file: st.sidebar.download_button( label="⬇️ Let's try it! Download Sample demokit (ZIP)", data=file, file_name="demokit.zip", mime="application/zip" ) st.sidebar.markdown("---") # 区切り線 st.sidebar.markdown("#### How to use mockjet-web") st.sidebar.video("mockup_web_demo.mp4") # Option: サイドバーで直接動画を再生 st.sidebar.markdown("---") # 区切り線 st.sidebar.markdown("#### How to create template.json file") st.sidebar.video("coordinate_maker.mp4") # 1) Space のリポ ID は環境変数 HF_SPACE_ID に入っています repo_id = os.getenv("HF_SPACE_ID", "junjiro1129/mockjet_web") # 2) resolve/main 形式で HTML ファイルの直リンクを組み立て coord_url = f"https://huggingface.co/spaces/{repo_id}/resolve/main/coordinatemaker.html" # 3) クリックで別タブを開く Markdown リンクを貼る st.sidebar.markdown( """ 🛠️ Create templates.json file/Edit Print Area (Coordinate Maker) """, unsafe_allow_html=True ) # --- ファイル保存 bases_paths, designs_paths = [], [] if bases_files: bases_paths = save_uploaded_files(bases_files, BASES_DIR) if design_files: designs_paths = save_uploaded_files(design_files, DESIGNS_DIR) # --- templates.jsonロード templates = None if templates_file is not None: templates = json.load(templates_file) # --- 画像表示 if bases_paths: st.subheader("Mockup Bases") num_cols = 6 cols = st.columns(num_cols) for idx, img_path in enumerate(bases_paths): with cols[idx % num_cols]: st.image(img_path, caption=os.path.basename(img_path), width=180) if designs_paths: st.subheader("Design Images") num_cols2 = 6 cols2 = st.columns(num_cols2) for idx, img_path in enumerate(designs_paths): with cols2[idx % num_cols2]: st.image(img_path, caption=os.path.basename(img_path), width=180) # --- 縦横比維持でワープ貼付 def warp_design_onto_mockup_fit(base_img, design_img, coords): base_cv = np.array(base_img.convert("RGBA")) design_cv = np.array(design_img.convert("RGBA")) h_des, w_des = design_cv.shape[:2] pts_dst = np.array(coords, dtype=np.float32) # 外接長方形 widthA = np.linalg.norm(pts_dst[0] - pts_dst[1]) widthB = np.linalg.norm(pts_dst[3] - pts_dst[2]) maxWidth = int(max(widthA, widthB)) heightA = np.linalg.norm(pts_dst[0] - pts_dst[3]) heightB = np.linalg.norm(pts_dst[1] - pts_dst[2]) maxHeight = int(max(heightA, heightB)) # デザイン画像をアスペクト比維持でフィット scale = min(maxWidth / w_des, maxHeight / h_des) new_w = int(w_des * scale) new_h = int(h_des * scale) design_resized = cv2.resize(design_cv, (new_w, new_h), interpolation=cv2.INTER_AREA) # 背景透明画像を中央配置 design_bg = np.zeros((maxHeight, maxWidth, 4), dtype=np.uint8) x_offset = (maxWidth - new_w) // 2 y_offset = (maxHeight - new_h) // 2 design_bg[y_offset:y_offset+new_h, x_offset:x_offset+new_w] = design_resized pts_src = np.array([[0,0],[maxWidth-1,0],[maxWidth-1,maxHeight-1],[0,maxHeight-1]], dtype=np.float32) M = cv2.getPerspectiveTransform(pts_src, pts_dst) warped = cv2.warpPerspective(design_bg, M, (base_cv.shape[1], base_cv.shape[0]), borderMode=cv2.BORDER_CONSTANT, borderValue=(0, 0, 0, 0)) mask = warped[:,:,3] > 0 base_cv[mask] = warped[mask] return Image.fromarray(base_cv) # --- モックアップ生成 output_images = [] if st.button("Generate Mockups") and bases_paths and designs_paths and templates: st.info("Generating mockups... Please wait.") total_tasks = len(bases_paths) * len(designs_paths) progress_bar = st.progress(0) eta_text = st.empty() start_time = time.time() current_task = 0 for base_path in bases_paths: base_name = os.path.basename(base_path) if base_name not in templates: st.warning(f"{base_name} is not found in templates.json, skipping.") continue coords = templates[base_name]['print_area'] # [[x1,y1], [x2,y2], [x3,y3], [x4,y4]] if isinstance(coords[0], dict): coords = [[p[0], p[1]] for p in coords.values()] elif isinstance(coords[0], list): coords = coords else: st.warning("Invalid coords format.") continue for design_path in designs_paths: base_img = Image.open(base_path).convert("RGBA") design_img = Image.open(design_path).convert("RGBA") result_img = warp_design_onto_mockup_fit(base_img, design_img, coords) result_name = f"{os.path.splitext(base_name)[0]}_{os.path.splitext(os.path.basename(design_path))[0]}.png" result_path = os.path.join(OUT_DIR, result_name) result_img.save(result_path) output_images.append(result_path) # --- プログレスバー・ETA更新ここから --- current_task += 1 progress = int(current_task / total_tasks * 100) progress_bar.progress(progress) elapsed = time.time() - start_time avg_time = elapsed / current_task remaining = avg_time * (total_tasks - current_task) min_left = int(remaining // 60) sec_left = int(remaining % 60) eta_text.markdown(f"**Estimated time left:** {min_left} min {sec_left} sec") # --- ここまで --- eta_text.markdown("✅ **All images processed!**") st.success(f"Generated {len(output_images)} mockup images!") # --- 生成画像表示＆ZIPダウンロード if output_images: st.subheader("Generated Mockups") cols3 = st.columns(min(len(output_images), 4)) for idx, out_img in enumerate(output_images): with cols3[idx % 4]: st.image(out_img, caption=os.path.basename(out_img), width=180) zip_buf = BytesIO() with zipfile.ZipFile(zip_buf, "w") as zipf: for img_path in output_images: zipf.write(img_path, os.path.basename(img_path)) st.download_button("Download All as ZIP", zip_buf.getvalue(), file_name="mockups.zip", mime="application/zip") # --- 後始末 def cleanup_tmp(): shutil.rmtree(TMP_ROOT, ignore_errors=True) import atexit atexit.register(cleanup_tmp)