Update app.py

app.py

@@ -1,34 +1,53 @@
 import gradio as gr
-import subprocess
 import os
+import subprocess
+import sys
 import shutil
 
+# 1. FORCE PYTHON TO SEE YOUR UPLOADED FOLDERS
+# This fixes the 'No module named third_partys' error
+current_dir = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(current_dir)
+sys.path.append(os.path.join(current_dir, "skeleton"))
+sys.path.append(os.path.join(current_dir, "skinning"))
+
 def run_rigging(input_model):
-    # 1. Create a clean work folder
-    if os.path.exists("temp_input"):
-        shutil.rmtree("temp_input")
-    os.makedirs("temp_input", exist_ok=True)
-    
-    # 2. Save your uploaded file into the input folder
-    input_path = os.path.join("temp_input", "model.glb")
+    # 2. MANUALLY CREATE MISSING FOLDERS
+    # This fixes the 'No such file or directory' errors in the .sh script
+    folders = [
+        "results/skel_results",
+        "results/skin_results/generate",
+        "results/final_rigging",
+        "results/rigging",
+        "temp_input"
+    ]
+    for folder in folders:
+        os.makedirs(folder, exist_ok=True)
+
+    # 3. SETUP INPUT
+    input_path = "temp_input/model.glb"
     shutil.copy(input_model.name, input_path)
-    
-    # 3. Trigger the Seed3D rigging script
-    # We use subprocess to run the .sh file you dragged into the repo
+
+    # 4. RUN THE RIGGING
     try:
+        # We run the bash script you uploaded
         subprocess.run(["bash", "demo_rigging.sh"], check=True)
-        # Seed3D usually saves results in a 'results' folder
-        output_path = "results/rigging/model_rigged.glb" 
-        return output_path
+        
+        # Check if the file was actually made
+        output_path = "results/rigging/model_rigged.glb"
+        if os.path.exists(output_path):
+            return output_path
+        else:
+            return "Error: Rigging script finished but no file was generated. Check logs for 3D math errors."
+            
     except Exception as e:
-        return f"Error during rigging: {str(e)}"
+        return f"System Error: {str(e)}"
 
-# This creates the "API Door" for your AI Studio app
 demo = gr.Interface(
     fn=run_rigging,
-    inputs=gr.File(label="Upload Model"),
-    outputs=gr.File(label="Download Rigged Model"),
-    api_name="rig" 
+    inputs=gr.File(label="Upload GLB"),
+    outputs=gr.File(label="Rigged Result"),
+    api_name="rig"
 )
 
 demo.launch()