Update app.py via AI Editor

app.py

@@ -598,6 +598,28 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                 threading.Thread(target=run_stream_for_doc, args=(session_id, list(state["messages"]), doc_question), daemon=True).start()
                 start_streaming = True
 
+            # After file upload (regardless of doc or image), DO NOT clear the user-input box, i.e. always return no_update below
+            chat_history = state.get("messages", [])
+            uploads = state.get("uploads", [])
+            chat_histories = state.get("chat_histories", [])
+            upload_cards = [uploaded_file_card(os.path.basename(f["name"]), f["is_img"]) for f in uploads]
+            chat_history_items = [
+                html.Li(
+                    dbc.Button(
+                        chat["name"],
+                        id={"type": "chat-history-item", "index": i},
+                        color="tertiary",
+                        outline=False,
+                        style={"width": "100%", "textAlign": "left", "fontWeight": "bold" if i == len(chat_histories)-1 else "normal", "fontSize": "0.92rem", "marginBottom": "0.2rem"}
+                    )
+                )
+                for i, chat in enumerate(chat_histories[-6:])
+            ]
+            chat_cards = []
+            for msg in chat_history:
+                chat_cards.append(chat_message_card(msg['content'], is_user=(msg['role'] == "user")))
+            return upload_cards, chat_history_items, chat_cards, error, (not state.get("streaming", False)), 0, no_update, selected_history
+
         send_triggered = False
         if trigger == "send-btn" or trigger == "hidden-send":
             send_triggered = True