Update app.py via AI Editor

app.py

@@ -222,6 +222,7 @@ def right_main_static():
 app.layout = html.Div([
     dcc.Store(id="session-id", storage_type="local"),
     dcc.Location(id="url"),
+    dcc.Store(id="selected-history", data=None),
     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"})
@@ -234,6 +235,20 @@ def _is_supported_doc(filename):
     ext = os.path.splitext(filename)[1].lower()
     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:
+                text = f.read()
+            return text
+        except Exception as e:
+            logger.error(f"Error reading {filepath}: {e}")
+            return ""
+    # For .pdf/.docx, could add extraction with extra dependencies
+    else:
+        return ""
+
 @app.callback(
     Output("session-id", "data"),
     Input("url", "href"),
@@ -254,27 +269,80 @@ def assign_session_id(_):
     Output("stream-interval", "disabled"),
     Output("stream-interval", "n_intervals"),
     Output("user-input", "value"),
+    Output("selected-history", "data"),
     Input("session-id", "data"),
     Input("send-btn", "n_clicks"),
     Input("file-upload", "contents"),
     Input("new-chat-btn", "n_clicks"),
     Input("stream-interval", "n_intervals"),
+    Input({"type": "chat-history-item", "index": dash.ALL}, "n_clicks"),
     State("file-upload", "filename"),
     State("user-input", "value"),
+    State("selected-history", "data"),
+    State("chat-history-list", "children"),
     prevent_initial_call=False
 )
-def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, stream_n, file_names, user_input):
+def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, stream_n, chat_history_clicks, file_names, user_input, selected_history, chat_history_list_children):
     trigger = callback_context.triggered[0]['prop_id'].split('.')[0] if callback_context.triggered else ""
-    if not session_id:
-        session_id = get_session_id()
+    session_id = session_id or get_session_id()
     session_lock = get_session_lock(session_id)
     with session_lock:
         load_session_state(session_id)
         state = get_session_state(session_id)
         error = ""
         start_streaming = False
