Update app.py via AI Editor

app.py

@@ -236,7 +236,6 @@ def left_navbar_static():
     ], style={"padding": "1rem", "backgroundColor": "#f8f9fa", "height": "100vh", "overflowY": "auto"})
 
 def chat_box_card():
-    # Use a scrollable vertical container for chat history
     return dbc.Card(
         dbc.CardBody([
             html.Div(
@@ -330,6 +329,21 @@ app.clientside_callback(
     State('user-input', 'value')
 )
 
+# Clientside callback to scroll chat window to bottom when scroll-bottom is incremented
+app.clientside_callback(
+    """
+    function(scrollIndex) {
+        var chatContainer = document.getElementById('chat-window-container');
+        if (chatContainer) {
+            chatContainer.scrollTop = chatContainer.scrollHeight;
+        }
+        return null;
+    }
+    """,
+    Output('clear-input', 'data'),  # dummy output
+    Input('scroll-bottom', 'data')
+)
+
 def _is_supported_doc(filename):
     ext = os.path.splitext(filename)[1].lower()
     return ext in [".txt", ".pdf", ".md", ".docx", ".xlsx"]
@@ -399,6 +413,7 @@ def assign_session_id(_):
     Output("stream-interval", "disabled"),
     Output("stream-interval", "n_intervals"),
     Output("user-input", "value"),
+    Output("scroll-bottom", "data"),
     Input("session-id", "data"),
     Input("send-btn", "n_clicks"),
     Input("file-upload", "contents"),
@@ -406,9 +421,10 @@ def assign_session_id(_):
     Input('hidden-send', 'n_clicks'),
     State("file-upload", "filename"),
     State("user-input", "value"),
+    State("scroll-bottom", "data"),
     prevent_initial_call=False
 )
-def main_callback(session_id, send_clicks, file_contents, stream_n, hidden_send_clicks, file_names, user_input):
+def main_callback(session_id, send_clicks, file_contents, stream_n, hidden_send_clicks, file_names, user_input, scroll_bottom):
     trigger = callback_context.triggered[0]['prop_id'].split('.')[0] if callback_context.triggered else ""
     session_id = session_id or get_session_id()
     session_lock = get_session_lock(session_id)
@@ -551,7 +567,7 @@ def main_callback(session_id, send_clicks, file_contents, stream_n, hidden_send_
             chat_cards = []
             for msg in chat_history:
                 chat_cards.append(chat_message_card(msg['content'], is_user=(msg['role'] == "user")))
-            return upload_cards, chat_cards, error, (not state.get("streaming", False)), 0, no_update
+            return upload_cards, chat_cards, error, (not state.get("streaming", False)), 0, no_update, scroll_bottom+1
 
         send_triggered = False
         if trigger == "send-btn" or trigger == "hidden-send":
@@ -650,7 +666,8 @@ def main_callback(session_id, send_clicks, file_contents, stream_n, hidden_send_
                     "",
                     False,
                     stream_n+1,
-                    no_update
+                    no_update,
+                    scroll_bottom+1
                 )
             else:
                 chat_cards = []
@@ -663,7 +680,8 @@ def main_callback(session_id, send_clicks, file_contents, stream_n, hidden_send_
                     "",
                     True,
                     0,
-                    no_update
+                    no_update,
+                    scroll_bottom+1
                 )
 
         chat_history = state.get("messages", [])
@@ -673,11 +691,11 @@ def main_callback(session_id, send_clicks, file_contents, stream_n, hidden_send_
         for msg in chat_history:
             chat_cards.append(chat_message_card(msg['content'], is_user=(msg['role'] == "user")))
         if trigger == "send-btn" or trigger == "hidden-send":
-            return upload_cards, chat_cards, error, (not state.get("streaming", False)), 0, ""
+            return upload_cards, chat_cards, error, (not state.get("streaming", False)), 0, "", scroll_bottom+1
         elif trigger == "file-upload":
-            return upload_cards, chat_cards, error, (not state.get("streaming", False)), 0, no_update
+            return upload_cards, chat_cards, error, (not state.get("streaming", False)), 0, no_update, scroll_bottom+1
         else:
-            return upload_cards, chat_cards, error, (not state.get("streaming", False)), 0, no_update
+            return upload_cards, chat_cards, error, (not state.get("streaming", False)), 0, no_update, scroll_bottom
 
 @app_flask.after_request
 def set_session_cookie(resp):