Update app.py

app.py

@@ -94,9 +94,9 @@ 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. If no book title is given, return not relevant_segment.
+    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."
@@ -135,9 +135,7 @@ books = """
 - Lord of the Flies
 - Hamlet
 """
-books2 = """
 
-"""
 
 # Setup the Gradio Blocks interface with custom layout components
 with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
@@ -146,7 +144,6 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
        # with gr.Column():
             gr.Markdown(books)  # Show the topics on the left side
             gr.Markdown(topics)
-            # gr.Markdown(books2)
     with gr.Row():
         with gr.Column():
             book = gr.Dropdown(
@@ -156,7 +153,7 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
             question = gr.Textbox(label="Your question", placeholder="What do you want to ask about?")
             answer = gr.Textbox(label="LitBot Response", placeholder="LitBot will respond here...", interactive=False, lines=20)
             submit_button = gr.Button("Submit")
-            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