Update app.py via AI Editor

app.py

@@ -178,25 +178,29 @@ def right_main_static():
     return html.Div([
         dbc.Card([
             dbc.CardBody([
-                html.Div(id="chat-window", style={"minHeight": "60vh", "display": "flex", "flexDirection": "column", "justifyContent": "flex-end"}),
+                html.Div(id="chat-window", style={
+                    "height": "60vh",
+                    "overflowY": "auto",
+                    "display": "flex",
+                    "flexDirection": "column",
+                    "justifyContent": "flex-end",
+                    "backgroundColor": "#fff",
+                    "padding": "0.5rem",
+                    "borderRadius": "0.5rem"
+                }),
                 html.Div([
                     dcc.Textarea(
                         id="user-input",
                         placeholder="Type your question...",
                         style={"width": "100%", "height": "60px", "resize": "vertical", "wordWrap": "break-word"},
                         wrap="soft",
-                        maxLength=1000
+                        maxLength=1000,
+                        autoFocus=True
                     ),
                     html.Div([
                         dbc.Button("Send", id="send-btn", color="primary", className="mt-2 me-2", style={"minWidth": "100px"}),
                         dbc.Button("New Chat", id="new-chat-btn", color="secondary", className="mt-2", style={"minWidth": "110px"}),
                     ], style={"float": "right", "display": "flex", "gap": "0.5rem"}),
-                    dcc.Input(
-                        id="chat-name-input",
-                        placeholder="Enter chat name...",
-                        type="text",
-                        style={"width": "100%", "marginTop": "0.5rem", "display": "none"}
-                    ),
                 ], style={"marginTop": "1rem"}),
                 html.Div(id="error-message", style={"color": "#bb2124", "marginTop": "0.5rem"}),
             ])
@@ -211,7 +215,9 @@ app.layout = html.Div([
     html.Div([
         html.Div(left_navbar_static(), id='left-navbar', style={"width": "30vw", "height": "100vh", "position": "fixed", "left": 0, "top": 0, "zIndex": 2, "overflowY": "auto"}),
         html.Div(right_main_static(), id='right-main', style={"marginLeft": "30vw", "width": "70vw", "overflowY": "auto"})
-    ], style={"display": "flex"})
+    ], style={"display": "flex"}),
+    dcc.Store(id="clear-input", data=False),
+    dcc.Store(id="scroll-bottom", data=0)
 ])
 
 def _is_supported_doc(filename):
@@ -237,7 +243,7 @@ def assign_session_id(_):
     Output("error-message", "children"),
     Output("stream-interval", "disabled"),
     Output("stream-interval", "n_intervals"),
-    Output("chat-name-input", "style"),
+    Output("user-input", "value"),
     Input("session-id", "data"),
     Input("send-btn", "n_clicks"),
     Input("file-upload", "contents"),
@@ -245,11 +251,9 @@ def assign_session_id(_):
     Input("stream-interval", "n_intervals"),
     State("file-upload", "filename"),
     State("user-input", "value"),
-    State("chat-name-input", "value"),
-    State("chat-name-input", "style"),
     prevent_initial_call=False
 )
-def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, stream_n, file_names, user_input, chat_name, chat_name_style):
+def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, stream_n, file_names, user_input):
     trigger = callback_context.triggered[0]['prop_id'].split('.')[0] if callback_context.triggered else ""
     if not session_id:
         session_id = get_session_id()
@@ -334,72 +338,22 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
             threading.Thread(target=run_stream, args=(session_id, list(state["messages"])), daemon=True).start()
             start_streaming = True
 
-        # Handle New Chat button logic
-        show_chat_name = chat_name_style if isinstance(chat_name_style, dict) else {}
+        # Handle New Chat button logic: auto-name and reset
         if trigger == "new-chat-btn":
