Update app.py

app.py

@@ -1,17 +1,21 @@
 import gradio as gr
 import pandas as pd
 import os
-from translator_utils import process_rows, revert_rows, download_dataset, LOCAL_FILENAME, ENV_HF_TOKEN
+from translator_utils import process_rows, revert_rows, download_dataset, LOCAL_FILENAME, ENV_HF_TOKEN, read_csv_robust
 
 def load_data():
-    """Load the dataset safely."""
+    """Load the dataset safely using robust reader."""
     if not os.path.exists(LOCAL_FILENAME):
         try:
             download_dataset()
         except:
             return pd.DataFrame()
+            
+    df = read_csv_robust(LOCAL_FILENAME)
+    if df.empty:
+        return df
+        
     try:
-        df = pd.read_csv(LOCAL_FILENAME)
         if 'Judul_Inggris' not in df.columns:
             df['Judul_Inggris'] = ""
         if 'Isi_Inggris' not in df.columns:
@@ -21,6 +25,14 @@ def load_data():
     except:
         return pd.DataFrame()
 
+def force_refresh_data():
+    """Forces a re-download and reload of the data."""
+    try:
+        download_dataset(force=True)
+        return load_data(), "Data refreshed successfully."
+    except Exception as e:
+        return pd.DataFrame(), f"Failed to refresh data: {e}"
+
 def get_pending_data(full_df):
     """Filter for rows that are NOT fully translated."""
     if full_df.empty:
@@ -97,6 +109,12 @@ def filter_history(date_input):
     df = load_data()
     return get_history_data(df, date_input)
 
+def refresh_views():
+    df, msg = force_refresh_data()
+    pending = get_pending_data(df)
+    history = get_history_data(df)
+    return pending, history, msg
+
 # CSS to ensure columns are readable
 custom_css = """
 /* Fix minimum width for content columns */
@@ -121,7 +139,11 @@ with gr.Blocks(title="Auto AI News Translator") as app:
         type="password",
         value=ENV_HF_TOKEN
     )
-    status_output = gr.Textbox(label="Status", value="Ready", interactive=False)
+    
+    with gr.Row():
+        status_output = gr.Textbox(label="Status", value="Ready", interactive=False, scale=4)
+        refresh_btn = gr.Button("🔄 Refresh Data (Force Redownload)", scale=1)
+        
     
     with gr.Tabs():
         with gr.TabItem("Translate Pending"):
@@ -138,6 +160,7 @@ with gr.Blocks(title="Auto AI News Translator") as app:
         with gr.TabItem("History (Revert Supported)"):
             gr.Markdown("Select rows to delete translation and move back to Pending.")
             date_filter = gr.Textbox(label="Filter by Date (YYYY-MM-DD)")
+            # History now has checkbox
             history_df_comp = gr.DataFrame(
                 value=get_history_data(load_data()),
                 label="Translation History",
@@ -160,6 +183,11 @@ with gr.Blocks(title="Auto AI News Translator") as app:
         inputs=[history_df_comp, hf_token],
         outputs=[pending_df_comp, history_df_comp, status_output]
     )
+    
+    refresh_btn.click(
+        fn=refresh_views,
+        outputs=[pending_df_comp, history_df_comp, status_output]
+    )
 
 if __name__ == "__main__":
     app.launch(css=custom_css)