interactive_test.py

"""
Interactive Test Script for RAG System
This allows you to test the system with your own questions.
"""

from simple_search import SimpleSearchRAG
import sys

def main():
    """Interactive test of the RAG system."""
    print("🤖 Interactive RAG Test")
    print("=" * 50)
    
    # Initialize RAG system
    rag = SimpleSearchRAG(document_dir="documents")
    
    if not rag.documents:
        print("❌ No documents loaded!")
        return
    
    print(f"✅ Loaded {len(rag.documents)} documents")
    print(f"📚 Documents: {[doc['filename'] for doc in rag.documents]}")
    print("\n💡 Try these sample questions:")
    print("   - 'artificial intelligence'")
    print("   - 'machine learning'") 
    print("   - 'neural networks'")
    print("   - 'deep learning'")
    print("   - 'computer vision'")
    print("\nType 'quit' to exit")
    print("-" * 50)
    
    while True:
        try:
            question = input("\n❓ Your question: ").strip()
            
            if question.lower() in ['quit', 'exit', 'q']:
                print("👋 Goodbye!")
                break
                
            if not question:
                continue
                
            # Show what documents we're searching
            print(f"🔍 Searching in {len(rag.documents)} documents...")
            
            # Get results
            result = rag.ask_question(question, top_k=3)
            
            if result['sources']:
                print(f"✅ Found {len(result['sources'])} relevant sources")
            else:
                print("❌ No relevant sources found")
                print("💡 Try different keywords or simpler terms")
            
        except KeyboardInterrupt:
            print("\n👋 Goodbye!")
            break
        except Exception as e:
            print(f"❌ Error: {e}")

if __name__ == "__main__":
    main()