+        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):
+                if n:
+                    history_index_clicked = idx
+                    break
+        elif trigger.startswith("{\"type\":\"chat-history-item\""):
+            for idx, n in enumerate(chat_history_clicks):
+                if n:
+                    history_index_clicked = idx
+                    break
+        if history_index_clicked is not None and history_index_clicked < len(chat_histories):
+            selected_dialog = chat_histories[history_index_clicked]["dialog"]
+            state["messages"] = selected_dialog.copy()
+            state["stream_buffer"] = ""
+            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")))
+            upload_cards = [uploaded_file_card(os.path.basename(f["name"]), f["is_img"]) for f in state.get("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:])
+            ]
+            return (
+                upload_cards,
+                chat_history_items,
+                chat_cards,
+                "",
+                True,
+                0,
+                "",
+                history_index_clicked
+            )
 
-        # Handle File Upload
+        # Handle File Upload -- now, if supported, send to OpenAI as system message
+        file_was_uploaded_and_sent = False
         if trigger == "file-upload" and file_contents and file_names:
             uploads = []
             if not isinstance(file_contents, list):
@@ -290,10 +358,69 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                 with open(fp, "wb") as f:
                     f.write(base64.b64decode(data))
                 uploads.append({"name": fname, "is_img": is_img, "path": fp})
+                # If document, extract text and send to OpenAI as a message (system or user)
+                if _is_supported_doc(n) and not is_img:
+                    text = _extract_text_from_upload(fp, ext)
+                    if text.strip():
+                        doc_intro = f"(User uploaded document '{n}'; content below):\n\n{text[:3800]}"
+                        # Add as user message and trigger streaming
+                        state["messages"].append({"role": "user", "content": doc_intro})
+                        state["streaming"] = True
+                        state["stream_buffer"] = ""
+                        file_was_uploaded_and_sent = True
+                        logger.info(f"Session {session_id}: Uploaded doc '{n}' sent to OpenAI")
             state["uploads"].extend(uploads)
             save_session_state(session_id)
             logger.info(f"Session {session_id}: Uploaded files {[u['name'] for u in uploads]}")
 
+        # If a supported doc was uploaded, start streaming OpenAI response
+        if file_was_uploaded_and_sent:
+            def run_stream(session_id, messages):
+                try:
+                    system_prompt = load_system_prompt()
+                    msg_list = [{"role": "system", "content": system_prompt}]
+                    for m in messages:
+                        msg_list.append({"role": m["role"], "content": m["content"]})
+                    response = openai.ChatCompletion.create(
+                        model="gpt-3.5-turbo",
+                        messages=msg_list,
+                        max_tokens=700,
+                        temperature=0.2,
+                        stream=True,
+                    )
+                    reply = ""
+                    for chunk in response:
+                        delta = chunk["choices"][0]["delta"]
+                        content = delta.get("content", "")
+                        if content:
+                            reply += content
+                            session_lock = get_session_lock(session_id)
+                            with session_lock:
+                                load_session_state(session_id)
+                                state = get_session_state(session_id)
+                                state["stream_buffer"] = reply
+                                save_session_state(session_id)
+                    session_lock = get_session_lock(session_id)
+                    with session_lock:
+                        load_session_state(session_id)
+                        state = get_session_state(session_id)
+                        state["messages"].append({"role": "assistant", "content": reply})
+                        state["stream_buffer"] = ""
+                        state["streaming"] = False
+                        save_session_state(session_id)
+                    logger.info(f"Session {session_id}: Doc Q&A: Assistant: {reply}")
+                except Exception as e:
+                    session_lock = get_session_lock(session_id)
+                    with session_lock:
+                        load_session_state(session_id)
+                        state = get_session_state(session_id)
+                        state["streaming"] = False
+                        state["stream_buffer"] = ""
+                        save_session_state(session_id)
+                    logger.error(f"Session {session_id}: Streaming error (doc upload): {e}")
+            threading.Thread(target=run_stream, args=(session_id, list(state["messages"])), daemon=True).start()
+            start_streaming = True
+
         # Handle Send
         if trigger == "send-btn" and user_input and user_input.strip():
             state["messages"].append({"role": "user", "content": user_input})
@@ -377,11 +504,15 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                 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(
+                        dbc.Button(
                             chat["name"],
-                            style={"fontSize": "0.92rem", "fontWeight": "bold" if i == len(state.get("chat_histories", []))-1 else "normal"}
+                            id={"type": "chat-history-item", "index": i},
+                            color="tertiary",
+                            outline=False,
+                            style={"width": "100%", "textAlign": "left", "fontWeight": "bold" if i == len(state.get("chat_histories", []) )-1 else "normal", "fontSize": "0.92rem", "marginBottom": "0.2rem"}
                         )
-                    ) for i, chat in enumerate(state.get("chat_histories", [])[-6:])
+                    )
+                    for i, chat in enumerate(state.get("chat_histories", [])[-6:])
                 ]
                 return (
                     upload_cards,
@@ -390,7 +521,8 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     "",
                     False,
                     stream_n+1,
-                    ""
+                    "",
+                    selected_history
                 )
             else:
                 chat_cards = []
@@ -399,11 +531,15 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                 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(
+                        dbc.Button(
                             chat["name"],
-                            style={"fontSize": "0.92rem", "fontWeight": "bold" if i == len(state.get("chat_histories", []))-1 else "normal"}
+                            id={"type": "chat-history-item", "index": i},
+                            color="tertiary",
+                            outline=False,
+                            style={"width": "100%", "textAlign": "left", "fontWeight": "bold" if i == len(state.get("chat_histories", []))-1 else "normal", "fontSize": "0.92rem", "marginBottom": "0.2rem"}
                         )
-                    ) for i, chat in enumerate(state.get("chat_histories", [])[-6:])
+                    )
+                    for i, chat in enumerate(state.get("chat_histories", [])[-6:])
                 ]
                 return (
                     upload_cards,
@@ -412,7 +548,8 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                     "",
                     True,
                     0,
-                    ""
+                    "",
+                    selected_history
                 )
 
         # Default: Build Uploads, Chat History and Chat Window
@@ -422,11 +559,15 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
         upload_cards = [uploaded_file_card(os.path.basename(f["name"]), f["is_img"]) for f in uploads]
         chat_history_items = [
             html.Li(
-                html.Span(
+                dbc.Button(
                     chat["name"],
-                    style={"fontSize": "0.92rem", "fontWeight": "bold" if i == len(chat_histories)-1 else "normal"}
+                    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:])
+            )
+            for i, chat in enumerate(chat_histories[-6:])
         ]
         chat_cards = []
         for msg in chat_history:
@@ -434,11 +575,11 @@ 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, ""
+            return upload_cards, chat_history_items, chat_cards, error, False, 0, "", selected_history
         # 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 ""
+            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
 
 @app_flask.after_request
 def set_session_cookie(resp):