|
import os |
|
import numpy as np |
|
from PIL import Image |
|
from stl import mesh |
|
from flask import Flask, render_template_string, request, send_file |
|
from werkzeug.utils import secure_filename |
|
import cv2 |
|
|
|
app = Flask(__name__) |
|
|
|
UPLOAD_FOLDER = 'uploads' |
|
OUTPUT_FOLDER = 'outputs' |
|
os.makedirs(UPLOAD_FOLDER, exist_ok=True) |
|
os.makedirs(OUTPUT_FOLDER, exist_ok=True) |
|
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER |
|
app.config['OUTPUT_FOLDER'] = OUTPUT_FOLDER |
|
|
|
HTML_TEMPLATE = """ |
|
<!DOCTYPE html> |
|
<html lang="en"> |
|
<head> |
|
<meta charset="UTF-8"> |
|
<title>3D Printing Model Generator</title> |
|
</head> |
|
<body> |
|
<h1>Upload Image to Generate 3D Printable Model</h1> |
|
<form action="/upload" method="post" enctype="multipart/form-data"> |
|
<label>Image:</label><input type="file" name="image" required><br><br> |
|
<input type="submit" value="Upload & Generate 3D Model"> |
|
</form> |
|
{% if download_link %} |
|
<h2>3D Model Ready:</h2> |
|
<a href="{{ download_link }}" download>Download STL File</a> |
|
{% endif %} |
|
</body> |
|
</html> |
|
""" |
|
|
|
def image_to_height_map(image_path, size=(256, 256)): |
|
image = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE) |
|
resized = cv2.resize(image, size) |
|
normalized = resized / 255.0 |
|
return normalized |
|
|
|
def height_map_to_stl(height_map, output_path, scale=10.0): |
|
rows, cols = height_map.shape |
|
vertices = [] |
|
faces = [] |
|
|
|
for i in range(rows): |
|
for j in range(cols): |
|
z = height_map[i, j] * scale |
|
vertices.append([j, i, z]) |
|
|
|
vertices = np.array(vertices).reshape((rows, cols, 3)) |
|
face_list = [] |
|
|
|
for i in range(rows - 1): |
|
for j in range(cols - 1): |
|
face_list.append([vertices[i, j], vertices[i + 1, j], vertices[i, j + 1]]) |
|
face_list.append([vertices[i + 1, j], vertices[i + 1, j + 1], vertices[i, j + 1]]) |
|
|
|
faces_np = np.array(face_list) |
|
model = mesh.Mesh(np.zeros(faces_np.shape[0], dtype=mesh.Mesh.dtype)) |
|
|
|
for i, f in enumerate(faces_np): |
|
model.vectors[i] = f |
|
|
|
model.save(output_path) |
|
|
|
def create_3d_model(image_path): |
|
height_map = image_to_height_map(image_path) |
|
output_path = os.path.join(app.config['OUTPUT_FOLDER'], 'model.stl') |
|
height_map_to_stl(height_map, output_path) |
|
return output_path |
|
|
|
@app.route('/', methods=['GET']) |
|
def index(): |
|
return render_template_string(HTML_TEMPLATE, download_link=None) |
|
|
|
@app.route('/upload', methods=['POST']) |
|
def upload(): |
|
uploaded_file = request.files['image'] |
|
if uploaded_file: |
|
filename = secure_filename(uploaded_file.filename) |
|
image_path = os.path.join(app.config['UPLOAD_FOLDER'], filename) |
|
uploaded_file.save(image_path) |
|
|
|
stl_path = create_3d_model(image_path) |
|
return render_template_string(HTML_TEMPLATE, download_link=f'/download/{os.path.basename(stl_path)}') |
|
return render_template_string(HTML_TEMPLATE, download_link=None) |
|
|
|
@app.route('/download/<filename>', methods=['GET']) |
|
def download_file(filename): |
|
return send_file(os.path.join(app.config['OUTPUT_FOLDER'], filename), as_attachment=True) |
|
|
|
|
