update style

app.py

@@ -5,6 +5,7 @@ from dotenv import load_dotenv
 import tempfile
 import pandas as pd
 from typing import List, Dict, Any, Tuple
+import shutil
 import json
 from core.ingest import DocumentChunker, HierarchyManager
 from core.index import VectorStore
@@ -126,6 +127,21 @@ def initialize_system():
     
     return f"System initialized successfully! Using persist directory: {persist_dir}"
 
+def reset_index() -> str:
+    """Clear Chroma persistence and reinitialize the vector store."""
+    global rag_manager, evaluator, persist_directory
+    try:
+        dir_path = persist_directory or ("/data/chroma" if os.path.exists("/data/chroma") else "./chroma_data")
+        if os.path.exists(dir_path):
+            shutil.rmtree(dir_path, ignore_errors=True)
+        os.makedirs(dir_path, exist_ok=True, mode=0o755)
+        persist_directory = dir_path
+        rag_manager = RAGManager(persist_directory=dir_path)
+        evaluator = RAGEvaluator(rag_manager)
+        return f"Index reset complete. Using fresh directory: {dir_path}"
+    except Exception as ex:
+        return f"Failed to reset index: {ex}"
+
 def upload_documents(files: List[str], hierarchy: str, doc_type: str, language: str, progress: Any = None) -> Tuple[str, List[Dict[str, Any]], Dict[str, Any], List[Dict[str, Any]]]:
     """Upload and process documents.
     Returns: (status_text, per_file_summaries, collection_stats)
@@ -508,16 +524,7 @@ def run_evaluation(queries_json: str, output_filename: str) -> Tuple[str, pd.Dat
 initialize_system()
 
 # Create Gradio interface
-APP_CSS = """
-.gradio-container { max-width: 1100px !important; margin: 0 auto !important; }
-.gradio-container img { max-width: 100%; height: auto; }
-/* Prevent toolbar and button icons from stretching */
-button img, .icon, .toolbar img { width: 20px !important; height: 20px !important; object-fit: contain; }
-/* Keep DataFrame within container width */
-.gr-dataframe { max-width: 100%; }
-"""
-
-with gr.Blocks(title="RAG Evaluation System", css=APP_CSS, fill_height=False) as demo:
+with gr.Blocks(title="RAG Evaluation System") as demo:
     gr.Markdown("# RAG Evaluation System: Hierarchical vs Standard RAG")
     
     with gr.Tab("Upload Documents"):
@@ -550,6 +557,7 @@ with gr.Blocks(title="RAG Evaluation System", css=APP_CSS, fill_height=False) as
                 build_output = gr.Textbox(label="Build Status", lines=4)
                 stats_table = gr.DataFrame(label="File Summary")
                 chunks_table = gr.DataFrame(label="Indexed Chunks (preview)")
+                reset_btn = gr.Button("Reset Index (Clear chroma_data)", variant="secondary")
     
     with gr.Tab("Search"):
         gr.Markdown("## Compare Retrieval Pipelines")
@@ -662,6 +670,12 @@ with gr.Blocks(title="RAG Evaluation System", css=APP_CSS, fill_height=False) as
         outputs=[build_output, stats_table, chunks_table],
         api_name="build_rag"
     )
+
+    reset_btn.click(
+        fn=reset_index,
+        inputs=None,
+        outputs=build_output
+    )
     
     # One button: uses manual if checked; otherwise auto
     search_btn.click(