Update app.py via AI Editor

app.py

@@ -8,7 +8,7 @@ import tempfile
 import glob
 from flask import Flask, request as flask_request, make_response
 import dash
-from dash import dcc, html, Input, Output, State, callback_context
+from dash import dcc, html, Input, Output, State, callback_context, no_update
 import dash_bootstrap_components as dbc
 import openai
 import base64
@@ -458,6 +458,7 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                 )
                 for i, chat in enumerate(chat_histories[-6:])
             ]
+            # Don't clear user input on chat history load; keep value unchanged
             return (
                 upload_cards,
                 chat_history_items,
@@ -465,7 +466,7 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                 "",
                 True,
                 0,
-                user_input if user_input is not None else "",
+                no_update,
                 history_index_clicked
             )
 
@@ -717,6 +718,7 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     )
                     for i, chat in enumerate(state.get("chat_histories", [])[-6:])
                 ]
+                # Don't clear user input during streaming refresh; keep value unchanged
                 return (
                     upload_cards,
                     chat_history_items,
@@ -724,7 +726,7 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     "",
                     False,
                     stream_n+1,
-                    user_input if user_input is not None else "",
+                    no_update,
                     selected_history
                 )
             else:
@@ -744,6 +746,7 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     )
                     for i, chat in enumerate(state.get("chat_histories", [])[-6:])
                 ]
+                # Don't clear user input after streaming ends; keep value unchanged
                 return (
                     upload_cards,
                     chat_history_items,
@@ -751,7 +754,7 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     "",
                     True,
                     0,
-                    user_input if user_input is not None else "",
+                    no_update,
                     selected_history
                 )
 
@@ -780,9 +783,10 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
         elif trigger == "new-chat-btn":
             return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, "", selected_history
         elif trigger == "file-upload":
-            return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, user_input if user_input is not None else "", selected_history
+            # Do not clear or update user input when a file is uploaded!
+            return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, no_update, selected_history
         else:
-            return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, user_input if user_input is not None else "", selected_history
+            return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, no_update, selected_history
 
 @app_flask.after_request
 def set_session_cookie(resp):