Update app.py

app.py

@@ -62,7 +62,7 @@ def find_relevant_segment(user_query, segments):
         print(f"Error in finding relevant segment: {e}")
         return ""
 
-def generate_response(user_query, book, relevant_segment):
+def generate_response(user_query, relevant_segment):
     """
     Generate a response emphasizing the bot's capability in providing literature information.
     """
@@ -100,7 +100,7 @@ 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, segments)
+    relevant_segment = find_relevant_segment(question and book, segments)
     if not relevant_segment: 
         return "Could not find specific information. Please refine your question."
     response = generate_response(question, relevant_segment)