Update app.py

app.py

@@ -11,8 +11,8 @@ import time
 # --- CONFIGURACIÓN DE ENTORNO ---
 try:
     subprocess.run(["playwright", "install", "chromium"], check=True)
-except Exception as e:
-    print(f"Playwright listo.")
+except Exception:
+    pass
 
 from playwright.sync_api import sync_playwright
 
@@ -21,128 +21,127 @@ USER_AGENTS = [
     "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
 ]
 
+# --- GENERADOR DE PDF (CORREGIDO) ---
+def generar_pdf_comparativo(zona, area, df):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", 'B', 16)
+    pdf.cell(0, 15, "TRAMITIA PRO - REPORTE COMPARATIVO", ln=True, align='C')
+    pdf.set_font("Arial", '', 10)
+    pdf.cell(0, 5, f"Fecha: {datetime.datetime.now().strftime('%d/%m/%Y')}", ln=True, align='C')
+    pdf.ln(10)
+
+    pdf.set_font("Arial", 'B', 12)
+    pdf.cell(0, 10, f"Analisis para: {zona.upper()} - Area: {area}m2", ln=True)
+    pdf.ln(5)
+
+    # Tabla de referencias
+    pdf.set_font("Arial", 'B', 9)
+    pdf.set_fill_color(200, 220, 255)
+    pdf.cell(20, 10, "Tipo", 1, 0, 'C', True)
+    pdf.cell(80, 10, "Descripcion", 1, 0, 'C', True)
+    pdf.cell(45, 10, "Precio", 1, 0, 'C', True)
+    pdf.cell(45, 10, "Valor m2", 1, 1, 'C', True)
+
+    pdf.set_font("Arial", '', 8)
+    for _, r in df.iterrows():
+        tipo = str(r['Tipo'])
+        desc = str(r['Descripción'])[:45]
+        precio = f"${r['Precio']:,.0f}"
+        val_m2 = f"${r['Precio_M2']:,.0f}"
+        
+        pdf.cell(20, 10, tipo, 1)
+        pdf.cell(80, 10, desc, 1)
+        pdf.cell(45, 10, precio, 1)
+        pdf.cell(45, 10, val_m2, 1, 1)
+
+    path_pdf = "Comparativa_TramitIA.pdf"
+    pdf.output(path_pdf)
+    return path_pdf
+
+# --- LÓGICA DE EXTRACCIÓN ---
 def extraer_datos(page, area_m2):
     referencias = []
-    # Esperar selectores de tarjetas de inmuebles
-    page.wait_for_selector("article, [class*='Card'], [class*='listing']", timeout=20000)
-    cards = page.query_selector_all("article, [class*='Card'], .listing-item")[:5]
-    
-    for card in cards:
-        try:
-            # Extraer URL
-            link_elem = card.query_selector("a")
-            url = "https://www.fincaraiz.com.co" + link_elem.get_attribute("href") if link_elem else "N/A"
-            
-            # Extraer Texto y Descripción
-            full_text = card.inner_text()
-            lineas = [l.strip() for l in full_text.split('\n') if len(l.strip()) > 5]
-            descripcion = lineas[0] if lineas else "Sin descripción"
-            
-            # Limpiar números para precio
-            solo_nums = full_text.replace('.', '').replace('$', '')
-            nums = [int(s) for s in solo_nums.split() if s.isdigit() and len(s) >= 6]
-            
-            if nums:
-                precio = max(nums)
-                # Intentar buscar área en el texto, si no usar la del usuario
-                area_ref = next((int(n) for n in solo_nums.split() if n.isdigit() and 20 < int(n) < 500), area_m2)
-                
-                referencias.append({
-                    "Portal": "Finca Raíz",
-                    "Descripción": descripcion[:100] + "...",
-                    "Precio": precio,
-                    "Area": area_ref,
-                    "Precio_M2": precio / area_ref,
-                    "URL": url
-                })
-        except: continue
+    try:
+        page.wait_for_selector("article, [class*='Card'], [class*='listing']", timeout=15000)
+        cards = page.query_selector_all("article, [class*='Card'], .listing-item")[:5]
+        for card in cards:
+            try:
+                link_elem = card.query_selector("a")
+                url = "https://www.fincaraiz.com.co" + link_elem.get_attribute("href") if link_elem else "N/A"
+                full_text = card.inner_text()
+                solo_nums = full_text.replace('.', '').replace('$', '')
+                nums = [int(s) for s in solo_nums.split() if s.isdigit() and len(s) >= 6]
+                if nums:
+                    precio = max(nums)
+                    area_ref = next((int(n) for n in solo_nums.split() if n.isdigit() and 20 < int(n) < 500), area_m2)
+                    referencias.append({
+                        "Portal": "Finca Raíz",
+                        "Descripción": full_text.split('\n')[0][:50],
+                        "Precio": precio,
+                        "Area": area_ref,
+                        "Precio_M2": precio / area_ref,
+                        "URL": url
+                    })
+            except: continue
+    except: pass
     return referencias
 
