优化输出参数

app.py

@@ -35,7 +35,7 @@ os.makedirs(SAVE_DIR, exist_ok=True)
 
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
-HTML_HEIGHT = 690
+HTML_HEIGHT = 500
 HTML_WIDTH = 500
 
 
@@ -203,9 +203,10 @@ def gen_shape(
     mesh = face_reduce_worker(mesh, target_face_num)
     save_folder = gen_save_folder()
 
+    progress(0.9,desc="Converting format")
     file_type = "obj"
-    objPath = export_mesh(mesh, save_folder, textured=False, type=file_type)
-    rel_objPath = os.path.relpath(objPath, SAVE_DIR)
+    sourceObjPath = export_mesh(mesh, save_folder, textured=False, type=file_type)
+    rel_objPath = os.path.relpath(sourceObjPath, SAVE_DIR)
     objPath = "/static/"+rel_objPath
 
     # for preview
@@ -219,11 +220,8 @@ def gen_shape(
     glbPath = "/static/"+rel_glbPath
 
 
-
-
     progress(1,desc="Complete")
-    return model_viewer_html, glbPath, objPath
-
+    return model_viewer_html, gr.update(value=sourceObjPath, interactive=True), glbPath, objPath
 
 
     
@@ -233,7 +231,7 @@ def get_example_img_list():
 example_imgs = get_example_img_list()
 
 HTML_OUTPUT_PLACEHOLDER = f"""
-<div style='height: {650}px; width: 100%; border-radius: 8px; border-color: #e5e7eb; border-style: solid; border-width: 1px; display: flex; justify-content: center; align-items: center;'>
+<div style='height: {500}px; width: 100%; border-radius: 8px; border-color: #e5e7eb; border-style: solid; border-width: 1px; display: flex; justify-content: center; align-items: center;'>
     <div style='text-align: center; font-size: 16px; color: #6b7280;'>
     <p style="color: #8d8d8d;">No mesh here.</p>
     </div>
@@ -274,8 +272,10 @@ with gr.Blocks().queue() as demo:
         with gr.Column(scale=6):
             gr.Markdown("#### Generated Mesh")
             html_export_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
-            objPath_output  = gr.Textbox(label="Obj Path")
-            glbPath_output  = gr.Textbox(label="Glb Path") 
+            file_export = gr.DownloadButton(label="Download", variant='primary', interactive=False)
+            with gr.Row():
+                objPath_output = gr.Text(label="Obj Path",interactive=False)
+                glbPath_output = gr.Text(label="Glb Path",interactive=False)
                 
         with gr.Column(scale=3):
             gr.Markdown("#### Image Examples")
@@ -285,7 +285,7 @@ with gr.Blocks().queue() as demo:
     gen_button.click(
         fn=gen_shape,
         inputs=[image,num_steps,cfg_scale,seed,octree_resolution,num_chunks,target_face_num, randomize_seed], 
-        outputs=[html_export_mesh, glbPath_output, objPath_output]
+        outputs=[html_export_mesh,file_export, glbPath_output, objPath_output]
     )    
 
  # https://discuss.huggingface.co/t/how-to-serve-an-html-file/33921/2