Update app.py

app.py

@@ -1394,14 +1394,12 @@ def langgraph_tab6_main(query: str, file=None):
         files = file if isinstance(file, list) else [file] if file else []
         all_docs, file_names = [], []
         for f in files:
-            # 使用原有的 get_file_path 函式
-            path = get_file_path(f)
+            path = get_file_path(f)  # 使用原有的 get_file_path
             if not path:
                 print("get_file_path returned None for file:", f)
                 continue
             file_names.append(os.path.basename(path))
             print("Processing file:", path)
-            # 根據副檔名選擇 Loader
             if path.lower().endswith(".pdf"):
                 loader = PyPDFLoader(path)
             elif path.lower().endswith(".docx"):
@@ -1412,7 +1410,6 @@ def langgraph_tab6_main(query: str, file=None):
             print("Docs loaded from", path, ":", docs)
             all_docs.extend(docs)
         
-        # 建立檢索器，合併所有檔案內容
         if not all_docs:
             print("No document content read. file_names:", file_names)
             retriever = None
@@ -1420,11 +1417,17 @@ def langgraph_tab6_main(query: str, file=None):
             chunks = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50).split_documents(all_docs)
             db = FAISS.from_documents(chunks, embeddings)
             retriever = db.as_retriever()
-            # 設定全域 session_retriever，供文件相關工具使用
+            # 將建立好的 retriever 指派到全域變數 session_retriever
             global session_retriever
             session_retriever = retriever
+            # 同時建立 Document QA 的 ConversationalRetrievalChain，供 uploaded_qa_tool_func 使用
+            global session_qa_chain
+            session_qa_chain = ConversationalRetrievalChain.from_llm(
+                llm=llm_gpt4,
+                retriever=retriever,
+                memory=ConversationBufferMemory(memory_key="chat_history", return_messages=True),
+            )
         
-        # 建立 LangGraph 流程
         graph = build_langgraph_pipeline()
         state = {"query": query, "file_names": file_names}
         if retriever is not None: