Update app.py

app.py

@@ -39,7 +39,8 @@ except Exception as e:
 
 # ── v2 agent ──────────────────────────────────────────────────────────────────
 try:
-    from agent_v2 import agent_v2, clean_thread_history_v2
+    
+    from agent_v2 import agent_v2, clean_thread_history_v2, reset_thread_v2
     AGENT_V2_OK = True
     print("Step 3b: v2 agent imported OK")
 except Exception as e:
@@ -47,6 +48,7 @@ except Exception as e:
     agent_v2 = None
     AGENT_V2_OK = False
     def clean_thread_history_v2(tid): pass
+    def reset_thread_v2(tid): pass
 
 # ── constants ──────────────────────────────────────────────────────────────────
 DATA_DIR = Path("data")
@@ -242,9 +244,9 @@ def _build_v2_cluster_table() -> pd.DataFrame:
         str(s.get("label", "Cluster {}".format(s.get("cluster_id", "")))),
         int(s.get("paper_count", 0)),
         str(s.get("vote_agreement", "")),
-        str(s.get("llm_vote_1_IS_THEORY", "")),
-        str(s.get("llm_vote_2_DIGITAL_MGT", "")),
-        str(s.get("llm_vote_3_COMP_SCI", "")),
+        str(s.get("llm_vote_1_MISTRAL", "")),
+        str(s.get("llm_vote_2_OPENAI",  "")),
+        str(s.get("llm_vote_3_GROQ",    "")),
         str(" | ".join(s.get("top3_titles", [])[:2])),
     ], summaries))
     cols = ["#", "Cluster Label", "Papers", "Vote Agreement",
@@ -305,6 +307,10 @@ def handle_file_upload(file_path) -> str:
 
 def reset_all_data() -> tuple:
     import shutil as _shutil
+    try:
+        reset_thread_v2("default")
+    except Exception:
+        pass
     if DATA_DIR.exists():
         _shutil.rmtree(str(DATA_DIR))
     DATA_DIR.mkdir(exist_ok=True)
@@ -866,6 +872,8 @@ with gr.Blocks(
                     "- `comparison_v2.csv` — one row per paper with cluster + PAJAIS\n"
                     "- `cluster_audit.csv` — full LLM voting record, per paper\n"
                     "- `narrative_v2.txt` — 500-word Section 7 discussion\n"
+                    "> 💡 **Cache:** `data/v2/llm_cache/` stores LLM responses — "
+                    "delete this folder to force fresh labels on re-run.\n"
                 )
                 download_files = gr.File(
                     label="Output Files", file_count="multiple", interactive=False,
@@ -949,7 +957,7 @@ with gr.Blocks(
         outputs=[table_status],
     )
     check_v2_btn.click(
-        fn=check_status,
+        fn=lambda: check_status("v2"),
         inputs=[current_mode_state],
         outputs=[table_status],
     )