Complete rewrite: Clean structure with robust visualizations

Fix: Improve probability plots with labels and better error handling

Fix: Add missing UI components for plots and results

Update: Simplified Gradio app with fine-tuned model