Update app.py

app.py

@@ -9,98 +9,108 @@ import plotly.express as px
 from playwright.sync_api import sync_playwright
 import time
 
-# --- CONFIGURACIÓN DE NAVEGADOR ---
+# --- CONFIGURACIÓN DE ENTORNO ---
 try:
     subprocess.run(["playwright", "install", "chromium"], check=True)
 except:
     pass
 
-# --- GENERADOR DE REPORTE TÉCNICO JUSTIFICADO ---
-def generar_pdf_justificado(zona, area_m2, df):
+# --- GENERADOR DE REPORTE TÉCNICO ---
+def generar_pdf_pro(zona, area_m2, df):
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", 'B', 16)
     pdf.set_text_color(26, 54, 93)
-    pdf.cell(0, 15, "TRAMITIA PRO - REPORTE TÉCNICO DE ESTIMACIÓN", ln=True, align='C')
+    pdf.cell(0, 15, "TRAMITIA PRO - INFORME TÉCNICO DE ESTIMACIÓN", ln=True, align='C')
     
     pdf.set_font("Arial", '', 10)
     pdf.set_text_color(0, 0, 0)
-    pdf.cell(0, 5, f"Zona: {zona.upper()} | Area Ref: {area_m2} m2", ln=True)
+    pdf.cell(0, 5, f"Zona Analizada: {zona.upper()}", ln=True)
+    pdf.cell(0, 5, f"Área de Referencia: {area_m2} m2", ln=True)
     pdf.cell(0, 5, f"Fecha: {datetime.datetime.now().strftime('%d/%m/%Y')}", ln=True)
     pdf.ln(10)
 
     for i, r in df.iterrows():
-        pdf.set_font("Arial", 'B', 10)
-        pdf.set_fill_color(240, 244, 248)
+        pdf.set_font("Arial", 'B', 11)
+        pdf.set_fill_color(230, 235, 245)
         pdf.cell(0, 8, f"REFERENCIA #{i+1} - {r['Portal']}", ln=True, fill=True)
         
         pdf.set_font("Arial", '', 9)
-        pdf.cell(0, 7, f"Precio: ${r['Precio']:,.0f} | Valor M2: ${r['Precio_M2']:,.0f} | Hab: {r['Habitaciones']} | Banos: {r['Banos']}", ln=True)
+        # Detalle técnico solicitado
+        detalle = f"Precio: ${r['Precio']:,.0f} | Hab: {r['Habitaciones']} | Baños: {r['Banos']} | Garajes: {r['Garajes']} | Edad: {r['Antiguedad']}"
+        pdf.cell(0, 7, detalle, ln=True)
         
         pdf.set_font("Arial", 'I', 8)
-        pdf.multi_cell(0, 5, f"Resumen: {r['Descripcion']}")
+        pdf.multi_cell(0, 5, f"Descripción: {r['Descripcion']}")
         
         pdf.set_font("Arial", 'U', 8)
         pdf.set_text_color(0, 0, 255)
-        # Link funcional
-        pdf.cell(0, 7, f"Link Directo: {r['URL']}", ln=True)
+        pdf.cell(0, 7, f"Link de la publicación: {r['URL']}", ln=True)
         pdf.set_text_color(0, 0, 0)
-        pdf.ln(4)
+        pdf.ln(5)
         
         if pdf.get_y() > 250: pdf.add_page()
 
-    path_pdf = "Reporte_TramitIA_Expert.pdf"
+    path_pdf = "Estimacion_Renta_TramitIA.pdf"
     pdf.output(path_pdf)
     return path_pdf
 
-# --- MOTOR DE BÚSQUEDA DUAL ---
+# --- MOTOR DE BÚSQUEDA HÍBRIDO ---
 def motor_tramitia_pro(zona, area, tipo, hab, ban, park, edad):
-    # Limpieza estricta de la zona para evitar saltos de ciudad
-    ciudad_busqueda = "bogota" # Puedes dinamizar esto si planeas otras ciudades
-    barrio_limpio = zona.lower().replace("bogota", "").strip().replace(" ", "-")
-    
+    # Forzar la ciudad para evitar saltos de localización
+    busqueda_url = zona.lower().replace(" ", "-")
+    if "bogota" not in busqueda_url:
+        busqueda_url += "-bogota"
+
     all_data = []
     
     with sync_playwright() as p:
         browser = p.chromium.launch(headless=True)
