from rag import get_answer
import sys

def main():
    print("🌟 Sagar's Personal Assistant: Hello! 👋 I'm here to answer questions about Sagar. (Type 'exit' to quit)")
    
    while True:
        try:
            query = input("\nYou: ")
            if query.lower() in ["exit", "quit", "q"]:
                print("🌟 Sagar's Personal Assistant: Goodbye! Take care! 👋😊")
                break
            
            if not query.strip():
                continue

            print("🌟 Sagar's Personal Assistant: Let me think... 🤔")
            response = get_answer(query)
            print(f"🌟 Sagar's Personal Assistant: {response}")

        except KeyboardInterrupt:
            print("\n🌟 Sagar's Personal Assistant: Goodbye! Take care! 👋😊")
            sys.exit()
        except Exception as e:
            print(f"Error: {e}")

if __name__ == "__main__":
    main()