Update app.py

app.py

@@ -409,11 +409,32 @@ kb = None
 
 def ensure_index():
     """Build index on first run or load from cache."""
-    if not kb.load():
-        if config.kb_directory.exists():
-            kb.build(config.kb_directory)
+    try:
+        # Try to load existing index first
+        if kb.load():
+            print(f"✅ Loaded existing index from {config.index_directory}")
+            return
+    except Exception as e:
+        print(f"⚠️ Could not load existing index: {e}")
+    
+    # Try to build new index if KB directory exists and has files
+    if config.kb_directory.exists():
+        md_files = list(config.kb_directory.glob("*.md"))
+        if md_files:
+            try:
+                print(f"🔨 Building index from {len(md_files)} markdown files...")
+                kb.build(config.kb_directory)
+                print(f"✅ Index built successfully!")
+            except Exception as e:
+                print(f"⚠️ Could not build index: {e}")
+                print(f"ℹ️ You can upload documents via the UI or add .md files to {config.kb_directory}")
         else:
-            print(f"Warning: KB directory {config.kb_directory} not found.")
+            print(f"ℹ️ No markdown files found in {config.kb_directory}")
+            print(f"ℹ️ Upload documents via the UI or add .md files to start using the knowledge base")
+    else:
+        print(f"ℹ️ KB directory {config.kb_directory} not found. Creating it...")
+        config.kb_directory.mkdir(exist_ok=True, parents=True)
+        print(f"ℹ️ Add .md files to {config.kb_directory} or upload documents via the UI")
 
 # ----------- Response Generation -----------
 def format_citations(citations: List[Dict]) -> str:
@@ -500,26 +521,18 @@ def handle_file_upload(file):
         return "ℹ️ No file uploaded.", ""
     
     try:
-        # Gradio File with type="filepath" -> file is a string
-        if isinstance(file, str):
-            file_path = file
-            filename = Path(file).name
-        else:
-            # Fallback for other types (e.g. temp file object)
-            file_path = getattr(file, "name", None)
-            if not file_path:
-                raise ValueError("Unsupported file object from Gradio.")
-            filename = Path(file_path).name
-
-        num_chunks, file_type = kb.build_from_uploaded_file(file_path, filename)
+        filename = Path(file.name).name
+        num_chunks, file_type = kb.build_from_uploaded_file(file.name, filename)
         
-        kb.uploaded_file_active = True
-        info = f"✅ Indexed {num_chunks} chunks from {file_type} file: **{filename}**"
-        return info, filename
-
+        return (
+            f"✅ **File processed successfully!**\n\n"
+            f"📄 **File:** {filename}\n"
+            f"📋 **Type:** {file_type}\n"
+            f"🔢 **Chunks:** {num_chunks}\n\n"
+            f"You can now ask questions about this document!"
+        ), filename
     except Exception as e:
-        return f"❌ Error processing file: {str(e)}", ""
-
+        return f"❌ **Error processing file:** {str(e)}\n\nPlease ensure the file is a valid PDF, DOCX, TXT, or MD file.", ""
 
 def clear_uploaded_file():
     """Clear uploaded file and reload KB index."""
@@ -666,12 +679,19 @@ def create_interface():
     return demo
 
 # ----------- Main Entry Point -----------
-
-# Load config and KB at import time
-config_file = os.getenv("RAG_CONFIG", "config.yaml")
-config = Config(config_file)
-kb = KBIndex()
-ensure_index()
-
-# Expose `demo` for Hugging Face Spaces
-demo = create_interface()
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='Configurable RAG Assistant')
+    parser.add_argument('--config', type=str, default='config.yaml',
+                        help='Path to configuration YAML file (default: config.yaml)')
+    args = parser.parse_args()
+    
+    # Load configuration
+    config = Config(args.config)
+    
+    # Initialize KB with config
+    kb = KBIndex()
+    ensure_index()
+    
+    # Create and launch interface
+    demo = create_interface()
+    demo.launch()