Update app.py

app.py

@@ -249,6 +249,40 @@ def handle_file_upload(file_path) -> str:
     return msg + "Type 'run abstract' in the chat to begin."
 
 
+def reset_all_data() -> tuple:
+    """
+    Delete everything in data/ so the user can start a completely fresh run.
+    Returns updated UI state for all outputs.
+    """
+    import shutil as _shutil
+    # Remove the entire data directory and recreate empty
+    if DATA_DIR.exists():
+        _shutil.rmtree(str(DATA_DIR))
+    DATA_DIR.mkdir(exist_ok=True)
+    empty_df    = pd.DataFrame(columns=REVIEW_COLUMNS)
+    empty_chart = "<p style='color:#9ca3af;padding:20px;'>Charts appear after Phase 2 completes.</p>"
+    status_msg  = (
+        "<div style='padding:10px;background:#fef3c7;border-radius:6px;"
+        "font-family:sans-serif;font-size:13px;'>"
+        "🔄 <b>All data cleared.</b> Upload a new CSV and type <b>run abstract</b> to begin."
+        "</div>"
+    )
+    return (
+        [],                          # chatbot history cleared
+        "",                          # chat input cleared
+        make_progress_html(0),       # progress reset
+        _run_status_html(),          # status badges reset
+        empty_df,                    # review table cleared
+        empty_chart,                 # chart cleared
+        None,                        # downloads cleared
+        new_thread_id(),             # fresh abstract thread
+        new_thread_id(),             # fresh title thread
+        "abstract",                  # reset current run
+        status_msg,                  # table status message
+        "",                          # file status cleared
+    )
+
+
 def _detect_phase(text: str) -> int:
     phase_map = {
         "phase 5.5": 5, "phase 6": 6, "phase 5": 4,
@@ -521,7 +555,45 @@ with gr.Blocks(title="BERTopic Thematic Analysis Agent") as demo:
 
     # ── Section 1 ─────────────────────────────────────────────────────────────
     with gr.Accordion("📂 Section 1 — Data Input", open=True):
-        file_input  = gr.File(label="Upload Scopus CSV", file_types=[".csv"], type="filepath")
+
+        # Show existing data state on load
+        def _startup_msg():
+            abs_done   = (DATA_DIR / "abstract" / "taxonomy.json").exists()
+            title_done = (DATA_DIR / "title"    / "taxonomy.json").exists()
+            csv_exists = (DATA_DIR / "uploaded.csv").exists()
+            has_data   = csv_exists or abs_done or title_done
+            return (
+                "<div style='padding:10px;background:#fef3c7;border:1px solid #fcd34d;"
+                "border-radius:8px;font-family:sans-serif;font-size:13px;'>"
+                "⚠️ <b>Previous session data detected.</b> "
+                "Abstract run: {abs} &nbsp;|&nbsp; Title run: {title} &nbsp;|&nbsp; CSV: {csv}<br>"
+                "Click <b>🗑️ Reset & Start Fresh</b> to clear everything and begin a new analysis. "
+                "Or continue where you left off by typing your next command."
+                "</div>"
+                if has_data else
+                "<div style='padding:10px;background:#f0fdf4;border:1px solid #86efac;"
+                "border-radius:8px;font-family:sans-serif;font-size:13px;'>"
+                "✅ Fresh session — no previous data. Upload your CSV to begin."
+                "</div>"
+            ).format(
+                abs="✅" if abs_done else "⏳",
+                title="✅" if title_done else "⏳",
+                csv="✅" if csv_exists else "❌",
+            )
+
+        startup_banner = gr.HTML(_startup_msg())
+
+        with gr.Row():
+            file_input  = gr.File(
+                label="Upload Scopus CSV", file_types=[".csv"],
+                type="filepath", scale=4,
+            )
+            reset_btn = gr.Button(
+                "🗑️ Reset & Start Fresh",
+                variant="stop", scale=1,
+                size="sm",
+            )
+
         file_status = gr.Textbox(
             label="Upload status", interactive=False, lines=2,
         )
@@ -631,6 +703,14 @@ with gr.Blocks(title="BERTopic Thematic Analysis Agent") as demo:
         abstract_thread_state, title_thread_state, current_run_state,
     ]
 
+    # Reset wires to ALL outputs including table_status, file_status, startup_banner
+    reset_outputs = [
+        chatbot, chat_input, progress_bar, run_status,
+        review_table, chart_display, download_files,
+        abstract_thread_state, title_thread_state, current_run_state,
+        table_status, file_status,
+    ]
+
     send_btn.click(
         fn=run_agent,
         inputs=[chat_input, chatbot,
@@ -649,6 +729,11 @@ with gr.Blocks(title="BERTopic Thematic Analysis Agent") as demo:
                 abstract_thread_state, title_thread_state, current_run_state],
         outputs=send_outputs,
     )
+    reset_btn.click(
+        fn=reset_all_data,
+        inputs=[],
+        outputs=reset_outputs,
+    )
 
 print("Step 5: UI built OK, launching...")