updated main

app/main.py

@@ -0,0 +1,182 @@
+# EUDR INGESTOR 
+
+import gradio as gr
+import os
+import logging
+from datetime import datetime
+from pathlib import Path
+from gradio_client import Client, handle_file
+import pandas as pd
+
+# Local imports
+from .utils import getconfig
+
+config = getconfig("params.cfg")
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+# WHISP API configuration
+WHISP_API_URL = config.get('whisp', 'WHISP_API_URL', fallback="https://giz-chatfed-whisp.hf.space/")
+
+def get_value(df, colname):
+    """Fetch value from WhispAPI-style Column/Value dataframe"""
+    if "Column" in df.columns and "Value" in df.columns:
+        match = df.loc[df["Column"] == colname, "Value"]
+        if not match.empty:
+            return match.values[0]
+    return "No disponible"
+
+def format_whisp_statistics(df):
+    """Format WhispAPI statistics into readable text for RAG context"""
+    try:
+        # Country code mapping
+        country_codes = {
+            'HND': 'Honduras', 'GTM': 'Guatemala', 'ECU': 'Ecuador',
+            'COL': 'Colombia', 'PER': 'Peru', 'BRA': 'Brasil',
+            'BOL': 'Bolivia', 'CRI': 'Costa Rica', 'PAN': 'Panamá',
+            'NIC': 'Nicaragua'
+        }
+        
+        country_raw = get_value(df, "Country")
+        country = country_codes.get(country_raw, country_raw)
+        admin_level = get_value(df, "Admin_Level_1")
+        area_raw = get_value(df, "Area")
+        
+        # Format area
+        try:
+            area_num = float(area_raw)
+            if area_num < 1:
+                area_text = f"{area_num:.3f} hectáreas"
+            elif area_num < 100:
+                area_text = f"{area_num:.2f} hectáreas"
+            else:
+                area_text = f"{area_num:,.1f} hectáreas"
+        except:
+            area_text = str(area_raw) if area_raw != "Not available" else "No disponible"
+
+        # Risk assessments
+        risk_pcrop = get_value(df, "risk_pcrop")
+        risk_acrop = get_value(df, "risk_acrop")
+        risk_timber = get_value(df, "risk_timber")
+        def_after_2020_raw = get_value(df, "TMF_def_after_2020")
+        def_before_2020_raw = get_value(df, "TMF_def_before_2020")
+
+        # Format for RAG context
+        context = f"""=== ANÁLISIS GEOGRÁFICO WHISP API ===
+País: {country}
+Región administrativa: {admin_level}
+Área total: {area_text}
+
+EVALUACIÓN DE RIESGO DE DEFORESTACIÓN:
+- Cultivos permanentes (Café, cacao, aceite de palma): {risk_pcrop}
+- Cultivos anuales (Soja, maíz, arroz): {risk_acrop}
+- Extracción de madera: {risk_timber}
+
+DATOS DE DEFORESTACIÓN:
+- Deforestación antes de 2020: {def_before_2020_raw} hectáreas
+- Deforestación después de 2020: {def_after_2020_raw} hectáreas
+
+Fuente: Forest Data Partnership (FDaP) WhispAPI
+Fecha de análisis: {datetime.now().isoformat()}"""
+
+        return context
+        
+    except Exception as e:
+        return f"Error en el análisis geográfico: {str(e)}"
+
+def process_geojson_whisp(file_content: bytes, filename: str) -> tuple[str, dict]:
+    """Process GeoJSON file through WHISP API and return formatted context"""
+    try:
+        # Create temporary file for WHISP API
+        import tempfile
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.geojson') as tmp_file:
+            tmp_file.write(file_content)
+            tmp_file_path = tmp_file.name
+        
+        try:
+            # Call WHISP API
+            client = Client(WHISP_API_URL)
+            result = client.predict(
+                file=handle_file(tmp_file_path),
+                api_name="/get_statistics"
+            )
+            
+            # Convert result to DataFrame
+            df = pd.DataFrame(result['data'], columns=result['headers'])
+            
+            # Format for RAG context
+            formatted_context = format_whisp_statistics(df)
+            
+            metadata = {
+                "analysis_type": "whisp_geojson",
+                "country": get_value(df, "Country"),
+                "admin_level": get_value(df, "Admin_Level_1"),
+                "area": get_value(df, "Area"),
+                "risk_levels": {
+                    "pcrop": get_value(df, "risk_pcrop"),
+                    "acrop": get_value(df, "risk_acrop"),
+                    "timber": get_value(df, "risk_timber")
+                }
+            }
+            
+            return formatted_context, metadata
+            
+        finally:
+            # Clean up temporary file
+            os.unlink(tmp_file_path)
+            
+    except Exception as e:
+        logger.error(f"WHISP API error: {str(e)}")
+        raise Exception(f"Failed to process GeoJSON through WHISP API: {str(e)}")
+
+def ingest(file):
+    """Main ingestion function - processes GeoJSON file and returns WHISP analysis context"""
+    if file is None:
+        return "No file uploaded", ""
+    
+    try:
+        with open(file.name, 'rb') as f:
+            file_content = f.read()
+        
+        filename = os.path.basename(file.name)
+        
+        # Check file extension
+        file_extension = os.path.splitext(filename)[1].lower()
+        if file_extension not in ['.geojson', '.json']:
+            raise ValueError(f"Unsupported file type: {file_extension}. Only GeoJSON files are supported.")
+        
+        # Process through WHISP API
+        context, metadata = process_geojson_whisp(file_content, filename)
+        
+        logger.info(f"Successfully processed GeoJSON {filename} through WHISP API")
+        
+        return context
+        
+    except Exception as e:
+        logger.error(f"GeoJSON processing failed: {str(e)}")
+        raise Exception(f"Processing failed: {str(e)}")
+
+if __name__ == "__main__":
+    ui = gr.Interface(
+        fn=ingest,
+        inputs=gr.File(
+            label="GeoJSON Upload",
+            file_types=[".geojson", ".json"]
+        ),
+        outputs=gr.Textbox(
+            label="WHISP Analysis Context",
+            lines=15,
+            show_copy_button=True
+        ),
+        title="EUDR Ingestion Module - WHISP API",
+        description="Processes GeoJSON files through WHISP API and returns geographic analysis context for RAG pipelines.",
+        api_name="ingest"
+    )
+
+    ui.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True
+    )