haystack_interactive_demo.py

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

from haystack_demo import HaystackStyleRAG
import os

def main():
    """Interactive Haystack RAG demo."""
    print("🤖 Interactive Haystack RAG Demo")
    print("=" * 50)
    
    # Initialize RAG system
    rag = HaystackStyleRAG(document_dir="documents")
    
    # Load documents
    rag.load_documents()
    
    if not rag.document_store:
        print("❌ No documents loaded. Please check your documents directory.")
        return
    
    print(f"✅ System ready! {len(rag.document_store)} documents in document store.")
    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
                
            # Run the Haystack-style pipeline
            results = rag.run_pipeline(question, top_k=3)
            
            if results['documents']:
                print(f"🤖 Answer: {results['answer']}")
                print(f"📚 Sources: {', '.join(results['sources'])}")
                print(f"🔍 Retrieved {len(results['documents'])} relevant documents")
            else:
                print("❌ No relevant documents 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()