-def analizar_mercado(zona, area_m2, tipo_inmueble, modo):
+def comparar_mercado_tramitia(zona, area_m2, tipo):
     with sync_playwright() as p:
         browser = p.chromium.launch(headless=True)
         context = browser.new_context(user_agent=random.choice(USER_AGENTS))
         page = context.new_page()
-        
         zona_url = zona.lower().replace(" ", "-")
-        # Diccionario de URLs según modo
-        path_modo = "arriendo" if modo == "Arriendo" else "venta"
-        url = f"https://www.fincaraiz.com.co/{tipo_inmueble.lower()}/{path_modo}/{zona_url}"
         
-        try:
-            page.goto(url, wait_until="domcontentloaded", timeout=60000)
-            time.sleep(random.uniform(1, 2))
-            datos = extraer_datos(page, area_m2)
-            browser.close()
-            return datos
-        except Exception as e:
-            if 'browser' in locals(): browser.close()
-            return []
-
-def comparativa_tramitia(zona, area_m2, tipo):
-    # Ejecutar ambas búsquedas
-    res_arriendo = analizar_mercado(zona, area_m2, tipo, "Arriendo")
-    res_venta = analizar_mercado(zona, area_m2, tipo, "Venta")
-    
-    if not res_arriendo and not res_venta:
-        return "❌ No se encontraron datos suficientes para comparar.", None, None
+        # Búsqueda Arriendo
+        page.goto(f"https://www.fincaraiz.com.co/{tipo.lower()}/arriendo/{zona_url}", wait_until="domcontentloaded")
+        data_a = extraer_datos(page, area_m2)
+        
+        # Búsqueda Venta
+        page.goto(f"https://www.fincaraiz.com.co/{tipo.lower()}/venta/{zona_url}", wait_until="domcontentloaded")
+        data_v = extraer_datos(page, area_m2)
+        
+        browser.close()
 
-    df_a = pd.DataFrame(res_arriendo)
-    df_v = pd.DataFrame(res_venta)
-    
-    html_resumen = f"<div style='padding:20px; border-radius:10px; background-color:#f8fafc; border:1px solid #e2e8f0;'>"
-    html_resumen += f"<h2 style='color:#1e40af;'>📊 Comparativa TramitIA Pro: {zona}</h2>"
-    
-    if not df_a.empty:
-        prom_a = df_a['Precio'].mean()
-        html_resumen += f"<p style='font-size:1.1em;'>🏠 <b>Promedio Arriendo:</b> ${prom_a:,.0f} COP</p>"
-    
-    if not df_v.empty:
-        prom_v = df_v['Precio'].mean()
-        html_resumen += f"<p style='font-size:1.1em;'>💰 <b>Promedio Venta:</b> ${prom_v:,.0f} COP</p>"
-        if not df_a.empty:
-            rentabilidad = ( (df_a['Precio'].mean() * 12) / prom_v ) * 100
-            html_resumen += f"<div style='margin-top:10px; padding:10px; background:#dcfce7; color:#166534; border-radius:5px;'>"
-            html_resumen += f"<b>Rentabilidad Estimada (PER):</b> {rentabilidad:,.2f}% anual</div>"
-    
-    html_resumen += "</div>"
-    
-    # Unir ambos dataframes para mostrar las referencias detalladas
-    df_a['Tipo'] = 'Arriendo'
-    df_v['Tipo'] = 'Venta'
+    if not data_a and not data_v:
+        return "⚠️ No se hallaron datos.", None, None
+
+    df_a = pd.DataFrame(data_a); df_a['Tipo'] = 'Arriendo'
+    df_v = pd.DataFrame(data_v); df_v['Tipo'] = 'Venta'
     df_final = pd.concat([df_a, df_v], ignore_index=True)
