haystack_interactive.py

"""
Interactive Haystack RAG Demo
Run this to interact with the Haystack RAG system.
"""

from haystack_rag import HaystackRAG
import os

def main():
    """Interactive Haystack RAG demo."""
    print("🤖 Interactive Haystack RAG Demo")
    print("=" * 50)
    
    # Initialize RAG system
    rag = HaystackRAG(document_dir="documents")
    
    if not rag.pipeline:
        print("❌ Pipeline setup failed. Please check your Haystack installation.")
        return
    
    # Load documents
    rag.load_documents()
    
    if not rag.document_store.get_document_count():
        print("❌ No documents loaded. Please check your documents directory.")
        return
    
    print(f"✅ System ready! {rag.document_store.get_document_count()} documents loaded.")
    print("\n💡 Try these sample questions:")
    print("   - 'What is artificial intelligence?'")
    print("   - 'What are neural networks?'") 
    print("   - 'What is machine learning?'")
    print("   - 'How does deep learning work?'")
    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
                
            # Ask the question
            result = rag.ask_question(question)
            
            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()