Spaces:

TruthLens
/

ai3d

Running

App Files Files Community

TruthLens commited on Feb 25

Commit

1277ccd

verified ·

1 Parent(s): ab2c761

Update app.py

Browse files

Files changed (1) hide show

app.py +51 -46

app.py CHANGED Viewed

@@ -2,9 +2,10 @@ import os
 import numpy as np
 from PIL import Image
 from stl import mesh
-import plotly.graph_objects as go
 from flask import Flask, render_template_string, request, send_file
 from werkzeug.utils import secure_filename
 app = Flask(__name__)
 UPLOAD_FOLDER = 'uploads'
@@ -19,7 +20,7 @@ HTML_TEMPLATE = """
 <html lang="en">
 <head>
     <meta charset="UTF-8">
-    <title>3D Model Generator for 3D Printing</title>
 </head>
 <body>
     <h1>Upload Image to Generate 3D Printable Model</h1>
@@ -27,67 +28,71 @@ HTML_TEMPLATE = """
         <label>Image:</label><input type="file" name="image" required><br><br>
         <input type="submit" value="Upload & Generate 3D Model">
     </form>
-    {% if plot_div %}
-    <h2>3D Model Preview:</h2>
-    {{ plot_div|safe }}
-    <br>
     <a href="{{ download_link }}" download>Download STL File</a>
     {% endif %}
 </body>
 </html>
 """
-def generate_3d_model(image_path, output_stl_path, scale=1.0, height_scale=5.0):
-    image = Image.open(image_path).convert('L')
-    img_array = np.array(image)
-    height, width = img_array.shape
-    x, y = np.meshgrid(np.linspace(0, width * scale, width), np.linspace(0, height * scale, height))
-    z = (img_array / 255.0) * height_scale
-    # Create vertices and faces
-    vertices = np.column_stack((x.flatten(), y.flatten(), z.flatten()))
     faces = []
-    for i in range(height - 1):
-        for j in range(width - 1):
-            idx = i * width + j
-            faces.append([idx, idx + 1, idx + width])
-            faces.append([idx + 1, idx + width + 1, idx + width])
-    # Create STL mesh
-    model_mesh = mesh.Mesh(np.zeros(len(faces), dtype=mesh.Mesh.dtype))
-    for i, f in enumerate(faces):
-        for j in range(3):
-            model_mesh.vectors[i][j] = vertices[f[j], :]
-    model_mesh.save(output_stl_path)
-    fig = go.Figure(data=[go.Surface(z=z, x=x, y=y, colorscale='Gray')])
-    fig.update_layout(scene=dict(aspectratio=dict(x=1, y=1, z=0.3)), margin=dict(l=0, r=0, b=0, t=0))
-    return fig.to_html(full_html=False)
 @app.route('/', methods=['GET'])
 def index():
-    return render_template_string(HTML_TEMPLATE, plot_div=None, download_link=None)
 @app.route('/upload', methods=['POST'])
 def upload():
-    image = request.files['image']
-    image_path = os.path.join(app.config['UPLOAD_FOLDER'], secure_filename(image.filename))
-    image.save(image_path)
-    stl_filename = os.path.splitext(image.filename)[0] + '.stl'
-    output_stl_path = os.path.join(app.config['OUTPUT_FOLDER'], stl_filename)
-    plot_div = generate_3d_model(image_path, output_stl_path)
-    download_link = f"/download/{stl_filename}"
-    return render_template_string(HTML_TEMPLATE, plot_div=plot_div, download_link=download_link)
-@app.route('/download/<filename>')
 def download_file(filename):
     return send_file(os.path.join(app.config['OUTPUT_FOLDER'], filename), as_attachment=True)
 if __name__ == '__main__':
-    app.run(debug=True)

 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
+from io import BytesIO
+import cv2
 app = Flask(__name__)
 UPLOAD_FOLDER = 'uploads'
 <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>
         <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)
 if __name__ == '__main__':
+    app.run(debug=True, host='0.0.0.0', port=5000)