mimic the naimate rig output stage

app.py

@@ -18,6 +18,7 @@ from gradio.processing_utils import move_resource_to_block_cache
 import traceback
 import sys
 import logging
+import requests
 
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -189,12 +190,15 @@ def generate_and_extract_glb(
     req: gr.Request,
 ) -> str:
     """
-    Runs the full text_to_3d and extract_glb pipeline internally.
+    Runs the full text_to_3d and extract_glb pipeline internally,
+    then uploads the GLB to the Node.js server and returns the persistent URL.
     """
     request_hash = str(req.session_hash)[:8]
     logging.info(f"[{request_hash}] ENTER generate_and_extract_glb")
     logging.info(f"[{request_hash}] Received parameters: prompt='{prompt}', seed={seed}, simplify={mesh_simplify}, tex_size={texture_size}, ...")
 
+    NODE_SERVER_UPLOAD_URL = "https://viverse-backend.onrender.com/api/upload-rigged-model"
+
     try:
         logging.info(f"[{request_hash}] Calling internal text_to_3d...")
         state, _ = text_to_3d(
@@ -213,20 +217,40 @@ def generate_and_extract_glb(
         if glb_path is None:
             logging.error(f"[{request_hash}] Internal extract_glb returned None path!")
             raise ValueError("Internal extract_glb failed to return GLB path")
-        logging.info(f"[{request_hash}] Internal extract_glb completed. Original GLB path: {glb_path}")
+        if not os.path.isfile(glb_path):
+            logging.error(f"[{request_hash}] GLB file not found at path: {glb_path}")
+            raise FileNotFoundError(f"Generated GLB file not found at {glb_path}")
+        logging.info(f"[{request_hash}] Internal extract_glb completed. GLB path: {glb_path}")
 
+        logging.info(f"[{request_hash}] Uploading GLB from {glb_path} to {NODE_SERVER_UPLOAD_URL}")
+        persistent_url = None
         try:
-            cached_glb_path = move_resource_to_block_cache(glb_path, demo)
-            logging.info(f"[{request_hash}] Copied GLB to Gradio cache: {cached_glb_path}")
-            logging.info(f"[{request_hash}] EXIT generate_and_extract_glb - Returning cached path: {cached_glb_path}")
-            return cached_glb_path
-        except Exception as cache_err:
-            logging.error(f"[{request_hash}] ERROR copying GLB to Gradio cache: {cache_err}", exc_info=True)
-            logging.warning(f"[{request_hash}] Falling back to returning original path: {glb_path}")
-            return glb_path
+            with open(glb_path, "rb") as f:
+                files = {"modelFile": (os.path.basename(glb_path), f, "model/gltf-binary")}
+                payload = {"clientType": "playcanvas"}
+                response = requests.post(NODE_SERVER_UPLOAD_URL, files=files, data=payload)
+                response.raise_for_status()
+                result = response.json()
+                persistent_url = result.get("persistentUrl")
+                if not persistent_url:
+                    logging.error(f"[{request_hash}] No persistent URL in Node.js server response: {result}")
+                    raise ValueError("Upload successful, but no persistent URL returned")
+                logging.info(f"[{request_hash}] Successfully uploaded to Node server. Persistent URL: {persistent_url}")
+        except requests.exceptions.RequestException as upload_err:
+            logging.error(f"[{request_hash}] FAILED to upload GLB to Node server: {upload_err}")
+            if hasattr(upload_err, 'response') and upload_err.response is not None:
+                logging.error(f"[{request_hash}] Node server response status: {upload_err.response.status_code}")
+                logging.error(f"[{request_hash}] Node server response text: {upload_err.response.text}")
+            raise gr.Error(f"Failed to upload result to backend server: {upload_err}")
+        except Exception as e:
+            logging.error(f"[{request_hash}] UNEXPECTED error during upload: {e}", exc_info=True)
+            raise gr.Error(f"Unexpected error during upload: {e}")
+
+        logging.info(f"[{request_hash}] EXIT generate_and_extract_glb - Returning persistent URL: {persistent_url}")
+        return persistent_url
 
     except Exception as e:
-        logging.error(f"[{request_hash}] ERROR in generate_and_extract_glb: {e}", exc_info=True)
+        logging.error(f"[{request_hash}] ERROR in generate_and_extract_glb pipeline: {e}", exc_info=True)
         raise gr.Error(f"Pipeline failed: {e}")