Sameer360/Document_based_chatbot

1### main.py ###
2from document_processor import process_documents
3from chatbot import ask
4from config import DOCUMENTS_DIR
5import os
6
7if __name__ == "__main__":
8    book_list = [
9        os.path.join(DOCUMENTS_DIR, file) for file in os.listdir(DOCUMENTS_DIR)
10    ]
11    db = process_documents(book_list)
12
13    while True:
14        query = input("Question: ")
15        if query.lower() == "exit":
16            break
17        ask(query, db)
18        print()
19