Update app.py

app.py

@@ -484,6 +484,37 @@ def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
         logger.error(f"Error in run_texture: {str(e)}")
         raise
 
+@spaces.GPU(duration=5)
+@torch.no_grad()
+def run_full_api(image, seed: int = 0, num_inference_steps: int = 50, guidance_scale: float = 7.5, simplify: bool = True, target_face_num: int = DEFAULT_FACE_NUMBER, req: gr.Request = None):
+    try:
+        logger.info("Running run_full_api")
+        # Handle FileData dict or URL
+        if isinstance(image, dict):
+            image_path = image.get("path") or image.get("url")
+            if not image_path:
+                logger.error("Invalid image input: no path or URL provided")
+                raise ValueError("Invalid image input: no path or URL provided")
+            if image_path.startswith("http"):
+                temp_image_path = os.path.join(TMP_DIR, f"input_{get_random_hex()}.png")
+                image_path = download_image(image_path, temp_image_path)
+        elif isinstance(image, str) and image.startswith("http"):
+            temp_image_path = os.path.join(TMP_DIR, f"input_{get_random_hex()}.png")
+            image_path = download_image(image, temp_image_path)
+        else:
+            image_path = image
+            if not isinstance(image, str) or not os.path.exists(image_path):
+                logger.error(f"Invalid image path: {image_path}")
+                raise ValueError(f"Invalid image path: {image_path}")
+
+        image_seg, mesh_path, textured_glb_path = run_full(image_path, seed, num_inference_steps, guidance_scale, simplify, target_face_num, req)
+        session_hash = os.path.basename(os.path.dirname(textured_glb_path))
+        logger.info(f"Generated textured model at /files/{session_hash}/{os.path.basename(textured_glb_path)}")
+        return {"file_url": f"/files/{session_hash}/{os.path.basename(textured_glb_path)}"}
+    except Exception as e:
+        logger.error(f"Error in run_full_api: {str(e)}")
+        raise
+
 # Define Gradio API endpoint
 try:
     logger.info("Initializing Gradio API interface")