Update app.py

app.py

@@ -226,14 +226,11 @@ def retry_last_response(history, use_web_search, model, temperature, num_calls):
     
     return chatbot_interface(last_user_msg, history, use_web_search, model, temperature, num_calls)
 
-def respond(message, history, model, temperature, num_calls, use_web_search):
+def respond(message, history, model, temperature, num_calls, use_web_search, selected_docs):
     logging.info(f"User Query: {message}")
     logging.info(f"Model Used: {model}")
     logging.info(f"Search Type: {'Web Search' if use_web_search else 'PDF Search'}")
 
-    # Get the selected documents from the document_selector
-    selected_docs = document_selector.value
-
     logging.info(f"Selected Documents: {selected_docs}")
 
     try:
@@ -280,7 +277,7 @@ def respond(message, history, model, temperature, num_calls, use_web_search):
         if "microsoft/Phi-3-mini-4k-instruct" in model:
             logging.info("Falling back to Mistral model due to Phi-3 error")
             fallback_model = "mistralai/Mistral-7B-Instruct-v0.3"
-            yield from respond(message, history, fallback_model, temperature, num_calls, use_web_search)
+            yield from respond(message, history, fallback_model, temperature, num_calls, use_web_search, selected_docs)
         else:
             yield f"An error occurred with the {model} model: {str(e)}. Please try again or select a different model."
 
@@ -385,6 +382,10 @@ def get_response_from_pdf(query, model, selected_docs, num_calls=3, temperature=
     # Filter relevant_docs based on selected documents
     filtered_docs = [doc for doc in relevant_docs if doc.metadata["source"] in selected_docs]
     
+    if not filtered_docs:
+        yield "No relevant information found in the selected documents. Please try selecting different documents or rephrasing your query."
+        return
+
     context_str = "\n".join([doc.page_content for doc in filtered_docs])
 
     if model == "@cf/meta/llama-3.1-8b-instruct":