Function main

examples/text_to_model.py:14–52

Create a 3D model from text. Args: prompt: The text prompt. negative_prompt: The negative text prompt. output_dir: The directory to save the output files.

(prompt: str, negative_prompt: str = None, output_dir: str = './output')

async def main(prompt: str, negative_prompt: str = None, output_dir: str = './output'):
    """
    Create a 3D model from text.

    Args:
        prompt: The text prompt.
        negative_prompt: The negative text prompt.
        output_dir: The directory to save the output files.
    """
    async with TripoClient() as client:
        # 创建任务
        task_id = await client.text_to_model(
            prompt=prompt,
            negative_prompt=negative_prompt
        )

        # 等待任务完成并显示进度
        task = await client.wait_for_task(task_id, verbose=True)

        if task.status == TaskStatus.SUCCESS:
            print(f"\nTask completed successfully!")

            # 创建输出目录（如果不存在）
            os.makedirs(output_dir, exist_ok=True)

            # 下载模型文件
            try:
                print("\nDownloading model files...")
                downloaded_files = await client.download_task_models(task, output_dir)

                # 打印下载的文件路径
                for model_type, file_path in downloaded_files.items():
                    if file_path:
                        print(f"Downloaded {model_type}: {file_path}")

            except Exception as e:
                print(f"Failed to download models: {str(e)}")
        else:
            print(f"\nTask failed with status: {task.status}")


if __name__ == "__main__":