modified: app.py

modified: custom_utils/smpl_visualizer.py

app.py

@@ -1,5 +1,15 @@
 import gradio as gr
 from custom_utils.setup_env import run_setup_script
+import viser
+from custom_utils import smpl_visualizer
+
+
+smpl_visualizer.run()
+
+iframe_html = """
+<iframe src="http://localhost:8080" width="100%" height="500" style="border:none;"></iframe>
+"""
+
 
 def update_submenu(clothing_type):
     """Функция для динамического обновления подменю в зависимости от типа одежды."""
@@ -101,10 +111,10 @@ with gr.Blocks(title="3D Garment Generator", theme=gr.themes.Default(text_size="
                 )
 
         with gr.Column(scale=3):
-            viewer = gr.Model3D(label="3D Просмотрщик", interactive=True, height=500,
-                               clear_color=[255.0, 255.0, 255.0, 0.0]
-                               )
-            
+            # viewer = gr.Model3D(label="3D Просмотрщик", interactive=True, height=500,
+            #                    clear_color=[255.0, 255.0, 255.0, 0.0]
+            #                    )
+            viewer = gr.HTML(iframe_html, label="3D Рендер")
             with gr.Row():
                 save_render_btn = gr.Button("Сохранить рендер")
                 save_obj_btn = gr.Button("Сохранить OBJ")

custom_utils/smpl_visualizer.py

@@ -302,5 +302,5 @@ def make_gui_elements(
     return out
 
 
-if __name__ == "__main__":
+def run():
     tyro.cli(main, description=__doc__)