pyvista-ai
is a Python library that leverages AI to automatically adjust PyVista's Plotter
settings. By entering a prompt, you can easily apply visually optimal plot configurations.
- AI-powered automatic configuration: Optimizes
pyvista.Plotter
settings based on user prompts - Customizable: Automatically adjusts background color, window size, lighting settings, and more
- Simple API: Intuitive operations for adjusting 3D visualizations
pip install pyvista-ai
import asyncio
import pyvista as pv
from pyvista_ai import PlotterAgent
agent = PlotterAgent(model="gemini:pro")
async def main() -> None:
plotter = await agent.run("Enhance realistic shading and display in widescreen mode")
return plotter.show()
asyncio.run(main())
To use pyvista-ai
, you need to set the GEMINI API key as an environment variable.
export GEMINI_API_KEY="your-api-key-here"
- Support for offline AI models: Integration with
Hugging Face Transformers
for local AI model support - GUI Integration: Compatibility with
PyQt
andStreamlit
- Additional Features: AI-assisted optimal camera angles and automatic mesh color selection
MIT License
Bug reports and feature suggestions are welcome via GitHub Issues. Pull requests are also highly encouraged!