Update app.py

app.py

@@ -75,7 +75,7 @@ def generate_response(user_query, relevant_segment):
         response = openai.ChatCompletion.create(
             model="gpt-4o",
             messages=messages,
-            max_tokens=5000,
+            max_tokens=500,
             temperature=0.2,
             top_p=1,
             frequency_penalty=0.5,
@@ -96,12 +96,12 @@ def generate_response(user_query, relevant_segment):
 
 def query_model(question):
     """
-    Process a question, find relevant information, and generate a response.
+    Process a question, find relevant information, and generate a response. If no book title is given, return not relevant_segment.
     """
     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)
-    if not relevant_segment:
+    if not relevant_segment: 
         return "Could not find specific information. Please refine your question."
     response = generate_response(question, relevant_segment)
     return response