update UI/UX

home.py

@@ -2,21 +2,26 @@ import streamlit as st
 import pandas as pd
 import numpy as np
 
-st.title("Some Streamlit Demo, maybe")
+st.set_page_config(page_title="Machine Translation App", page_icon="🌍", layout="centered")
 
-st.sidebar.header("Input")
-num_rows = st.sidebar.slider("Number of rows", min_value=10, max_value=100, value=20)
-num_cols = st.sidebar.slider("Number of columns", min_value=2, max_value=10, value=3)
+st.title("🌍 Machine Translation App")
 
-data = np.random.randn(num_rows, num_cols)
-columns = [f"Column {i+1}" for i in range(num_cols)]
-df = pd.DataFrame(data, columns=columns)
+st.write(
+    "Chào mừng bạn đến với ứng dụng dịch máy tự động!\n"
+    "Ứng dụng này giúp bạn dịch tài liệu một cách nhanh chóng và chính xác."
+)
 
-st.subheader("Generated Data Table")
-st.dataframe(df)
+st.header("📌 Hướng dẫn sử dụng")
+st.write("1. Truy cập [trang upload](#).")
+st.write("2. Chọn tệp tin cần dịch (hỗ trợ .docx, .txt, .pdf).")
+st.write("3. Chọn ngôn ngữ muốn dịch sang.")
+st.write("4. Nhấn **Upload** để bắt đầu dịch.")
+st.write("5. Chờ một vài phút để hệ thống xử lý.")
+st.write("6. Khi hoàn tất, nhấn **Tải về** để nhận tệp đã dịch.")
+
+# Điều hướng đến trang Upload
+to_upload = st.button("📤 Đi đến trang Upload")
+if to_upload:
+    st.switch_page("upload.py")  # Điều hướng đến trang upload
 
-st.subheader("Line Chart of the Data")
-st.line_chart(df)
 
-st.subheader("Statistics")
-st.write(df.describe())

pages/upload.py

@@ -8,129 +8,53 @@ from translate.translator import translate_text_dict
 from powerpoint.pptx_object import create_translated_ppt
 from excel.excel_translate import translate_xlsx, translate_csv
 from word.word_translate import translate_docx_from_mongodb
-
 import dotenv
 import os
 
 dotenv.load_dotenv(".env")
