Update app.py

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 # 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
+TEMP_FILE = "temp.txt"  # File tạm lưu dữ liệu gốc
 
 def write_file(file_path, data):
     with open(file_path, 'w', encoding='utf-8') as file:
@@ -43,7 +43,7 @@ def split_emails(input_data, uploaded_file):
         if match:
             gmail_addresses.append(match.group(1) + '\n')
     
-    # Lưu dữ liệu gốc vào biến tạm (không tự xóa)
+    # Ghi đè dữ liệu gốc vào file temp.txt
     write_file(TEMP_FILE, [line + '\n' for line in lines])
     
     output_data = "".join(gmail_addresses)
@@ -75,71 +75,27 @@ def fill_missing_parts(input_data, uploaded_file):
     output_total = count_emails(output_data)
     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_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 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, uploaded_file):
-    if not output_data:
-        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])
+    return "Đã xóa các file output.", gr.update(value="", label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None
 
-# Giao diện Gradio
+# Giao diện Gradio (chỉ giữ lại 2 tab chính)
 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")
-            with gr.Row():
-                upload_file = gr.File(label="📤 Upload File (.txt)", file_types=[".txt"])
-            with gr.Row():
-                clear_input_btn = gr.Button("🧹 Clear Input")
+                                  placeholder="Nhập dữ liệu hoặc tải file .txt lên...")
+            upload_file = gr.File(label="📤 Upload File (.txt)", file_types=[".txt"])
+            clear_input_btn = gr.Button("🧹 Clear Input")
         
         with gr.Column(scale=1):
             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")
+            download_file = gr.File(label="📥 Download Files", interactive=False, file_count="multiple")
+            clear_output_btn = gr.Button("🧹 Clear Output")
+            download_btn = gr.Button("💾 Download Output")
 
     status = gr.Textbox(label="🔄 Status", interactive=False)
     
@@ -149,15 +105,14 @@ with gr.Blocks(title="Email Processor Pro", theme=gr.themes.Soft()) as demo:
                 split_btn = gr.Button("✂️ Split Emails")
                 fill_btn = gr.Button("🔍 Fill Missing")
 
-    with gr.Row():
-        clear_files_btn = gr.Button("🧹 Clear All Data", size="lg")
+    clear_files_btn = gr.Button("🧹 Clear All Data", size="lg")
 
     # 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])
+    clear_files_btn.click(clear_files, outputs=[status, input_text, output_text, download_file])
+    clear_input_btn.click(lambda: ["", gr.update(value="", label="Input Text (Total: 0)"), gr.update(value="", label="Output Text (Total: 0)"), None, "Đã xóa input."], outputs=[input_text, input_text, output_text, upload_file, status])
+    clear_output_btn.click(clear_files, 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()