utils/retriever.py

import gradio as gr
from gradio_client import Client


def retrieve_paragraphs(query, country):
    """Connect to retriever and retrieve paragraphs"""
    try:   
        # Call the API with the uploaded file
        client = Client("https://giz-chatfed-retriever0-3.hf.space/")
        result = client.predict(
                query=query,
                collection_name= "EUDR",
                filter_metadata = {'country':country},
                api_name="/retrieve"
        )
        return result
        
    except Exception as e:
        error_msg = f"Error retrieving paragraphs: {str(e)}"
        return (
                error_msg,
                gr.update(visible=True),  # upload_status
                gr.update(visible=False)  # results_table
            )