Update app.py

app.py

@@ -74,7 +74,7 @@ def start_chat(query,history):
 def finish_chat():
     return (gr.update(interactive = True,value = ""))
     
-async def chat(query,history,sources,reports):
+async def chat(query,history,sources,reports,subtype,year):
     """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)"""
 
@@ -83,17 +83,17 @@ async def chat(query,history,sources,reports):
     #print(f"audience:{audience}")
     print(f"sources:{sources}")
     print(f"reports:{reports}")
+    print(f"reports:{subtype}")
+    print(f"reports:{year}")
     docs_html = ""
     output_query = ""
 
 
-    if reports is not None:
-        print(reports)
+    if len(reports) == 0:
         print(sources)
         #vectorstore = vectorstores[sources]
     else:
         print(reports)
-        print(sources)
         #vectorstore = vectorstores["allreports"]
 
 
@@ -304,12 +304,12 @@ with gr.Blocks(title="Audit Q&A", css= "style.css", theme=theme,elem_id = "main-
     # https://www.gradio.app/docs/gradio/textbox#event-listeners-arguments
     (textbox
      .submit(start_chat, [textbox,chatbot], [textbox,tabs,chatbot],queue = False,api_name = "start_chat_textbox")
-     .then(chat, [textbox,chatbot, dropdown_sources,dropdown_reports], [chatbot,sources_textbox],concurrency_limit = 8,api_name = "chat_textbox")
+     .then(chat, [textbox,chatbot, dropdown_sources,dropdown_reports,dropdown_category,dropdown_year], [chatbot,sources_textbox],concurrency_limit = 8,api_name = "chat_textbox")
      .then(finish_chat, None, [textbox],api_name = "finish_chat_textbox"))
 
     (examples_hidden
         .change(start_chat, [examples_hidden,chatbot], [textbox,tabs,chatbot],queue = False,api_name = "start_chat_examples")
-        .then(chat, [examples_hidden,chatbot, dropdown_sources,dropdown_reports], [chatbot,sources_textbox],concurrency_limit = 8,api_name = "chat_examples")
+        .then(chat, [examples_hidden,chatbot, dropdown_sources,dropdown_reports,dropdown_category,dropdown_year], [chatbot,sources_textbox],concurrency_limit = 8,api_name = "chat_examples")
         .then(finish_chat, None, [textbox],api_name = "finish_chat_examples")
     )