No info found warning

app.py

@@ -179,6 +179,31 @@ async def chat(query, history, sources, reports, subtype, year, client_ip=None,
 
     context_retrieved = get_context(vectorstore=vectorstore,query=query,reports=reports,
                                                 sources=sources,subtype=subtype,year=year)
+    
+    # Check if any paragraphs were retrieved - add warning if none found
+    if not context_retrieved or len(context_retrieved) == 0:
+        warning_message = "⚠️ **No relevant information was found in the audit reports pertaining your query.** Please try rephrasing your question or selecting different report filters."
+        history[-1] = (query, warning_message)
+        # Update logs with the warning instead of answer
+        logs_data = {
+            "record_id": str(uuid4()),
+            "session_id": session_id,
+            "session_duration_seconds": session_duration,
+            "client_location": session_data['location_info'],
+            "platform": session_data['platform_info'],
+            "year": year,
+            "question": query,
+            "retriever": model_config.get('retriever','MODEL'),
+            "endpoint_type": model_config.get('reader','TYPE'),
+            "reader": model_config.get('reader','NVIDIA_MODEL'),
+            "answer": warning_message,
+            "no_results": True  # Flag to indicate no results were found
+        }
+        yield [tuple(x) for x in history], "", logs_data, session_id
+        # Save log for the warning response
+        save_logs(scheduler, JSON_DATASET_PATH, logs_data)
+        return
+    
     context_retrieved_formatted = "||".join(doc.page_content for doc in context_retrieved)
     context_retrieved_lst = [doc.page_content for doc in context_retrieved]