Update app.py

app.py

@@ -6,8 +6,6 @@ import shutil
 
 # Configuración inicial para ZeroGPU
 os.environ['SPCONV_ALGO'] = 'native'
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
-os.environ["CUDA_VISIBLE_DEVICES"] = "0"  # Asegura usar la GPU asignada por ZeroGPU
 
 from typing import *
 import torch
@@ -102,13 +100,11 @@ def text_to_3d(
             "cfg_strength": slat_guidance_strength,
         },
     )
-    
-    # Renderizado de video
     video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
+    video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
+    video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
     video_path = os.path.join(user_dir, 'sample.mp4')
     imageio.mimsave(video_path, video, fps=15)
-    
-    # Empaquetar estado y limpiar memoria
     state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
     torch.cuda.empty_cache()
     return state, video_path
@@ -122,23 +118,15 @@ def extract_glb(
 ) -> Tuple[str, str]:
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     gs, mesh = unpack_state(state)
-    
-    # Exportar GLB con simplificación de malla
-    glb = postprocessing_utils.to_glb(
-        gs, 
-        mesh, 
-        simplify=mesh_simplify, 
-        texture_size=texture_size, 
-        verbose=False
-    )
+    glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
     glb_path = os.path.join(user_dir, 'sample.glb')
     glb.export(glb_path)
-    
     torch.cuda.empty_cache()
     return glb_path, glb_path
+
     
 # Interfaz Gradio
-with gr.Blocks(delete_cache=(600, 600)) as demo:
+with gr.Blocks() as demo:
     gr.Markdown("""
     # UTPL - Conversión de Texto a objetos 3D usando IA  
     ### Tesis: *"Objetos tridimensionales creados por IA: Innovación en entornos virtuales"*  
@@ -149,62 +137,35 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 
     with gr.Row():
         with gr.Column():
-            with gr.Tabs():
-                with gr.Tab(label="Input Image"):
-                    text_prompt = gr.Textbox(
-                        label="Text Prompt", 
-                        lines=5
-                    )
-
-            with gr.Accordion(".Generation Settings", open=False):
-                with gr.Column(variant="panel"):
-                    seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
-                    randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
-                    
-                    with gr.Group():
-                        gr.Markdown("#### Stage 1: Structure")
-                        ss_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance", value=7.5, step=0.1)
-                        ss_sampling_steps = gr.Slider(1, 50, label="Steps", value=12, step=1)
-                    
-                    with gr.Group():
-                        gr.Markdown("#### Stage 2: Detail")
-                        slat_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance", value=3.0, step=0.1)
-                        slat_sampling_steps = gr.Slider(1, 50, label="Steps", value=12, step=1)
+            text_prompt = gr.Textbox(label="Text Prompt", lines=5)
             
-            generate_btn = gr.Button("Generate 3D Asset", variant="primary", size="lg")
+            with gr.Accordion(label="Generation Settings", open=False):
+                seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
+                randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
+                with gr.Row():
+                    ss_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
+                    ss_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=25, step=1)
+                with gr.Row():
+                    slat_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
+                    slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=25, step=1)
+
+            generate_btn = gr.Button("Generate")
             
-            with gr.Accordion("GLB Export Settings", open=False):
-                with gr.Column(variant="panel"):
-                    mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify Mesh", value=0.95, step=0.01)
-                    texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
+            with gr.Accordion(label="GLB Extraction Settings", open=False):
+                mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)
+                texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
             
-            extract_glb_btn = gr.Button("Export GLB", interactive=False, size="lg")
+            with gr.Row():
+                extract_glb_btn = gr.Button("Extract GLB", interactive=False)
 
-        # Columna de salida
-        with gr.Column(scale=3, min_width=600):
-            with gr.Group():
-                video_output = gr.Video(
-                    label="3D Preview", 
-                    autoplay=True, 
-                    loop=True, 
-                    height=300,
-                    show_label=False
-                )
-                model_output = LitModel3D(
-                    label="3D Model Viewer", 
-                    exposure=10.0, 
-                    height=400
-                )
+        with gr.Column():
+            video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
+            model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
             
             with gr.Row():
-                download_glb = gr.DownloadButton(
-                    label="Download GLB File", 
-                    interactive=False,
-                    variant="secondary",
-                    size="lg"
-                )
-
-    output_buf = gr.State({})
+                download_glb = gr.DownloadButton(label="Download GLB", interactive=False)  
+    
+    output_buf = gr.State()
 
     # Eventos
     demo.load(start_session)