initial commit

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.stl filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,3 @@
+flagged/
+__pycache__/
+.vscode

Dockerfile

@@ -0,0 +1,16 @@
+FROM python:3.10-slim
+
+ENV DEBIAN_FRONTEND noninteractive
+ENV LANG C.UTF-8
+ENV GRADIO_SERVER_NAME 0.0.0.0
+
+RUN useradd -m -u 1000 user
+WORKDIR /home/user/app
+
+RUN --mount=target=requirements.txt,source=requirements.txt :\
+    && pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+EXPOSE 7860
+
+ENTRYPOINT ["python", "app.py"]

app.py

@@ -0,0 +1,48 @@
+import trimesh
+import tempfile
+import atexit
+import shutil
+import pathlib
+import gradio as gr
+
+
+def cleanup_temp_directories():
+    print("Deleting temporary files")
+    for temp_dir in temp_directories:
+        try:
+            shutil.rmtree(temp_dir)
+        except FileNotFoundError:
+            print(f"Could not delete directory {temp_dir}")
+
+
+def stl2glb(glb_file):
+    if not glb_file.endswith(".stl"):
+        raise gr.Error("Please upload a .stl file")
+
+    temp_dir = pathlib.Path(tempfile.mkdtemp())
+    temp_directories.append(temp_dir)
+
+    mesh = trimesh.load(glb_file)
+    glb = trimesh.exchange.gltf.export_glb(mesh)
+
+    output_file = (temp_dir / glb_file).with_suffix(".glb")
+    with open(output_file, "wb") as f:
+        f.write(glb)
+
+    return str(output_file)
+
+
+if __name__ == "__main__":
+    temp_directories = []
+    atexit.register(cleanup_temp_directories)
+
+    demo = gr.Interface(
+        fn=stl2glb,
+        inputs=gr.Model3D(label="STL File"),
+        outputs=gr.Model3D(label="GLB File"),
+        examples=["space_shuttle.stl"],
+        title="STL2GLB Converter",
+        description="Convert a .stl file to a .glb file",
+        analytics_enabled=False,
+    )
+    demo.launch(share=False)

requirements.txt

@@ -0,0 +1,2 @@
+trimesh
+gradio

space_shuttle.stl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f8b96742bda278c6069d96f5054a320e3a21db2ad9db15c493a79f1f7b6482c6
+size 7500084