Spaces:
Running
Running
File size: 2,163 Bytes
823cc73 |
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 |
import gradio as gr
import trimesh
import numpy as np
from PIL import Image
import base64
def text_image_to_glb(text, image):
"""
Converts text and image into a 3D GLB object.
Args:
text (str): Input text.
image (PIL.Image): Input image.
Returns:
bytes: GLB file as bytes.
str: HTML for 3D preview.
"""
# Create a simple 3D mesh (e.g., a cube)
mesh = trimesh.creation.box(extents=[1, 1, 1]) # Create a 1x1x1 cube
# Add text to the 3D object (as metadata)
mesh.metadata['text'] = text
# Convert the image to a texture and apply it to the mesh
if image:
image = np.array(image) # Convert PIL image to numpy array
texture = trimesh.visual.TextureVisuals(image=image)
mesh.visual = texture
# Export the mesh to GLB format
glb_data = trimesh.exchange.gltf.export_glb(mesh)
# Encode the GLB data as base64 for the HTML preview
glb_base64 = base64.b64encode(glb_data).decode("utf-8")
# Create an HTML preview for the 3D object
html_preview = f"""
<div style="text-align:center;">
<h3>3D Object Preview</h3>
<p><strong>Text:</strong> {text}</p>
<p><strong>Image:</strong> Applied as texture</p>
<model-viewer style="width:100%; height:400px;" src="data:model/gltf-binary;base64,{glb_base64}" auto-rotate camera-controls></model-viewer>
<script type="module" src="https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js"></script>
</div>
"""
return glb_data, html_preview
# Gradio interface
def create_interface():
iface = gr.Interface(
fn=text_image_to_glb,
inputs=[
gr.Textbox(label="Enter Text"),
gr.Image(label="Upload Image", type="pil")
],
outputs=[
gr.File(label="Download GLB File"),
gr.HTML(label="3D Object Preview")
],
title="Text & Image to 3D GLB Converter",
description="Convert text and an image into a 3D GLB object with a preview."
)
return iface
# Launch the app
if __name__ == "__main__":
iface = create_interface()
iface.launch() |