Update app.py

app.py

@@ -7,7 +7,7 @@ from huggingface_hub import HfApi, create_repo, upload_file
 from datasets import Dataset
 
 # File paths
-INPUT_CSV = "summaries_boe.csv"  # Change this to your input CSV path
+INPUT_CSV = "summaries.csv"  # Change this to your input CSV path
 OUTPUT_CSV = "results.csv"   # Local backup file path
 TEMP_JSON = "temp_results.jsonl"  # Temporary file for storing results as JSONL
 
@@ -22,7 +22,7 @@ def load_data():
         return pd.read_csv(INPUT_CSV)
     else:
         # Create empty dataframe with required columns if file doesn't exist
-        return pd.DataFrame(columns=["id", "boe_text_cleaned", "tweet_text_cleaned", "tweet_original"])
+        return pd.DataFrame(columns=["text_id", "text", "summary_a", "summary_b"])
 
 def initialize_hf_dataset():
     """Initialize a HuggingFace dataset repository if it doesn't exist"""
@@ -222,11 +222,11 @@ with gr.Blocks(title="Summary Chooser") as interface:
         result_box = gr.Textbox(label="Result")
     
     # Initialize with the first item
-    id, boe_text_cleaned, tweet_text_cleaned, tweet_original, prog = app.get_current_item()
+    text_id, text, summary_a, summary_b, prog = app.get_current_item()
     text_id_box.value = id
-    text_box.value = boe_text_cleaned
-    summary_a.value = tweet_text_cleaned
-    summary_b.value = tweet_original
+    text_box.value = text
+    summary_a.value = summary_a
+    summary_b.value = summary_b
     progress_label.value = prog
     
     # Set up event handlers