Bug fixes.

app.py

@@ -6474,7 +6474,7 @@ def create_ui(
 
         with gr.Row():
             # LEFT: legend + settings panel
-            with gr.Column(scale=0.5) as legend_col:
+            with gr.Column(scale=0.2) as legend_col:
                 with gr.Group(visible=False) as legend_group:
                     # with gr.Row():
                     gr.Markdown("### 🎨 Legend")
@@ -6590,7 +6590,8 @@ def create_ui(
                     end_btn   = gr.Button("End conversation", variant="stop", scale=1)
 
             # RIGHT: risk panel
-            with gr.Column(scale=2) as risk_col:
+            right_col_needed = show_risk_default or show_pii_hl_default or (show_infr_default != 0)
+            with gr.Column(scale=2, visible=right_col_needed) as risk_col:
                 risk_title = gr.Markdown("### 🛡️ Privacy Risk", visible=False)
                 risk_html = gr.HTML(create_risk_display(0), visible=False)
                 analysis_title = gr.Markdown("### 📝 Analysis", visible=False)
@@ -6815,7 +6816,7 @@ def create_ui(
                 "",                                 # 1. access_control_css - clear it
                 legend,                             # 2. legend_html
                 gr.update(visible=show_settings_v), # 3. settings_panel
-                gr.update(visible=True),            # 4. risk_col — always visible container
+                gr.update(visible=show_risk_v or show_pii_hl_v or (show_infr_v != 0)),  # 4. risk_col
                 use_rag,                            # 5. rag_st
                 eps,                                # 6. eps_st
                 mdl,                                # 7. mdl_st
@@ -7062,7 +7063,7 @@ def create_ui(
                 result[2],                                                           # warn_html
                 gr.update(value=result[3], visible=show_pii_now),                   # analysis_md (value + visibility)
                 privacy_update,                                                      # privacy_settings_html
-                gr.update(visible=True),                                             # risk_col — always visible container
+                gr.update(visible=show_risk_now or show_pii_now or (show_infr_v != 0 and not llm_errored)),  # risk_col
                 gr.update(visible=show_legend_now),                                  # legend_group
                 infer_card_update,                                                   # infer_card_html
                 gr.update(visible=show_risk_now),                                    # risk_title