From fc23345c6d23447417d89c4b45fbb58c40c434f1 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 17 Jun 2025 14:59:03 -0700
Subject: [PATCH] Send the default input to the notebook textbox when switching
 2 columns to 1 (instead of the output)

---
 modules/ui_session.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/ui_session.py b/modules/ui_session.py
index 431aeeb0..a69e155b 100644
--- a/modules/ui_session.py
+++ b/modules/ui_session.py
@@ -64,7 +64,7 @@ def handle_default_to_notebook_change(show_two_columns, default_input, default_o
             gr.update(visible=True),
             gr.update(visible=False),
             notebook_input,
-            notebook_input,
+            "",
             gr.update(value=notebook_prompt, choices=utils.get_available_prompts()),
             gr.update(),
             gr.update(),
@@ -77,7 +77,7 @@ def handle_default_to_notebook_change(show_two_columns, default_input, default_o
             gr.update(),
             gr.update(),
             gr.update(),
-            default_output,
+            default_input,
             gr.update(value=default_prompt, choices=utils.get_available_prompts())
         ]