Update app.py

app.py

@@ -83,16 +83,18 @@ def generate_response(
 
     context = search_relevant_text(message)  # Get relevant content from PDF
 
-    # Start with the system message in the first user message
-    messages = []
-    first_msg = f"{system_message}\n\nContext: {context}\nQuestion: {message}"
-    messages.append({"role": "user", "content": first_msg})
+    if not context.strip():  # If no relevant content is found, refuse to answer
+        return "I can only answer based on the uploaded PDF. Your question is outside the document's content."
+
+    messages = [
+        {"role": "user", "content": f"{system_message}\n\nContext: {context}\nQuestion: {message}"}
+    ]
 
     # Add conversation history ensuring alternating pattern (user, assistant, user, assistant...)
     for user_msg, bot_msg in history:
-        if user_msg.strip():  # Check if user message is not empty
+        if user_msg.strip():
             messages.append({"role": "user", "content": user_msg})
-        if bot_msg.strip():  # Check if assistant message is not empty
+        if bot_msg.strip():
             messages.append({"role": "assistant", "content": bot_msg})
 
     try:
@@ -106,6 +108,7 @@ def generate_response(
         print(f"Error generating response: {str(e)}")
         return "I apologize, but I encountered an error while generating the response. Please try again."
 
+
 @app.route('/')
 def index():
     """Serve the HTML page for the user interface"""