keep 2 filter options in ui and allow api to take more

app.py

@@ -28,6 +28,8 @@ def retrieve(
     query:str, 
     collection_name:str = None,
     filter_metadata:dict = None,
+    sources_input:str = None,
+    sources_value:str = None,
 ) -> list:
     """
     Retrieve semantically similar documents from the vector database for MCP clients.
@@ -42,7 +44,9 @@ def retrieve(
     Returns:
         list: List of dictionaries containing document content, metadata, and scores
     """
-    print(filter_metadata)
+    if not filter_metadata:
+        if sources_input and sources_value:
+            filter_metadata = {sources_input: sources_value}
     
     # Call retriever function and return raw results
     results = get_context(
@@ -85,9 +89,7 @@ with gr.Blocks() as ui:
                 lines=1, 
                 placeholder="Ecuador, Guatemala",
                 info="Filter by document subtype (leave empty for all)"
-            )
-
-            filter_metadata_state = gr.State(None)            
+            )           
             submit_btn = gr.Button("Submit", variant="primary")
         
         # Output needs to be in json format to be added as tool in HuggingChat
@@ -106,14 +108,9 @@ with gr.Blocks() as ui:
     #     api_name="retrieve"
     # )
     submit_btn.click(
-        fn=create_metadata_dict,
-        inputs=[sources_input, sources_value],
-        outputs=filter_metadata_state,
-        queue=False
-    ).then(
         # 3. Use the updated state as an input to the retrieve function.
         fn=retrieve,
-        inputs=[query_input, collection_name, filter_metadata_state],
+        inputs=[query_input, collection_name, None, sources_input, sources_value],
         outputs=output,
         queue=False
     )