Update app.py

app.py

@@ -94,16 +94,17 @@ def generate_response(user_query, relevant_segment):
         print(f"Error in generating response: {e}")
         return f"Error in generating response: {e}"
 
-def query_model(question):
+def query_model(question, book):
     """
     Process a question, find relevant information, and generate a response using inputted book title. 
     """
     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)
+    full_query = f"{book}: {question}"
+    relevant_segment = find_relevant_segment(full_query, segments)  
     if not relevant_segment: 
         return "Could not find specific information. Please refine your question."
-    response = generate_response(question, relevant_segment)
+    response = generate_response(full_query, relevant_segment)
     return response
 
 # Define the welcome message and specific topics the chatbot can provide information about
@@ -149,15 +150,15 @@ with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
             gr.Markdown(topics)
     with gr.Row():
         with gr.Column():
-            # book = gr.Dropdown(
-            #     ["The Great Gatsby", "The Crucible", "Fahrenheit 451", "Of Mice and Men", "To Kill a Mockingbird", "Romeo and Juliet", "The Catcher in the Rye", "Pride and Prejudice", "Lord of the Flies", "Hamlet"],
-            #     label = "Choose a book!",
-            #     interactive = True )
+            book = gr.Dropdown(
+                ["The Great Gatsby", "The Crucible", "Fahrenheit 451", "Of Mice and Men", "To Kill a Mockingbird", "Romeo and Juliet", "The Catcher in the Rye", "Pride and Prejudice", "Lord of the Flies", "Hamlet"],
+                label = "Choose a book!",
+                interactive = True )
             question = gr.Textbox(label="Your question", placeholder="What do you want to ask about?")
             gr.Markdown(space)
             submit_button = gr.Button("Submit")
             answer = gr.Textbox(label="LitBot Response", placeholder="LitBot will respond here...", interactive=False, lines=30)
-            submit_button.click(fn=query_model, inputs=question, outputs=answer)
+            submit_button.click(fn=query_model, inputs=[question, book], outputs=answer)
             
     
 # Launch the Gradio app to allow user interaction