Update app/main.py

app/main.py

@@ -68,26 +68,70 @@ def format_whisp_statistics(df):
         def_after_2020_raw = get_value(df, "TMF_def_after_2020")
         def_before_2020_raw = get_value(df, "TMF_def_before_2020")
 
+        
+        # Helper function to format risk levels with colors/emojis
+        def format_risk(risk_val):
+            if not risk_val or risk_val in ["Not available", "not available"]:
+                return "**No disponible**"
+            elif isinstance(risk_val, str):
+                risk_lower = risk_val.lower().strip()
+                if risk_lower == "low":
+                    return "*riesgo bajo*"
+                elif risk_lower == "medium":
+                    return "*riesgo medio*"
+                elif risk_lower == "high":
+                    return "*riesgo alto*"
+                elif risk_lower == "very high":
+                    return "*riesgo muy alto*"
+                elif risk_lower == "more_info_needed":
+                    return "*Se necesita más información.*"
+                else:
+                    return f"ℹ️ **{risk_val.title()}**"
+            return str(risk_val)
+
+        # Format deforestation data
+        def format_deforestation(def_val):
+            if not def_val or def_val in ["Not available", "not available"]:
+                return "*No disponible*"
+            try:
+                def_num = float(def_val)
+                if def_num == 0:
+                    return "* No se detectó deforestación.*"
+                elif def_num < 0.1:
+                    return f"*{def_num:.3f} hectáreas*"
+                else:
+                    return f"*{def_num:.2f} hectáreas*"
+            except:
+                return f"ℹ️ **{def_val}**"
+
+        deforestation_after_2020_formatted = format_deforestation(def_after_2020_raw)
+        deforestation_before_2020_formatted = format_deforestation(def_before_2020_raw)
+
         # Format for RAG context
-        context = f"""=== ANÁLISIS GEOGRÁFICO WHISP API ===
+        context = f"""
+
+**Respuesta generada mediante inteligencia artificíal:** \n\n
+        
+**Resultados del análisis geográfico**  \n\n
+La siguiente información ha sido generada por la [WhispAPI creada por Forest Data Partnership (FDaP)](https://openforis.org/solutions/whisp/).
 
-📍 Detalles de la ubicación
+📍 **Detalles de la ubicación:**
 
 - País: {country}
 - Región administrativa: {admin_level}
 - Área total: {area_text}
 
-⚠️ Evaluación del riesgo de deforestación
+⚠️ **Evaluación del riesgo de deforestación:**
 Los niveles de riesgo se basan en patrones históricos, factores ambientales y datos sobre el uso del suelo.
 
 - 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:
+🌳 **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
+- Deforestación antes de 2020: {deforestation_before_2020_formatted}
+- Deforestación después de 2020: {deforestation_after_2020_formatted}
 
 Fuente: Forest Data Partnership (FDaP) WhispAPI
 Fecha de análisis: {datetime.now().isoformat()}"""