Update app.py

app.py

@@ -32,6 +32,8 @@ pipeline = None
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+logger.info("Trellis App: Script starting.")
+
 def start_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     os.makedirs(user_dir, exist_ok=True)
@@ -313,15 +315,39 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 
 # Launch the Gradio app and FastAPI server
 if __name__ == "__main__":
-    logger.info("Initializing Trellis Pipeline...")
-    # Make pipeline global so Gradio functions and API endpoint can access it
-    pipeline = TrellisTextTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-text-xlarge")
-    pipeline.cuda()
-    logger.info("Trellis Pipeline Initialized.")
-
-    # Start the background API server using the integration module
-    trellis_fastAPI_integration.start_api_thread(pipeline)
+    logger.info("Trellis App: Initializing Trellis Pipeline...")
+    try:
+        # Make pipeline global so Gradio functions and API endpoint can access it
+        pipeline = TrellisTextTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-text-xlarge")
+        pipeline.cuda()
+        logger.info("Trellis App: Trellis Pipeline Initialized successfully.")
+    except Exception as e:
+        logger.error(f"Trellis App: FATAL ERROR initializing pipeline: {e}", exc_info=True)
+        pipeline = None # Ensure pipeline is None if initialization failed
+        # Optionally exit if pipeline is critical
+        # import sys
+        # sys.exit("Pipeline initialization failed.")
+
+    # Start the background API server using the integration module only if pipeline loaded
+    if pipeline:
+        logger.info("Trellis App: Attempting to start FastAPI server thread...")
+        try:
+            api_thread = trellis_fastAPI_integration.start_api_thread(pipeline)
+            if api_thread and api_thread.is_alive():
+                 logger.info("Trellis App: FastAPI server thread started successfully (is_alive check passed).")
+            elif api_thread:
+                 logger.warning("Trellis App: FastAPI server thread was created but is not alive shortly after starting.")
+            else:
+                 logger.error("Trellis App: start_api_thread returned None, thread not created.")
+        except Exception as e:
+            logger.error(f"Trellis App: Error occurred during start_api_thread call: {e}", exc_info=True)
+    else:
+        logger.error("Trellis App: Skipping FastAPI server start because pipeline failed to initialize.")
 
     # Launch the Gradio interface (blocking call)
-    logger.info("Launching Gradio Demo...")
-    demo.launch()
+    logger.info("Trellis App: Launching Gradio Demo...")
+    try:
+        demo.launch()
+        logger.info("Trellis App: Gradio Demo launched.") # This might not be reached if launch blocks indefinitely
+    except Exception as e:
+        logger.error(f"Trellis App: Error launching Gradio demo: {e}", exc_info=True)