Update main.py

main.py

@@ -8,6 +8,7 @@ from langchain.chains import ConversationalRetrievalChain
 import pdfplumber
 import os
 import google.generativeai as genai
+from deep_translator import GoogleTranslator
 
 # 初始化 FastAPI 應用
 app = FastAPI()
@@ -58,6 +59,18 @@ if docs:
 else:
     raise ValueError("No documents found in the 'data' directory.")
 
+
+# 定義翻譯函數
+def translate_to_english(text):
+    return GoogleTranslator(source='auto', target='en').translate(text)
+
+def translate_to_chinese(text):
+    return GoogleTranslator(source='auto', target='zh-TW').translate(text)
+
+
+
+
+
 # 初始化 chat_history 作為全局變數
 chat_history = []
 
@@ -71,8 +84,11 @@ def invoke(question: str):
         )
 
         # 呼叫 QA chain 並處理回應
+        question = translate_to_english(question)
         response = qa_chain.invoke({"question": question, "chat_history": chat_history})
-        return response['answer']
+        response = translate_to_chinese(response['answer'])
+        # return response['answer']
+        return response
 
     except Exception as e:
         print(f"Error during invoke: {e}")