move retriever functions

app.py

@@ -5,6 +5,7 @@ import asyncio
 from uuid import uuid4
 from gradio_client import Client, handle_file
 from utils.whisp_api import handle_geojson_upload
+from utils.retriever import retrieve_paragraphs
 
 # Sample questions for examples 
 SAMPLE_QUESTIONS = {
@@ -26,35 +27,6 @@ SAMPLE_QUESTIONS = {
 }
 
 
-
-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"
-        )    
-        print(result)
-        return (
-            result,  
-            gr.update(visible=True),  # Keep status visible
-            gr.update(visible=False)  # Always hide results table
-        )
-
-    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
-            )
-
 def start_chat(query, history):
     """Start a new chat interaction"""
     history = history + [(query, None)]

utils/retriever.py

@@ -0,0 +1,31 @@
+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"
+        )    
+        print(result)
+        return (
+            result,  
+            gr.update(visible=True),  # Keep status visible
+            gr.update(visible=False)  # Always hide results table
+        )
+
+    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
+            )