ts startup

app/main.py

@@ -1,5 +1,4 @@
 #CHATFED_ORCHESTRATOR
-
 import gradio as gr
 from fastapi import FastAPI, UploadFile, File, Form, Request
 from fastapi.responses import StreamingResponse
@@ -318,7 +317,7 @@ workflow.add_edge("geojson_direct", END)
 
 compiled_graph = workflow.compile()
 
-# TODO: PROBABLY REMOVE - NON-STREAMING VERSION
+# NON-STREAMING VERSION
 def process_query_core(
     query: str,
     reports_filter: str = "",

app/nodes.py

@@ -5,9 +5,22 @@ import tempfile
 import os
 from gradio_client import Client, file
 import logging
+from utils import getconfig
 
 logger = logging.getLogger(__name__)
 
+config = getconfig("params.cfg")
+RETRIEVER = config.get("retriever", "RETRIEVER", fallback="https://giz-chatfed-retriever.hf.space")
+GENERATOR = config.get("generator", "GENERATOR", fallback="https://giz-chatfed-generator.hf.space")
+INGESTOR = config.get("ingestor", "INGESTOR", fallback="https://mtyrrell-chatfed-ingestor.hf.space")
+GEOJSON_INGESTOR = config.get("ingestor", "GEOJSON_INGESTOR", fallback="https://giz-eudr-chatfed-ingestor.hf.space")
+MAX_CONTEXT_CHARS = config.get("general", "MAX_CONTEXT_CHARS")
+
+ingestor_url = INGESTOR
+retriever_url = RETRIEVER
+generator_url = GENERATOR
+geojson_ingestor_url = GEOJSON_INGESTOR
+
 # CORE PROCESSING NODES
 #----------------------------------------
 # File type detection node

app/utils.py

@@ -9,6 +9,8 @@ from typing import Optional, Dict, Any, List
 # Local .env file
 load_dotenv()
 
+logger = logging.getLogger(__name__)
+
 def getconfig(configfile_path: str):
     """
     Read the config file

requirements.txt

@@ -20,6 +20,7 @@ typing_extensions>=4.5.0
 python-multipart>=0.0.6
 pydantic>=2.0.0
 httpx>=0.24.0
+python-dotenv