Update app.py

app.py

@@ -1,47 +1,49 @@
 import gradio as gr
 import subprocess
 import os
-import shutil
+
+def check_files():
+    # Let's see exactly what we have
+    files = os.listdir('.')
+    return f"Files found in root: {files}"
 
 def rig_model(input_file):
-    # Verify the official script exists
-    if not os.path.exists("demo_rigging.sh"):
-        return None, "Error: demo_rigging.sh not found in root. Clone failed."
+    # Absolute path check
+    script = os.path.abspath("demo_rigging.sh")
+    
+    if not os.path.exists(script):
+        return None, f"Error: demo_rigging.sh NOT found. Root contains: {os.listdir('.')}"
 
-    # Puppeteer outputs to this specific folder
     output_dir = "results/final_rigging"
-    if os.path.exists(output_dir):
-        shutil.rmtree(output_dir)
-    os.makedirs(output_dir, exist_ok=True)
-            
+    
     try:
-        # Run the bash command exactly as described in the GitHub
-        # We pass the input file path as the first argument
+        # Run the official bash command
         result = subprocess.run(
-            ["bash", "demo_rigging.sh", input_file.name], 
+            ["bash", script, input_file.name], 
             capture_output=True, 
             text=True
         )
         
-        # Search for the generated GLB in the output directory
-        generated_files = [f for f in os.listdir(output_dir) if f.endswith('.glb')]
-        
-        if generated_files:
-            return os.path.join(output_dir, generated_files[0]), "Rigging Complete!"
+        # Check for output
+        if os.path.exists(output_dir) and os.listdir(output_dir):
+            out_file = [f for f in os.listdir(output_dir) if f.endswith('.glb')][0]
+            return os.path.join(output_dir, out_file), "Success!"
         else:
-            return None, f"Script ran but no GLB was found in {output_dir}.\n\nLog:\n{result.stderr}"
+            return None, f"Failed. Bash Log:\n{result.stderr}\n{result.stdout}"
                 
     except Exception as e:
         return None, f"System Error: {str(e)}"
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🎭 Puppeteer Auto-Rigging (Seed3D Official Setup)")
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎭 Puppeteer Official Bash Setup")
+    gr.Markdown(check_files()) # This will show the file list at the top
     
-    input_file = gr.File(label="Upload Static GLB", file_types=[".glb"])
-    run_btn = gr.Button("Start Rigging", variant="primary")
+    with gr.Row():
+        input_file = gr.File(label="Upload GLB")
+        run_btn = gr.Button("Start Rigging", variant="primary")
     
-    out_file = gr.File(label="Rigged GLB Output")
-    out_logs = gr.Textbox(label="Console Output", lines=10)
+    out_file = gr.File(label="Rigged Result")
+    out_logs = gr.Textbox(label="Logs", lines=10)
 
     run_btn.click(fn=rig_model, inputs=[input_file], outputs=[out_file, out_logs])