Update app.py via AI Editor

app.py

@@ -257,7 +257,6 @@ def user_input_card():
                     style={"width": "100%", "height": "60px", "resize": "vertical", "wordWrap": "break-word"},
                     wrap="soft",
                     maxLength=1000,
-                    n_submit=0,
                     n_blur=0,
                 ),
                 dcc.Store(id="enter-triggered", data=False),
@@ -296,7 +295,6 @@ app.layout = html.Div([
     dcc.Store(id="enter-pressed", data=False)
 ])
 
-# JS callback to intercept Enter/Shift+Enter for dcc.Textarea
 app.clientside_callback(
     """
     function(n, value) {
@@ -325,7 +323,6 @@ def _is_supported_doc(filename):
     return ext in [".txt", ".pdf", ".md", ".docx"]
 
 def _extract_text_from_upload(filepath, ext):
-    # Only .txt and .md are extracted for now
     if ext in [".txt", ".md"]:
         try:
             with open(filepath, "r", encoding="utf-8", errors="ignore") as f:
@@ -383,7 +380,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
         chat_histories = state.get("chat_histories", [])
         uploads = state.get("uploads", [])
 
-        # Handle clickable chat history loading
         history_index_clicked = None
         if trigger == "" and chat_history_clicks:
             for idx, n in enumerate(chat_history_clicks):
@@ -402,7 +398,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
             state["streaming"] = False
             save_session_state(session_id)
             logger.info(f"Session {session_id}: Loaded chat history index {history_index_clicked}")
-            # Rebuild downstream outputs with this dialog
             chat_cards = []
             for msg in state["messages"]:
                 chat_cards.append(chat_message_card(msg['content'], is_user=(msg['role'] == "user")))
@@ -430,7 +425,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                 history_index_clicked
             )
 
-        # Handle File Upload
         file_was_uploaded_and_sent = False
         if trigger == "file-upload" and file_contents and file_names:
             uploads = []
@@ -456,12 +450,10 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
             save_session_state(session_id)
             logger.info(f"Session {session_id}: Uploaded files {[u['name'] for u in uploads]}")
 
-        # Determine if send was triggered (via send-btn, hidden-send, or enter)
         send_triggered = False
         if trigger == "send-btn" or trigger == "hidden-send":
             send_triggered = True
 
-        # Handle Send
         if send_triggered and user_input and user_input.strip():
             question = user_input.strip()
             state["messages"].append({"role": "user", "content": question})
@@ -472,10 +464,8 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
             def run_stream(session_id, messages, question):
                 try:
                     system_prompt = load_system_prompt()
-                    # Retrieve relevant context from global RAG
                     rag_chunks = []
                     try:
-                        # Search global docs
                         global_embeds = []
                         global_texts = []
                         global_fnames = []
@@ -487,7 +477,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                         if global_rag:
                             for r in global_rag:
                                 rag_chunks.append(f"Global doc [{r['filename']}]:\n{r['text'][:1000]}")
-                        # Search user docs
                         user_embeds, user_texts, user_fnames = get_user_embeddings(session_id)
                         user_rag = semantic_search(question, user_embeds, user_texts, user_fnames, top_k=2)
                         if user_rag:
@@ -544,7 +533,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
             threading.Thread(target=run_stream, args=(session_id, list(state["messages"]), question), daemon=True).start()
             start_streaming = True
 
-        # Handle New Chat button logic: auto-name and reset
         if trigger == "new-chat-btn":
             chat_dialog = list(state.get("messages", []))
             if not chat_dialog:
@@ -561,7 +549,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                 save_session_state(session_id)
                 logger.info(f"Session {session_id}: Saved chat history '{chat_title}'")
 
-        # Handle polling for streaming
         if trigger == "stream-interval":
             chat_history = state.get("messages", [])
             chat_cards = []
@@ -621,7 +608,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     selected_history
                 )
 
-        # Default: Build Uploads, Chat History and Chat Window
         chat_history = state.get("messages", [])
         uploads = state.get("uploads", [])
         chat_histories = state.get("chat_histories", [])
@@ -645,7 +631,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
             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
-        # Always clear input after send
         if send_triggered:
             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