Update app.py

app.py

@@ -115,7 +115,7 @@ class SessionManager:
 # Initialize session manager
 session_manager = SessionManager()
     
-async def chat(query,history,sources,reports,subtype, client_ip=None, session_id = None, request:gr.Request = None):
+async def chat(query,history, method, sources,reports,subtype, client_ip=None, session_id = None, request:gr.Request = None):
     """taking a query and a message history, use a pipeline (reformulation, retriever, answering) 
        to yield a tuple of:(messages in gradio format/messages in langchain format, source documents)
     """
@@ -138,6 +138,51 @@ async def chat(query,history,sources,reports,subtype, client_ip=None, session_id
     #print(f"year:{year}")
     docs_html = ""
     output_query = ""
+    if method == "Search by Report Name":
+        if len(reports) == 0:
+            warning_message = "⚠️ **No Report Selected"
+            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'],
+                "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
+    else:
+        if sources is None and subtype is None:
+            warning_message = "⚠️ **No Report Selected"
+            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'],
+                "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
+            
 
     ##------------------------fetch collection from vectorstore------------------------------
     vectorstore = vectorstores["docling"]
@@ -150,6 +195,29 @@ async def chat(query,history,sources,reports,subtype, client_ip=None, session_id
                                                 sources=sources,subtype=subtype)
     end_time = time.time()
     print("Time for retriever:",end_time - start_time)
+
+        
+    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'],
+            "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]
     
@@ -633,7 +701,7 @@ with gr.Blocks(title="Audit Q&A", css= "style.css", theme=theme,elem_id = "main-
         .submit(get_client_ip_handler, [textbox], [client_ip], api_name="get_ip_textbox")
         .then(start_chat, [textbox, chatbot], [textbox, tabs, chatbot], queue=False, api_name="start_chat_textbox")
         .then(chat, 
-            [textbox, chatbot, dropdown_sources, dropdown_reports, dropdown_category, client_ip, session_id], 
+            [textbox, chatbot, search_method, dropdown_sources, dropdown_reports, dropdown_category, client_ip, session_id], 
             [chatbot, sources_textbox, feedback_state, session_id], 
             queue=True, concurrency_limit=8, api_name="chat_textbox")
         .then(show_feedback, [feedback_state], [feedback_row, feedback_thanks, feedback_state], api_name="show_feedback_textbox")
@@ -643,7 +711,7 @@ with gr.Blocks(title="Audit Q&A", css= "style.css", theme=theme,elem_id = "main-
         .change(start_chat, [examples_hidden, chatbot], [textbox, tabs, chatbot], queue=False, api_name="start_chat_examples")
         .then(get_client_ip_handler, [examples_hidden], [client_ip], api_name="get_ip_examples")
         .then(chat, 
-            [examples_hidden, chatbot, dropdown_sources, dropdown_reports, dropdown_category,  client_ip, session_id], 
+            [examples_hidden, chatbot, searh_method, dropdown_sources, dropdown_reports, dropdown_category,  client_ip, session_id], 
             [chatbot, sources_textbox, feedback_state, session_id], 
             concurrency_limit=8, api_name="chat_examples")
         .then(show_feedback, [feedback_state], [feedback_row, feedback_thanks, feedback_state], api_name="show_feedback_examples")