Update app.py

app.py

@@ -2,27 +2,24 @@ import os
 import re
 import gradio as gr
 import pandas as pd
-from mega import Mega
-from io import StringIO
-import logging
-
-# Cấu hình logging để debug
-logging.basicConfig(level=logging.DEBUG)
-logger = logging.getLogger(__name__)
-
-# Lấy thông tin từ biến môi trường
-MEGA_EMAIL = os.getenv("MEGA_EMAIL", "your_mega_email@example.com")
-MEGA_PASSWORD = os.getenv("MEGA_PASSWORD", "your_mega_password")
-MEGA_FOLDER = os.getenv("MEGA_FOLDER", "/EmailProcessor")
-
-# Khởi tạo Mega client
-try:
-    logger.debug(f"Đang đăng nhập với email: {MEGA_EMAIL}")
-    mega = Mega().login(MEGA_EMAIL, MEGA_PASSWORD)
-    logger.debug("Đăng nhập thành công!")
-except Exception as e:
-    logger.error(f"Lỗi đăng nhập Mega: {str(e)}")
-    raise
+
+# Các biến toàn cục
+OUTPUT_FILE_TXT = "output.txt"
+OUTPUT_FILE_XLSX = "output.xlsx"
+TEMP_FILE = "temp.txt"  # Biến tạm lưu dữ liệu sau split
+
+def write_file(file_path, data):
+    with open(file_path, 'w', encoding='utf-8') as file:
+        file.writelines(data)
+
+def read_file(file):
+    if file is None:
+        return ""
+    try:
+        with open(file, 'r', encoding='utf-8') as f:
+            return f.read()
+    except Exception as e:
+        return f"Lỗi khi đọc file: {str(e)}"
 
 def count_emails(data):
     if not data or isinstance(data, str) and data.startswith("Lỗi"):
@@ -31,10 +28,9 @@ def count_emails(data):
     return len(lines)
 
 def split_emails(input_data, uploaded_file):
-    global TEMP_FILE_CONTENT
     data = input_data if input_data else read_file(uploaded_file)
     if not data:
-        return "Cảnh báo: Không có dữ liệu.", gr.update(label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None, None
+        return "Cảnh báo: Không có dữ liệu.", gr.update(label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None
     
     lines = data.splitlines()
     gmail_addresses = []
@@ -42,45 +38,30 @@ def split_emails(input_data, uploaded_file):
     for line in lines:
         if not line.strip():
             continue
+            
         match = re.search(r'([a-zA-Z0-9._%+-]+@gmail\.com)', line)
         if match:
             gmail_addresses.append(match.group(1) + '\n')
     
-    # Lưu dữ liệu tạm
-    TEMP_FILE_CONTENT = "\n".join(lines) + "\n"
+    # Lưu dữ liệu gốc vào biến tạm (không tự xóa)
+    write_file(TEMP_FILE, [line + '\n' for line in lines])
+    
     output_data = "".join(gmail_addresses)
     input_total = count_emails(data)
     output_total = count_emails(output_data)
-    
-    # Upload lên Mega
-    file_name = "private_output.txt"
-    try:
-        with open(file_name, "w", encoding="utf-8") as f:
-            f.writelines(gmail_addresses)
-        m = mega.get_root()
-        m = mega.find(m, MEGA_FOLDER)
-        if not m:
-            logger.debug(f"Tạo thư mục {MEGA_FOLDER} trên Mega")
-            m = mega.create_folder(MEGA_FOLDER)
-        mega.upload(file_name, m)
-        os.remove(file_name)
-        logger.debug(f"Đã upload {file_name} lên {MEGA_FOLDER}")
-    except Exception as e:
-        logger.error(f"Lỗi upload lên Mega: {str(e)}")
-        return f"Lỗi khi upload lên Mega: {str(e)}. Vui lòng kiểm tra email/mật khẩu hoặc kết nối.", gr.update(value=data, label=f"Input Text (Total: {input_total})"), gr.update(value=output_data, label=f"Output Text (Total: {output_total})"), None, None
-    
-    return f"Đã lưu kết quả lên Mega tại {MEGA_FOLDER}/{file_name}.", gr.update(value=data, label=f"Input Text (Total: {input_total})"), gr.update(value=output_data, label=f"Output Text (Total: {output_total})"), download_output(data, output_data), None
+    return f"Đã lưu kết quả.", gr.update(value=data, label=f"Input Text (Total: {input_total})"), gr.update(value=output_data, label=f"Output Text (Total: {output_total})"), None
 
 def fill_missing_parts(input_data, uploaded_file):
-    global TEMP_FILE_CONTENT
     data = input_data if input_data else read_file(uploaded_file)
     if not data:
-        return "Cảnh báo: Không có dữ liệu.", gr.update(label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None, None
+        return "Cảnh báo: Không có dữ liệu.", gr.update(label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None
+    
+    if not os.path.exists(TEMP_FILE):
+        return "Lỗi: Không tìm thấy file tạm. Vui lòng chạy Split Emails trước.", gr.update(label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None
     
-    if not TEMP_FILE_CONTENT:
-        return "Lỗi: Không tìm thấy dữ liệu tạm. Vui lòng chạy Split Emails trước.", gr.update(label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None, None
+    with open(TEMP_FILE, 'r', encoding='utf-8') as file:
+        temp_lines = {line.strip() for line in file}
     
-    temp_lines = {line.strip() for line in TEMP_FILE_CONTENT.splitlines() if line.strip()}
     input_emails = [line.strip() for line in data.splitlines() if line.strip()]
     output_lines = []
     
@@ -92,139 +73,91 @@ def fill_missing_parts(input_data, uploaded_file):
     output_data = "".join(output_lines)
     input_total = count_emails(data)
     output_total = count_emails(output_data)
-    return f"Đã xử lý.", gr.update(value=data, label=f"Input Text (Total: {input_total})"), gr.update(value=output_data, label=f"Output Text (Total: {output_total})"), download_output(data, output_data), None
+    return f"Đã lưu kết quả.", gr.update(value=data, label=f"Input Text (Total: {input_total})"), gr.update(value=output_data, label=f"Output Text (Total: {output_total})"), None
 
-def clear_files():
-    # Không xóa dữ liệu tạm theo yêu cầu
-    return "Nút xóa không có tác dụng. Dữ liệu tạm được giữ nguyên.", gr.update(value="", label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None, None
+def clear_input(input_data, output_data, uploaded_file):
+    return "", gr.update(value="", label="Input Text (Total: 0)"), gr.update(value=output_data, label=f"Output Text (Total: {count_emails(output_data)})"), None, "Đã xóa input."
 
-def read_file(file):
-    if file is None:
-        return ""
-    try:
-        with open(file, 'r', encoding='utf-8') as f:
-            return f.read()
-    except Exception as e:
-        return f"Lỗi khi đọc file: {str(e)}"
+def clear_output(input_data, output_data, uploaded_file):
+    if os.path.exists(OUTPUT_FILE_TXT):
+        os.remove(OUTPUT_FILE_TXT)
+    if os.path.exists(OUTPUT_FILE_XLSX):
+        os.remove(OUTPUT_FILE_XLSX)
+    return "Đã xóa output.", gr.update(value=input_data, label=f"Input Text (Total: {count_emails(input_data)})"), gr.update(value="", label="Output Text (Total: 0)"), None
+
+def clear_files():
+    # Chỉ xóa output files, không xóa TEMP_FILE
+    for file_path in [OUTPUT_FILE_TXT, OUTPUT_FILE_XLSX]:
+        if os.path.exists(file_path):
+            os.remove(file_path)
+    return "Đã xóa các file output.", gr.update(value="", label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None, None
 
-def download_output(input_data, output_data):
+def download_output(input_data, output_data, uploaded_file):
     if not output_data:
-        return None
-    return gr.File(value=StringIO(output_data), label="Download Output", file_type="text/plain", file_name="private_output.txt")
-
-# Biến toàn cục để lưu dữ liệu tạm
-TEMP_FILE_CONTENT = ""
-
-custom_css = """
-:root {
-    --primary-color: #6e48aa;
-    --dark-color: #121212;
-    --darker-color: #0a0a0a;
-    --light-color: #e0e0e0;
-    --text-color: #f5f5f5;
-    --border-color: #333;
-    --card-bg: #1e1e1e;
-    --input-bg: #252525;
-}
-
-body {
-    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-    background-color: var(--darker-color) !important;
-    color: var(--text-color) !important;
-    margin: 0;
-    padding: 20px;
-}
-
-.gr-block {
-    border-radius: 8px !important;
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.3) !important;
-    background-color: var(--card-bg) !important;
-    border: 1px solid var(--border-color) !important;
-    margin-bottom: 15px !important;
-}
-
-.gr-box {
-    border: 1px solid var(--border-color) !important;
-    border-radius: 6px !important;
-    padding: 12px !important;
-    background-color: var(--input-bg) !important;
-}
-
-.textbox-fixed-height textarea {
-    height: 250px !important;
-    overflow-y: auto !important;
-    resize: none !important;
-    font-family: monospace !important;
-    background-color: var(--input-bg) !important;
-    color: var(--light-color) !important;
-    border: 1px solid var(--border-color) !important;
-}
-
-button {
-    border-radius: 6px !important;
-    font-weight: 500 !important;
-    transition: all 0.3s ease !important;
-    padding: 8px 16px !important;
-    margin: 4px !important;
-    border: none !important;
-}
-
-button:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 6px 12px rgba(0,0,0,0.2) !important;
-    opacity: 0.9 !important;
-}
-
-.primary-btn {
-    background: linear-gradient(135deg, var(--primary-color), #9d50bb) !important;
-    color: white !important;
-}
-
-.danger-btn {
-    background: linear-gradient(135deg, #f85032, #e73827) !important;
-    color: white !important;
-}
-
-.status-box {
-    background-color: var(--card-bg) !important;
-    border-left: 4px solid var(--primary-color) !important;
-    padding: 14px !important;
-    margin-top: 15px !important;
-    color: var(--light-color) !important;
-}
-"""
-
-with gr.Blocks(title="Email Processor", theme=gr.themes.Soft(), css=custom_css) as demo:
+        if os.path.exists(OUTPUT_FILE_TXT):
+            os.remove(OUTPUT_FILE_TXT)
+        if os.path.exists(OUTPUT_FILE_XLSX):
+            os.remove(OUTPUT_FILE_XLSX)
+        return "Không có dữ liệu để tải.", gr.update(label=f"Input Text (Total: {count_emails(input_data)})"), gr.update(label=f"Output Text (Total: {count_emails(output_data)})"), None
+
+    write_file(OUTPUT_FILE_TXT, output_data.splitlines(keepends=True))
+    
+    lines = output_data.splitlines()
+    data_rows = []
+    for line in lines:
+        if line.strip():
+            separator = re.search(r'[\t|;:]', line)
+            if separator:
+                sep_char = separator.group()
+                parts = line.strip().split(sep_char)
+                data_rows.append([part.strip() for part in parts])
+            else:
+                data_rows.append([line.strip()])
+    
+    df = pd.DataFrame(data_rows)
+    df.to_excel(OUTPUT_FILE_XLSX, index=False, header=False)
+    
+    return ("Đã tạo file output.txt và output.xlsx.", 
+            gr.update(label=f"Input Text (Total: {count_emails(input_data)})"), 
+            gr.update(label=f"Output Text (Total: {count_emails(output_data)})"), 
+            [OUTPUT_FILE_TXT, OUTPUT_FILE_XLSX])
+
+# Giao diện Gradio
+with gr.Blocks(title="Email Processor Pro", theme=gr.themes.Soft()) as demo:
     with gr.Row(equal_height=True):
         with gr.Column(scale=1):
             input_text = gr.Textbox(label="Input Text", lines=10, 
-                                  placeholder="Nhập dữ liệu hoặc tải file .txt lên...\nĐịnh dạng hỗ trợ: email|data hoặc email;data hoặc email:data hoặc email<tab>data",
-                                  elem_classes="textbox-fixed-height")
+                                  placeholder="Nhập dữ liệu hoặc tải file .txt lên...\nĐịnh dạng hỗ trợ: email|data hoặc email;data hoặc email:data hoặc email<tab>data")
+            with gr.Row():
+                upload_file = gr.File(label="📤 Upload File (.txt)", file_types=[".txt"])
             with gr.Row():
-                upload_file = gr.File(label="📤 Upload File (.txt)", file_types=[".txt"], elem_classes="file-upload")
+                clear_input_btn = gr.Button("🧹 Clear Input")
         
         with gr.Column(scale=1):
-            output_text = gr.Textbox(label="Output Text", lines=10, interactive=False, 
-                                   elem_classes="textbox-fixed-height")
-            download_file = gr.File(label="📥 Download Output", interactive=False)
+            output_text = gr.Textbox(label="Output Text", lines=10, interactive=False)
+            with gr.Row():
+                download_file = gr.File(label="📥 Download Files", interactive=False, file_count="multiple")
+            with gr.Row():
+                clear_output_btn = gr.Button("🧹 Clear Output")
+                download_btn = gr.Button("💾 Download Output")
 
-    status = gr.Textbox(label="🔄 Status", interactive=False, elem_classes="status-box")
+    status = gr.Textbox(label="🔄 Status", interactive=False)
     
     with gr.Tabs():
         with gr.Tab("🔧 Email Tools"):
             with gr.Row():
-                with gr.Column():
-                    split_btn = gr.Button("✂️ Split Emails", elem_classes="primary-btn")
-                
-                with gr.Column():
-                    fill_btn = gr.Button("🔍 Fill Missing", elem_classes="primary-btn")
-    
+                split_btn = gr.Button("✂️ Split Emails")
+                fill_btn = gr.Button("🔍 Fill Missing")
+
     with gr.Row():
-        clear_files_btn = gr.Button("🧹 Clear All Data", elem_classes="danger-btn", size="lg")
+        clear_files_btn = gr.Button("🧹 Clear All Data", size="lg")
 
-    # Kết nối các sự kiện
-    split_btn.click(split_emails, inputs=[input_text, upload_file], outputs=[status, input_text, output_text, download_file, upload_file])
-    fill_btn.click(fill_missing_parts, inputs=[input_text, upload_file], outputs=[status, input_text, output_text, download_file, upload_file])
+    # Kết nối sự kiện
+    split_btn.click(split_emails, inputs=[input_text, upload_file], outputs=[status, input_text, output_text, download_file])
+    fill_btn.click(fill_missing_parts, inputs=[input_text, upload_file], outputs=[status, input_text, output_text, download_file])
     clear_files_btn.click(clear_files, outputs=[status, input_text, output_text, download_file, upload_file])
+    clear_input_btn.click(clear_input, inputs=[input_text, output_text, upload_file], outputs=[input_text, input_text, output_text, upload_file, status])
+    clear_output_btn.click(clear_output, inputs=[input_text, output_text, upload_file], outputs=[status, input_text, output_text, download_file])
+    download_btn.click(download_output, inputs=[input_text, output_text, upload_file], outputs=[status, input_text, output_text, download_file])
 
-demo.launch()
+demo.launch()