-            # If chat name input box is not yet visible, show it
-            if show_chat_name.get("display", "none") == "none":
-                show_chat_name["display"] = "block"
-                upload_cards = [uploaded_file_card(os.path.basename(f["name"]), f["is_img"]) for f in state.get("uploads", [])]
-                chat_history_items = [
-                    html.Li(
-                        html.Span(
-                            chat["name"],
-                            style={"fontSize": "0.92rem"}
-                        )
-                    ) for chat in state.get("chat_histories", [])[-6:]
-                ]
-                chat_cards = [
-                    chat_message_card(msg['content'], is_user=(msg['role'] == "user"))
-                    for msg in state.get("messages", [])
-                ] + (
-                    [chat_message_card(state["stream_buffer"], is_user=False)]
-                    if state.get("streaming", False) and state.get("stream_buffer", "") else []
-                )
-                return (
-                    upload_cards,
-                    chat_history_items,
-                    chat_cards,
-                    "",
-                    not state.get("streaming", False),
-                    0,
-                    show_chat_name
-                )
-            # If input box is visible and has a value, save chat history
+            chat_dialog = list(state.get("messages", []))
+            if not chat_dialog:
+                error = "No chat to save. Start chatting!"
             else:
-                chat_dialog = list(state.get("messages", []))
-                if not chat_dialog:
-                    error = "Cannot save empty chat."
-                else:
-                    chat_title = chat_name if chat_name and chat_name.strip() else "Chat " + datetime.datetime.utcnow().strftime("%Y-%m-%d %H:%M")
-                    state.setdefault("chat_histories", []).append({
-                        "name": chat_title,
-                        "dialog": chat_dialog
-                    })
-                    state["messages"] = []
-                    state["stream_buffer"] = ""
-                    state["streaming"] = False
-                    save_session_state(session_id)
-                    logger.info(f"Session {session_id}: Saved chat history '{chat_title}'")
-                show_chat_name["display"] = "none"
-                upload_cards = [uploaded_file_card(os.path.basename(f["name"]), f["is_img"]) for f in state.get("uploads", [])]
-                chat_history_items = [
-                    html.Li(
-                        html.Span(
-                            chat["name"],
-                            style={"fontSize": "0.92rem", "fontWeight": "bold"}
-                        )
-                    ) for chat in state.get("chat_histories", [])[-6:]
-                ]
-                return (
-                    upload_cards,
-                    chat_history_items,
-                    [],
-                    error,
-                    not state.get("streaming", False),
-                    0,
-                    show_chat_name
-                )
+                chat_title = "Chat " + datetime.datetime.utcnow().strftime("%Y-%m-%d %H:%M")
+                state.setdefault("chat_histories", []).append({
+                    "name": chat_title,
+                    "dialog": chat_dialog
+                })
+                state["messages"] = []
+                state["stream_buffer"] = ""
+                state["streaming"] = False
+                save_session_state(session_id)
+                logger.info(f"Session {session_id}: Saved chat history '{chat_title}'")
 
         # Handle polling for streaming
         if trigger == "stream-interval":
@@ -426,7 +380,7 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     "",
                     False,
                     stream_n+1,
-                    chat_name_style
+                    ""
                 )
             else:
                 chat_cards = []
@@ -448,7 +402,7 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     "",
                     True,
                     0,
-                    chat_name_style
+                    ""
                 )
 
         # Default: Build Uploads, Chat History and Chat Window
@@ -470,8 +424,32 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
         if state.get("streaming", False):
             if state.get("stream_buffer", ""):
                 chat_cards.append(chat_message_card(state["stream_buffer"], is_user=False))
-            return upload_cards, chat_history_items, chat_cards, error, False, 0, chat_name_style
-        return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, chat_name_style
+            return upload_cards, chat_history_items, chat_cards, error, False, 0, ""
+        # Always clear input after send
+        if trigger == "send-btn":
+            return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, ""
+        return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, user_input or ""
+
+# Clientside callback for pressing Enter to send
+app.clientside_callback(
+    """
+    function(n_clicks) {
+        var ta = document.getElementById('user-input');
+        if (ta) {
+            ta.addEventListener('keydown', function(e) {
+                if (e.key === 'Enter' && !e.shiftKey) {
+                    e.preventDefault();
+                    var sendBtn = document.getElementById('send-btn');
+                    if(sendBtn){sendBtn.click();}
+                }
+            }, {once: true});
+        }
+        return window.dash_clientside.no_update;
+    }
+    """,
+    Output('user-input', 'value'),
+    [Input('send-btn', 'n_clicks')]
+)
 
 @app_flask.after_request
 def set_session_cookie(resp):