| import os | |
| import gradio as gr | |
| from gradio_ui import PIIMaskingUI | |
| def main(): | |
| """Launch the PII Detection & Masking UI.""" | |
| # Create output directory if it doesn't exist | |
| output_dir = "tmp" | |
| os.makedirs(output_dir, exist_ok=True) | |
| # Create the UI | |
| ui = PIIMaskingUI(output_dir=output_dir) | |
| # Get the Gradio Blocks interface | |
| demo = ui.demo | |
| # Launch the interface on a different port to avoid conflicts | |
| demo.launch( | |
| # share=True, | |
| # debug=True, | |
| server_name="0.0.0.0", | |
| mcp_server=True, | |
| # server_port=7869 # let the port be selected | |
| ) | |
| if __name__ == "__main__": | |
| main() | |