utils/retriever.py

import gradio as gr
from gradio_client import Client


def retrieve_paragraphs(query):
    """Connect to retriever and retrieve paragraphs"""
    try:   
        # Call the API with the uploaded file
        client = Client("https://giz-eudr-retriever.hf.space/")
        result = client.predict(
                query=query,
                reports_filter="",
                sources_filter="",
                subtype_filter="",
                year_filter="",
                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
            )