-        context = browser.new_context(user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64)")
+        # Cambiamos el contexto para parecer un navegador real en Colombia
+        context = browser.new_context(user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36")
         
-        # URLs ESTRUCTURADAS
+        # FUENTES
         portales = [
             {
                 "id": "Finca Raiz",
-                "url": f"https://www.fincaraiz.com.co/{tipo.lower()}/arriendo/{barrio_limpio}-bogota?habitaciones={int(hab)}&banos={int(ban)}"
+                "url": f"https://www.fincaraiz.com.co/{tipo.lower()}/arriendo/{busqueda_url}?habitaciones={int(hab)}&banos={int(ban)}"
             },
             {
                 "id": "Metrocuadrado",
-                "url": f"https://www.metrocuadrado.com/{tipo.lower()}/arriendo/bogota/{barrio_limpio}/{int(hab)}-habitaciones/{int(ban)}-banos/"
+                "url": f"https://www.metrocuadrado.com/{tipo.lower()}/arriendo/bogota/{busqueda_url.replace('-bogota','')}/{int(hab)}-habitaciones/{int(ban)}-banos/"
             }
         ]
 
         for p_info in portales:
             page = context.new_page()
             try:
+                # Navegar y esperar carga real
                 page.goto(p_info["url"], wait_until="load", timeout=60000)
-                time.sleep(4)
-                page.mouse.wheel(0, 2000) # Scroll para cargar descripciones
+                time.sleep(5) 
+                page.mouse.wheel(0, 2000) # Scroll para cargar contenido dinámico
                 
-                cards = page.query_selector_all("article, [class*='Card'], .listing-item")[:3]
+                # Buscamos anuncios que coincidan con la zona para evitar falsos positivos
+                cards = page.query_selector_all("article, [class*='Card'], .listing-item")
                 
                 for card in cards:
+                    if len(all_data) >= 6: break # Límite de referencias
+                    
                     try:
-                        # 1. Link Absoluto
+                        full_text = card.inner_text()
+                        # Validación de seguridad: Si el texto del anuncio no menciona la zona o ciudad, lo ignoramos
+                        if "medellin" in full_text.lower() and "bogota" in busqueda_url:
+                            continue
+
                         link_elem = card.query_selector("a")
                         href = link_elem.get_attribute("href") if link_elem else ""
                         url_final = href if "http" in href else f"https://www.{p_info['id'].lower().replace(' ', '')}.com.co{href}"
                         
-                        # 2. Descripción y Precio
-                        texto_completo = card.inner_text()
-                        precios = [int(s) for s in texto_completo.replace('.', '').replace('$', '').split() if s.isdigit() and len(s) >= 6]
+                        precios = [int(s) for s in full_text.replace('.', '').replace('$', '').split() if s.isdigit() and len(s) >= 6]
                         
                         if precios:
-                            # Limpieza de descripción: tomamos las primeras líneas que no son números
-                            desc_lineas = [l for l in texto_completo.split('\n') if len(l) > 15][:2]
-                            descripcion = " ".join(desc_lineas) if desc_lineas else "Ver detalle en link"
+                            # Extraer descripción real (buscando líneas con texto largo)
+                            desc_lineas = [l.strip() for l in full_text.split('\n') if len(l) > 20]
+                            descripcion = desc_lineas[0] if desc_lineas else "Ver más detalles en el portal."
 
                             all_data.append({
                                 "Portal": p_info["id"],
@@ -119,38 +129,44 @@ def motor_tramitia_pro(zona, area, tipo, hab, ban, park, edad):
         browser.close()
 
     if not all_data:
-        return "⚠️ Sin datos. Prueba con 'Modelia Bogota' o reduce filtros.", None, None, None, None
+        return "⚠️ No se hallaron datos exactos. Intenta con una zona más amplia o verifica la ortografía.", None, None, None, None
 
     df = pd.DataFrame(all_data)
-    pdf_path = generar_pdf_justificado(zona, area, df)
-    fig = px.bar(df, x="Portal", y="Precio_M2", color="Portal", title="Comparativa Valor M2")
+    pdf_path = generar_pdf_pro(zona, area, df)
+    
+    # Gráfica comparativa
+    fig = px.bar(df, x="Portal", y="Precio_M2", color="Portal", barmode="group",
+                 title="Comparativa Valor M2 por Fuente")
+
+    # Mapa centrado en Bogotá
     m = folium.Map(location=[4.66, -74.11], zoom_start=14)
+    folium.Marker([4.66, -74.11], popup=f"Análisis {zona}").add_to(m)
 
-    return f"✅ Análisis Exitoso: {len(df)} referencias encontradas.", df, pdf_path, fig, m._repr_html_()
+    return f"✅ Análisis Exitoso para {zona}. Se encontraron {len(df)} referencias justificadas.", df, pdf_path, fig, m._repr_html_()
 
-# --- INTERFAZ ---
+# --- INTERFAZ PANEL DE CONTROL ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.HTML("<h1 style='text-align:center'>🤖 TramitIA Pro: Centro de Estimación</h1>")
+    gr.HTML("<h1 style='text-align:center'>🤖 TramitIA Pro</h1><p style='text-align:center;'>Sistema de Estimación de Renta Justificada</p>")
     
     with gr.Row():
         with gr.Column(scale=1):
-            zona_in = gr.Textbox(label="Barrio y Ciudad", value="Modelia Bogota")
-            area_in = gr.Number(label="Área M2", value=70)
-            tipo_in = gr.Dropdown(["Apartamento", "Casa"], label="Tipo", value="Apartamento")
+            zona_in = gr.Textbox(label="Ubicación (Barrio y Ciudad)", value="Salitre Bogota")
+            area_in = gr.Number(label="Área M2", value=150)
+            tipo_in = gr.Dropdown(["Apartamento", "Casa"], label="Tipo de Inmueble", value="Apartamento")
             with gr.Row():
                 hab_in = gr.Number(label="Habitaciones", value=3)
                 ban_in = gr.Number(label="Baños", value=2)
             with gr.Row():
-                gar_in = gr.Number(label="Garajes", value=1)
+                gar_in = gr.Number(label="Parqueaderos", value=1)
                 ant_in = gr.Dropdown(["Nuevo", "1-5 años", "5-10 años", "Más de 10"], label="Antigüedad", value="5-10 años")
-            btn = gr.Button("GENERAR ESTIMACIÓN JUSTIFICADA", variant="primary")
+            btn = gr.Button("EJECUTAR ANÁLISIS PROFUNDO", variant="primary")
             
         with gr.Column(scale=2):
-            msg = gr.Markdown()
+            msg = gr.Markdown("Esperando consulta...")
             with gr.Tabs():
-                with gr.TabItem("📋 Ficha Técnica"):
+                with gr.TabItem("📋 Ficha Técnica y PDF"):
                     table = gr.Dataframe(interactive=False)
-                    file = gr.File(label="Descargar PDF de Consultoría")
+                    file = gr.File(label="Descargar Reporte de Consultoría")
                 with gr.TabItem("📊 Mapa y Gráficas"):
                     plot = gr.Plot()
                     mapa = gr.HTML()