Update app.py

app.py

@@ -100,10 +100,10 @@ def query_model(question, book):
     """
     if question == "":
         return "Welcome to LitBot! Ask me anything about literature, book themes, and the historical context behind your book."
-    relevant_segment = find_relevant_segment(question, book, segments)
+    relevant_segment = find_relevant_segment(question, segments)
     if not relevant_segment: 
         return "Could not find specific information. Please refine your question."
-    response = generate_response(question, relevant_segment)
+    response = generate_response(question, book, relevant_segment)
     return response
 
 # Define the welcome message and specific topics the chatbot can provide information about