Update app.py via AI Editor

app.py

@@ -465,7 +465,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 "",
                 history_index_clicked
             )
 
@@ -724,7 +724,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 "",
                     selected_history
                 )
             else:
@@ -751,7 +751,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 "",
                     selected_history
                 )
 
@@ -774,13 +774,15 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
         chat_cards = []
         for msg in chat_history:
             chat_cards.append(chat_message_card(msg['content'], is_user=(msg['role'] == "user")))
-        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, "", selected_history
-        if send_triggered:
+        # Only clear user input when message is SENT or when new chat is started
+        if trigger == "send-btn" or trigger == "hidden-send":
+            return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, "", selected_history
+        elif trigger == "new-chat-btn":
             return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, "", selected_history
-        return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, user_input or "", 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
+        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
 
 @app_flask.after_request
 def set_session_cookie(resp):