Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import pandas as pd
 import os
-from translator_utils import process_rows, download_dataset, LOCAL_FILENAME
+from translator_utils import process_rows, revert_rows, download_dataset, LOCAL_FILENAME
 
 def load_data():
     """Load the dataset safely."""
@@ -12,7 +12,6 @@ def load_data():
             return pd.DataFrame()
     try:
         df = pd.read_csv(LOCAL_FILENAME)
-        # Ensure translation columns exist
         if 'Judul_Inggris' not in df.columns:
             df['Judul_Inggris'] = ""
         if 'Isi_Inggris' not in df.columns:
@@ -26,10 +25,12 @@ def get_pending_data(full_df):
     """Filter for rows that are NOT fully translated."""
     if full_df.empty:
         return full_df
-    # A row is pending if either Judul_Inggris or Isi_Inggris is empty
     mask = (full_df['Judul_Inggris'].str.strip() == "") | (full_df['Isi_Inggris'].str.strip() == "")
-    # Add a 'Select' column for checkbox, initialized to False
+    
     df_view = full_df[mask].copy()
+    df_view.reset_index(inplace=True) 
+    df_view.rename(columns={'index': 'ID'}, inplace=True)
+    
     df_view.insert(0, "Select", False)
     return df_view
 
@@ -40,39 +41,52 @@ def get_history_data(full_df, date_filter=""):
     mask = (full_df['Judul_Inggris'].str.strip() != "") & (full_df['Isi_Inggris'].str.strip() != "")
     df_view = full_df[mask].copy()
     
+    df_view.reset_index(inplace=True)
+    df_view.rename(columns={'index': 'ID'}, inplace=True)
+    
     if date_filter:
         df_view = df_view[df_view['Tanggal'].str.contains(date_filter, case=False, na=False)]
-        
+    
+    # Add 'Select' column for Revert functionality
+    df_view.insert(0, "Select", False)
     return df_view
 
 def on_translate_selected(df_data, token, progress=gr.Progress()):
-    """
-    Callback when user clicks 'Translate Selected'.
-    df_data is the dataframe from the component (with checkboxes).
-    """
-    # Filter rows where 'Select' is True
-    # Gradio Dataframe sends back a pandas DataFrame
+    """Callback when user clicks 'Translate Selected'."""
     selected_rows = df_data[df_data["Select"] == True]
     
     if selected_rows.empty:
         return gr.DataFrame(), gr.DataFrame(), "No rows selected!"
     
-    # We need the ORIGINAL indices. 
-    # Since we filtered the view, the index of the dataframe should still be the original index 
-    # IF we didn't reset_index. 
-    # Let's check. Gradio might reset it or treat it as data.
-    # To be safe, we rely on the fact that we passed the dataframe with original index?
-    # Actually Gradio usually converts index to 0..n.
-    # We should have preserved the original index as a column or rely on matching content?
-    # Better approach: We passed the DF. Gradio returns the modified DF.
-    # If the index is preserved, we are good.
-    # Let's try to trust the index.
-    
-    indices = selected_rows.index.tolist()
+    try:
+        indices = selected_rows['ID'].tolist()
+        indices = [int(i) for i in indices]
+    except KeyError:
+        return gr.DataFrame(), gr.DataFrame(), "Error: ID column missing."
     
-    # Call backend
     updated_df, msg = process_rows(indices, token=token, progress=progress)
     
+    pending = get_pending_data(updated_df)
+    history = get_history_data(updated_df)
+    
+    return pending, history, msg
+
+def on_revert_selected(df_data, token, progress=gr.Progress()):
+    """Callback when user clicks 'Revert Selected'."""
+    selected_rows = df_data[df_data["Select"] == True]
+    
+    if selected_rows.empty:
+        return gr.DataFrame(), gr.DataFrame(), "No rows selected!"
+    
+    try:
+        indices = selected_rows['ID'].tolist()
+        indices = [int(i) for i in indices]
+    except KeyError:
+        return gr.DataFrame(), gr.DataFrame(), "Error: ID column missing."
+    
+    # Call revert
+    updated_df, msg = revert_rows(indices, token=token, progress=progress)
+    
     # Refresh views
     pending = get_pending_data(updated_df)
     history = get_history_data(updated_df)
@@ -87,34 +101,35 @@ with gr.Blocks(title="Auto AI News Translator") as app:
     gr.Markdown("# Auto AI News Translator")
     gr.Markdown("Translate news from Indonesian to English (with Chunking). Select rows to translate.")
     
-    # Store full data state if needed, but we read from disk to be single-source-of-truth safe
-    # token input
     hf_token = gr.Textbox(label="HF Token (Optional for Persistence)", placeholder="hf_...", type="password")
     status_output = gr.Textbox(label="Status", value="Ready", interactive=False)
     
     with gr.Tabs():
         with gr.TabItem("Translate Pending"):
             gr.Markdown("Select rows to translate and click Submit.")
-            # We use wrap=True for long text
-            # We use interactive=True to allow checking the 'Select' box
             pending_df_comp = gr.DataFrame(
                 value=get_pending_data(load_data()),
                 label="Pending Translations",
                 interactive=True,
                 wrap=True,
-                datatype=["bool"] + ["str"] * 6 # approximate types
+                datatype=["bool", "number"] + ["str"] * 6 
             )
             translate_btn = gr.Button("Translate Selected Rows", variant="primary")
             
-        with gr.TabItem("History"):
+        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",
-                interactive=False,
-                wrap=True
+                interactive=True,
+                wrap=True,
+                datatype=["bool", "number"] + ["str"] * 6
             )
-            date_filter.change(fn=filter_history, inputs=[date_filter], outputs=[history_df_comp])
+            with gr.Row():
+                date_filter.change(fn=filter_history, inputs=[date_filter], outputs=[history_df_comp])
+                revert_btn = gr.Button("Revert Selected (Delete & Move to Pending)", variant="stop")
 
     translate_btn.click(
         fn=on_translate_selected,
@@ -122,7 +137,11 @@ with gr.Blocks(title="Auto AI News Translator") as app:
         outputs=[pending_df_comp, history_df_comp, status_output]
     )
     
-    # Auto refresh on load? (Handled by initial value)
+    revert_btn.click(
+        fn=on_revert_selected,
+        inputs=[history_df_comp, hf_token],
+        outputs=[pending_df_comp, history_df_comp, status_output]
+    )
 
 if __name__ == "__main__":
     app.launch()