-# st.set_option("server.maxUploadSize", 1024)
-
-# Cấu hình API key
-api_key = os.getenv("GEMINI_API_KEY")
-genai.configure(api_key=api_key)
-model = genai.GenerativeModel("gemini-2.0-flash")
-
-# Giao diện Streamlit
-st.title("Please chose your PPTX, Excel file to translate")
-
-# st.set_option("server.fileUploader.allowMediaFiles", True)
-uploaded_file = st.file_uploader("Chọn file")
-target_lang = st.selectbox("Chọn ngôn ngữ bạn muốn dịch sang", ["english", "vietnamese"])
-
-final_pptx_id = None  # Biến lưu ID file sau khi xử lý
-
-if uploaded_file is not None:
-    if st.button("Upload"):
-        file_type = detect_file_type(uploaded_file)
-        st.write(f"Detected file type: {file_type}")
-        if file_type == "PPTX":
-            
-            file_id = save_file_to_mongodb(uploaded_file=uploaded_file, db_name="ppt", collection_name="root_file")
-            st.write(f"File ID: {file_id}")
-
-            xml_file_id = ppt_to_xml_mongodb(file_id)
-            text_dict = extract_text_from_xml(file_id=xml_file_id)
-            translated_dict = translate_text_dict(text_dict, target_lang=target_lang, gemini_api=api_key)
-
-            final_xml_id = update_xml_with_translated_text_mongodb(xml_file_id, translated_dict)
-            st.write(f"Final XML ID: {final_xml_id}")
-
-            # Lưu ID file PPTX cuối cùng
-            final_pptx_id = create_translated_ppt(
-                db_name="ppt", original_ppt_id=file_id, 
-                translated_xml_id=final_xml_id, output_collection="final_pptx"
-            )
-            st.write(f"Final PPTX ID: {final_pptx_id}")
-
-            # Hiển thị ảnh slide trước khi tải xuống
-            if final_pptx_id:
-                st.write("✅ File đã sẵn sàng để tải xuống!")
-
-                pptx_io, pptx_filename = fetch_file_from_mongodb("ppt", "final_pptx", final_pptx_id)
-
-                if pptx_io:
-                    # Nút tải file sau khi xem trước
-                    st.download_button(
-                        label="Click to Download",
-                        data=pptx_io.getvalue(),  # Chuyển thành bytes để tải về
-                        file_name=pptx_filename,
-                        mime="application/vnd.openxmlformats-officedocument.presentationml.presentation"
-                    )
-                else:
-                    st.error("❌ Không thể tải xuống file. Kiểm tra lại ID hoặc thử lại sau.")
-
-
-
-        elif file_type == "Excel":
-            file_id = save_file_to_mongodb(uploaded_file=uploaded_file, db_name="excel", collection_name="root_file")
-            st.write(f"File ID: {file_id}")
-             
-            final_id = translate_xlsx(file_id=file_id, from_lang="en", target_lang="vi", gemini_api=api_key)
-            st.write(f"Final Excel ID: {final_id}")
-            if final_id:
-                st.write("✅ File đã sẵn sàng để tải xuống!")
-
-                excel_io, excel_filename = fetch_file_from_mongodb("excel", "final_file", final_id)
-
-                if excel_io:
-                    st.download_button(
-                        label="Click to Download",
-                        data=excel_io.getvalue(),  
-                        file_name=excel_filename,
-                        mime="application/vnd.ms-excel"
-                    )
-                else:
-                    st.error("❌ Không thể tải xuống file. Kiểm tra lại ID hoặc thử lại sau.")
-        elif file_type == "CSV":
-            file_id = save_file_to_mongodb(uploaded_file=uploaded_file, db_name="csv", collection_name="root_file")
-            st.write(f"File ID: {file_id}")
-
-            final_id = translate_csv(file_id=file_id, source_lang="en", target_lang="vi", gemini_api=api_key)
-            st.write(f"Final CSV ID: {final_id}")
-            if final_id:
-                st.write("✅ File đã sẵn sàng để tải xuống!")
-
-                csv_io, csv_filename = fetch_file_from_mongodb("csv", "final_file", final_id)
-
-                if csv_io:
-                    st.download_button(
-                        label="Click to Download",
-                        data=csv_io.getvalue(),  
-                        file_name=csv_filename,
-                        mime="text/csv"
-                    )
-                else:
-                    st.error("❌ Không thể tải xuống file. Kiểm tra lại ID hoặc thử lại sau.")
-
-        elif file_type == "Word":
-            file_id = save_file_to_mongodb(uploaded_file=uploaded_file, db_name="word", collection_name="root_file")
-            st.write(f"File ID: {file_id}")
-
-            final_id = translate_docx_from_mongodb(file_id = file_id, target_lang="Vietnamese")
-            st.write(f"Final CSV ID: {final_id}")
-            if final_id:
-                st.write("✅ File đã sẵn sàng để tải xuống!")
-
-                docx_io, docx_filename = fetch_file_from_mongodb("word", "final_file", final_id)
-
-                if docx_io:
-                    st.download_button(
-                        label="Click to Download",
-                        data=docx_io.getvalue(),  
-                        file_name=docx_filename,
-                        mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-                    )
-                else:
-                    st.error("❌ Không thể tải xuống file. Kiểm tra lại ID hoặc thử lại sau.")
 
-            
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+
+st.title("Translate Your File Easily! 🌍")
+
+uploaded_file = st.file_uploader("📂 Chọn file để dịch")
+target_lang = st.selectbox("🌐 Chọn ngôn ngữ", ["english", "vietnamese"])
+
+def process_file(file, file_type):
+    progress_bar = st.progress(0)
+    file_id = save_file_to_mongodb(uploaded_file=file, db_name=file_type.lower(), collection_name="root_file")
+    progress_bar.progress(20)
+    st.write(f"📂 File ID: {file_id}")
+
+    if file_type == "PPTX":
+        xml_file_id = ppt_to_xml_mongodb(file_id)
+        progress_bar.progress(40)
+        text_dict = extract_text_from_xml(file_id=xml_file_id)
+        translated_dict = translate_text_dict(text_dict, target_lang=target_lang)
+        progress_bar.progress(60)
+        final_xml_id = update_xml_with_translated_text_mongodb(xml_file_id, translated_dict)
+        final_id = create_translated_ppt("ppt", file_id, final_xml_id, "final_pptx")
+    elif file_type == "Excel":
+        final_id = translate_xlsx(file_id, "en", target_lang, os.getenv("GEMINI_API_KEY"))
+    elif file_type == "CSV":
+        final_id = translate_csv(file_id, "en", target_lang, os.getenv("GEMINI_API_KEY"))
+    elif file_type == "Word":
+        final_id = translate_docx_from_mongodb(file_id, target_lang)
+    else:
+        st.error("❌ Loại file không hỗ trợ!")
+        return
+
+    progress_bar.progress(80)
+    st.write("✅ File đã được dịch xong!")
+    file_io, file_name = fetch_file_from_mongodb(file_type.lower(), "final_file", final_id)
+    progress_bar.progress(100)
+
+    if file_io:
+        st.download_button("⬇️ Tải file về", data=file_io.getvalue(), file_name=file_name)
+    else:
+        st.error("❌ Không thể tải xuống file. Vui lòng thử lại!")
+
+if uploaded_file and st.button("🚀 Upload và dịch ngay!"):
+    file_type = detect_file_type(uploaded_file)
+    st.write(f"🔍 Loại file phát hiện: {file_type}")
+    process_file(uploaded_file, file_type)