+
+    # CÁLCULOS
+    prom_a = df_a['Precio'].mean() if not df_a.empty else 0
+    prom_v = df_v['Precio'].mean() if not df_v.empty else 0
+    
+    resumen_html = f"""
+    <div style='padding:15px; background:#f0f9ff; border-radius:10px; border:1px solid #bae6fd;'>
+        <h3 style='color:#0369a1; margin:0;'>📊 Comparativa TramitIA Pro</h3>
+        <p>Promedio Arriendo: <b>${prom_a:,.0f} COP</b></p>
+        <p>Promedio Venta: <b>${prom_v:,.0f} COP</b></p>
+    </div>
+    """
     
-    # Generar PDF
-    pdf_path = "Comparativa_TramitIA.pdf"
-    # (Lógica simplificada de PDF para el ejemplo)
+    # GENERACIÓN OBLIGATORIA DEL PDF
+    pdf_generado = generar_pdf_comparativo(zona, area_m2, df_final)
     
-    return html_resumen, df_final, pdf_path
+    return resumen_html, df_final, pdf_generado
 
-# --- INTERFAZ GRADIO ---
+# --- INTERFAZ ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.HTML("<h1 style='text-align:center; color:#1e40af;'>🤖 TramitIA Pro</h1>")
-    gr.HTML("<p style='text-align:center;'>Análisis Comparativo de Mercado (Arriendo vs Venta)</p>")
-    
+    gr.HTML("<h1 style='text-align:center'>🤖 TramitIA Pro</h1>")
     with gr.Row():
         with gr.Column(scale=1):
-            zona_in = gr.Textbox(label="Zona/Barrio y Ciudad", placeholder="Ej: El Salitre Bogota")
-            area_in = gr.Number(label="Área M2", value=80)
-            tipo_in = gr.Dropdown(["Apartamento", "Casa", "Oficina"], label="Tipo", value="Apartamento")
-            btn_comp = gr.Button("COMPARAR ARRIENDO VS VENTA", variant="primary")
-            
+            zona = gr.Textbox(label="Zona (Ciudad y Barrio)")
+            area = gr.Number(label="M2", value=70)
+            tipo = gr.Dropdown(["Apartamento", "Casa"], label="Tipo", value="Apartamento")
+            btn = gr.Button("EJECUTAR COMPARATIVA", variant="primary")
         with gr.Column(scale=2):
-            res_html = gr.HTML()
-            with gr.Tabs():
-                with gr.TabItem("Detalle de Referencias"):
-                    data_view = gr.Dataframe(interactive=False)
-                with gr.TabItem("Descargas"):
-                    file_out = gr.File(label="Reporte Técnico PDF")
-
-    btn_comp.click(
-        comparativa_tramitia, 
-        inputs=[zona_in, area_in, tipo_in], 
-        outputs=[res_html, data_view, file_out]
-    )
-
-if __name__ == "__main__":
-    demo.launch()
+            res = gr.HTML()
+            table = gr.Dataframe(label="Referencias Detalladas")
+            file = gr.File(label="Descargar Reporte PDF")
+
+    btn.click(comparar_mercado_tramitia, [zona, area, tipo], [res, table, file])
+
+demo.launch()