minor change

db/mongodb.py

@@ -13,44 +13,31 @@ def connect_mongodb(db_name, collection_name):
 
 
 
-def save_file_to_mongodb(uploaded_file, db_name="ppt", collection_name="root_file", file_name=None, file_tail=".pptx"):
+def save_file_to_mongodb(uploaded_file, db_name="ppt", collection_name="root_file", file_tail=".pptx"):
     """
-    Lưu file PowerPoint (pptx) vào MongoDB bằng GridFS
-    nhưng không lưu nếu tên file đã tồn tại.
+    Lưu file vào MongoDB bằng GridFS mà không kiểm tra trùng lặp.
     
     :param uploaded_file: đối tượng UploadedFile từ Streamlit
     :param db_name: Tên database trong MongoDB
     :param collection_name: Tên collection GridFS
-    :param file_name: Tên file muốn lưu (không cần .pptx). Nếu để None, lấy tên gốc.
-    :return: file_id nếu lưu thành công, None nếu file đã tồn tại
+    :param file_tail: Phần mở rộng mặc định của file nếu không có
+    :return: file_id nếu lưu thành công
     """
     client = MongoClient("mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0")
     db = client[db_name]
     fs = gridfs.GridFS(db, collection=collection_name)
 
     # Xác định tên file
-    if not file_name:
-        # Lấy tên file từ uploaded_file (VD: "slide.pptx")
-        file_name = uploaded_file.name
-    else:
-        # Nếu người dùng chỉ truyền tên, thêm .pptx nếu chưa có
-        if not file_name.endswith(file_tail):
-            file_name = file_name + file_tail
-
-    # Kiểm tra file đã tồn tại trong MongoDB chưa
-    existing_file = fs.find_one({"filename": file_name})
-    if existing_file:
-        print(f"⚠️ File '{file_name}' đã tồn tại trong MongoDB. Không lưu lại. Xin vui lòng đổi tên.")
-        client.close()
-        return None
+    file_name = uploaded_file.name
 
     # Đảm bảo con trỏ file đang ở đầu
     uploaded_file.seek(0)
     file_bytes = uploaded_file.read()
 
-    # Lưu nội dung file (bytes) vào MongoDB
+    # Lưu file vào MongoDB (không kiểm tra trùng lặp)
     file_id = fs.put(file_bytes, filename=file_name)
     print(f"✅ File '{file_name}' đã được lưu vào '{collection_name}' với ID: {file_id}")
+    
     client.close()
     return file_id
 

excel/excel_translate.py

@@ -48,7 +48,7 @@ def translate_xlsx(file_id: str, sheet_name: str = None, from_lang: str = 'en',
                     cell_map[key] = cell
         
         # Gọi hàm dịch theo dạng bulk
-        translated_dict = translate_text_dict(text_dict, source_lang=from_lang, target_lang=target_lang, gemini_api=gemini_api)
+        translated_dict = translate_text_dict(text_dict, target_lang=target_lang, gemini_api=gemini_api)
         
         # Cập nhật lại các cell với nội dung đã dịch
         for key, cell in cell_map.items():

pages/upload.py

@@ -24,7 +24,6 @@ st.title("Upload PPTX to MongoDB")
 
 # st.set_option("server.fileUploader.allowMediaFiles", True)
 uploaded_file = st.file_uploader("Chọn file")
-file_name_input = st.text_input("Tên file để lưu (không cần .pptx)", value="")
 target_lang = st.selectbox("Chọn ngôn ngữ dịch", ["english", "vietnamese"])
 
 final_pptx_id = None  # Biến lưu ID file sau khi xử lý
@@ -35,7 +34,7 @@ if uploaded_file is not None:
         st.write(f"Detected file type: {file_type}")
         if file_type == "PPTX":
             
-            file_id = save_file_to_mongodb(uploaded_file=uploaded_file, file_name=file_name_input)
+            file_id = save_file_to_mongodb(uploaded_file=uploaded_file)
             st.write(f"File ID: {file_id}")
 
             xml_file_id = ppt_to_xml_mongodb(file_id)

powerpoint/xml_handling.py

@@ -83,7 +83,14 @@ def ppt_to_xml_mongodb(ppt_file_id: str, db_name="ppt"):
     :return: ID của file XML trong MongoDB (original_xml)
     """
     # Kết nối MongoDB
-    client = MongoClient("mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0")
+    client = MongoClient(
+        "mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0",
+        connectTimeoutMS=60000,  # 60 giây thay vì 20 giây
+        serverSelectionTimeoutMS=60000,  # Chờ phản hồi lâu hơn
+        socketTimeoutMS=60000,  # Tăng thời gian chờ socket
+        tls=True,
+        tlsAllowInvalidCertificates=True  # Giữ kết nối lâu hơn
+    )    
     db = client[db_name]
 
     fs_ppt = gridfs.GridFS(db, collection="root_file")  # PPT gốc
@@ -120,6 +127,7 @@ def ppt_to_xml_mongodb(ppt_file_id: str, db_name="ppt"):
         xml_file_id = fs_xml.put(xml_output, filename=f"{ppt_file.filename}.xml")
 
         print(f"✅ XML đã được lưu vào MongoDB (original_xml) với file_id: {xml_file_id}")
+        client.close()
 
         return xml_file_id