text2room / app.py
Mattia2700's picture
chore: update requirements
37402b8
raw
history blame
1.02 kB
import os
import subprocess
import gradio as gr
def load_mesh(mesh_file_name):
try:
subprocess.run(["python", "text2room/generate_scene.py"])
except Exception as e:
print(e)
search_files(".ply")
return mesh_file_name
def search_files(extension):
# recursively search for files with the given extension
files = []
# print content of the current directory
print(os.listdir())
# print content of the text2room directory
print(os.listdir("text2room"))
return files
iface = gr.Interface(fn=load_mesh,
inputs=gr.Model3D(),
outputs=gr.Model3D(
clear_color=[0.0, 0.0, 0.0, 0.0], label="3D Model"),
examples=[
[os.path.join(os.path.dirname(__file__), "files/Bunny.obj")],
[os.path.join(os.path.dirname(__file__), "files/Duck.glb")],
[os.path.join(os.path.dirname(__file__), "files/Fox.gltf")],
[os.path.join(os.path.dirname(__file__), "files/face.obj")],
],
)
if __name__ == "__main__":
iface.launch()