| from plantweb import render | |
| import re | |
| def sanitize_plantuml_code(plantuml_code): | |
| match = re.search(r'```plantuml\s*(.*?)\s*```', plantuml_code, re.DOTALL) | |
| if match: | |
| return match.group(1).strip() | |
| return "@startuml\n'Error\n@enduml" | |
| def render_plantuml(plantuml_code): | |
| try: | |
| svg_output, _, _, _ = render.render(plantuml_code, engine='plantuml', format='svg') | |
| return svg_output.decode('utf-8') | |
| except Exception as e: | |
| return f"Error rendering PlantUML: {e}" |