Update app.py

app.py

@@ -318,28 +318,64 @@ def process_obj_file(file_obj):
         return error_msg
 
 # Gradio Interface
-demo = gr.Interface(
-    fn=process_obj_file,
-    inputs=gr.File(
-        label="Upload OBJ File",
-        file_types=[".obj"],
-        type="file"
-    ),
-    outputs=gr.Textbox(
-        label="Analysis Results (First 10 Lines)",
-        lines=20,
-        max_lines=30
-    ),
-    title="🔷 OBJ File Analyzer",
-    description="Upload a 3D OBJ file to see the first 10 lines of raw content and mesh analysis",
-    examples=None,
-    cache_examples=False
-)
+# demo = gr.Interface(
+#     fn=process_obj_file,
+#     inputs=gr.File(
+#         label="Upload OBJ File",
+#         file_types=[".obj"],
+#         type="file"
+#     ),
+#     outputs=gr.Textbox(
+#         label="Analysis Results (First 10 Lines)",
+#         lines=20,
+#         max_lines=30
+#     ),
+#     title="🔷 OBJ File Analyzer",
+#     description="Upload a 3D OBJ file to see the first 10 lines of raw content and mesh analysis",
+#     examples=None,
+#     cache_examples=False
+# )
 
 if __name__ == "__main__":
     print("="*60, flush=True)
     print("🚀 Starting OBJ File Analyzer...", flush=True)
     print("="*60, flush=True)
+    load_models()
+
+
+    demo = gr.Interface(
+        fn=rignet_inference,
+        inputs=[
+            gr.File(label="Upload OBJ File", file_types=[".obj"], type="file"),
+            gr.Slider(0.02, 0.08, value=0.04, step=0.001, label="Bandwidth", info="Joint clustering density (default: 0.04)"),
+            gr.Slider(0.1, 3.0, value=1.0, step=0.1, label="Threshold (×10⁻⁵)", info="Joint filtering threshold (default: 1.0)")
+        ],
+        outputs=[
+            gr.File(label="Download Rig TXT"),
+            gr.Textbox(label="Status", lines=5)
+        ],
+        title="🎭 RigNet: Neural Rigging for 3D Characters",
+        description="""
+Upload a 3D character mesh (OBJ format) to automatically generate skeletal rig and skinning weights.
+
+**Recommended:** OBJ files with 1K-5K vertices work best.
+**Processing time:** 1-3 minutes on CPU depending on mesh complexity.
+        """,
+        article="""
+### 📚 About the Output
+
+The generated `*_rig.txt` file contains:
+- **joints**: 3D positions of skeletal joints
+- **root**: Root joint of the hierarchy  
+- **hier**: Parent-child relationships (skeleton hierarchy)
+- **skin**: Skinning weights for each vertex
+
+This format can be imported into 3D animation software.
+
+**Reference:** [RigNet: Neural Rigging for Articulated Characters (SIGGRAPH 2020)](https://arxiv.org/abs/2005.00559)
+        """,
+        allow_flagging="never"
+    )
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,