Update app.py

app.py

@@ -13,7 +13,7 @@ try:
     subprocess.run(["playwright", "install", "chromium"], check=True)
 except: pass
 
-# --- 1. GENERADOR DE URLS (Tu lógica exacta) ---
+# --- 1. GENERADOR DE URLS ---
 def construir_urls_final(zona, ciudad, tipo, hab, ban, park, antiguedad):
     # Mapeo Antigüedad
     mapa_ant = {
@@ -24,11 +24,7 @@ def construir_urls_final(zona, ciudad, tipo, hab, ban, park, antiguedad):
         "Más de 30 años": "mas-de-30-anos"
     }
     slug_ant = mapa_ant.get(antiguedad, "de-1-a-8-anios")
-    
-    # Lógica singular/plural
     slug_park = f"{int(park)}-parqueadero" if int(park) == 1 else f"{int(park)}-parqueaderos"
-    
-    # Searchstring
     query_geo = f"{zona.lower().replace(' ', '-')}-{ciudad.lower().replace(' ', '-')}"
     
     # URL FINCA RAÍZ
@@ -40,55 +36,71 @@ def construir_urls_final(zona, ciudad, tipo, hab, ban, park, antiguedad):
     
     return url_fr, url_mc
 
-# --- 2. GENERADOR PDF CON URLS VISIBLES ---
-def generar_pdf_con_fuentes(zona, df, url_fr, url_mc):
-    pdf_path = f"Reporte_{int(time.time())}.pdf"
+# --- 2. GENERADOR PDF CON ESTIMACIÓN FINANCIERA ---
+def generar_pdf_con_estimacion(zona, area, df, url_fr, url_mc):
+    # Cálculos Matemáticos
+    promedio_m2 = df['Precio_M2'].mean()
+    precio_sugerido = promedio_m2 * area
+    precio_min = df['Precio'].min()
+    precio_max = df['Precio'].max()
+    
+    pdf_path = f"Estimacion_{int(time.time())}.pdf"
     pdf = FPDF()
     pdf.add_page()
     
-    # Título
-    pdf.set_font("Arial", 'B', 14)
-    pdf.cell(0, 10, f"ESTUDIO DE MERCADO: {zona.upper()}", ln=True)
+    # Encabezado
+    pdf.set_font("Arial", 'B', 16)
+    pdf.cell(0, 10, f"ESTIMACION DE CANON: {zona.upper()}", ln=True, align='C')
+    pdf.ln(5)
+    
+    # --- BLOQUE DE VALORACIÓN (LO NUEVO) ---
+    pdf.set_fill_color(220, 230, 240)
+    pdf.rect(10, 30, 190, 40, 'F')
+    pdf.set_y(35)
     
-    # BLOQUE DE FUENTES (URLs Visibles)
+    pdf.set_font("Arial", 'B', 12)
+    pdf.cell(0, 8, f"CANON SUGERIDO (Area: {area}m2):", ln=True, align='C')
+    
+    pdf.set_font("Arial", 'B', 20)
+    pdf.set_text_color(0, 100, 0)
+    pdf.cell(0, 10, f"${precio_sugerido:,.0f}", ln=True, align='C')
+    
+    pdf.set_font("Arial", '', 10)
+    pdf.set_text_color(0, 0, 0)
+    pdf.cell(0, 8, f"Rango de Mercado: ${precio_min:,.0f} - ${precio_max:,.0f}", ln=True, align='C')
+    pdf.ln(15)
+
+    # Fuentes
     pdf.set_font("Arial", 'B', 8)
     pdf.set_text_color(100, 100, 100)
-    pdf.cell(0, 5, "FUENTES DE BÚSQUEDA GENERADAS:", ln=True)
+    pdf.cell(0, 5, "URLS DE BÚSQUEDA (AUDITORÍA):", ln=True)
     pdf.set_font("Courier", '', 7)
-    pdf.multi_cell(0, 4, f"FR: {url_fr}")
-    pdf.multi_cell(0, 4, f"MC: {url_mc}")
+    pdf.multi_cell(0, 4, f"FR: {url_fr}\nMC: {url_mc}")
     pdf.ln(5)
     
-    # Tabla de Resultados
+    # Detalle de Comparables
     pdf.set_text_color(0, 0, 0)
-    pdf.set_font("Arial", '', 10)
+    pdf.set_font("Arial", 'B', 12)
+    pdf.cell(0, 10, "COMPARABLES UTILIZADOS:", ln=True)
     
+    pdf.set_font("Arial", '', 10)
     for _, r in df.iterrows():
-        pdf.set_fill_color(240, 240, 240)
-        # Título
-        pdf.set_font("Arial", 'B', 10)
-        pdf.cell(0, 8, f"{r['Portal']} - ${r['Precio']:,.0f}", ln=True, fill=True)
-        # Detalle
-        pdf.set_font("Arial", '', 9)
-        pdf.multi_cell(0, 5, f"Desc: {r['Descripcion']}")
+        pdf.set_fill_color(245, 245, 245)
+        pdf.cell(0, 8, f"{r['Portal']} - ${r['Precio']:,.0f} (M2: ${r['Precio_M2']:,.0f})", ln=True, fill=True)
         pdf.set_font("Arial", 'U', 8)
         pdf.set_text_color(0, 0, 255)
-        pdf.cell(0, 5, f"Link: {r['URL']}", ln=True)
+        pdf.cell(0, 5, f"{r['URL']}", ln=True)
         pdf.set_text_color(0, 0, 0)
         pdf.ln(2)
         
     pdf.output(pdf_path)
-    return pdf_path
+    return pdf_path, precio_sugerido, precio_min, precio_max
 
 # --- 3. MOTOR DE EXTRACCIÓN ---
-def motor_tramitia_debug(zona, ciudad, area, tipo, hab, ban, park, antiguedad):
+def motor_tramitia_estimador(zona, ciudad, area, tipo, hab, ban, park, antiguedad):
     resultados = []
-    
-    # 1. Generamos URLs
     url_fr, url_mc = construir_urls_final(zona, ciudad, tipo, hab, ban, park, antiguedad)
-    
-    # INICIAMOS EL LOG VISIBLE
-    log_visible = f"✅ URLs GENERADAS POR EL SISTEMA:\n\n🌍 FINCA RAÍZ:\n{url_fr}\n\n🌍 METROCUADRADO:\n{url_mc}\n\n--- PROCESANDO ---\n"
+    log_visible = f"✅ URLs AUDITABLES:\nFR: {url_fr}\nMC: {url_mc}\n\n--- INICIANDO ESCANEO ---\n"
 
     with sync_playwright() as p:
         browser = p.chromium.launch(headless=True)
@@ -113,13 +125,11 @@ def motor_tramitia_debug(zona, ciudad, area, tipo, hab, ban, park, antiguedad):
                         "Portal": "Finca Raiz",
                         "Precio": max(precios),
                         "Precio_M2": max(precios) / area,
-                        "Descripcion": txt.split('\n')[0][:60],
                         "URL": href if href else url_fr
                     })
                 except: continue
             page.close()
-        except Exception as e:
-            log_visible += f"⚠️ Error FR: {str(e)}\n"
+        except Exception as e: log_visible += f"⚠️ Error FR: {e}\n"
 
         # --- METROCUADRADO ---
         try:
@@ -127,7 +137,6 @@ def motor_tramitia_debug(zona, ciudad, area, tipo, hab, ban, park, antiguedad):
             page.goto(url_mc, wait_until="domcontentloaded", timeout=60000)
             for _ in range(3): page.mouse.wheel(0, 800); time.sleep(1)
             
-            # Selector Detail específico que pediste
             details = page.query_selector_all("div.property-card__detail")
             for d in details[:5]:
                 try:
@@ -135,7 +144,6 @@ def motor_tramitia_debug(zona, ciudad, area, tipo, hab, ban, park, antiguedad):
                     precios = [int(s) for s in txt.replace('.', '').replace('$', '').split() if s.isdigit() and len(s) >= 6]
                     if not precios: continue
                     
-                    # Búsqueda de link hacia arriba
                     href = d.evaluate("""el => {
                         let card = el.closest('li') || el.closest('div[class*="card"]');
                         return card && card.querySelector('a') ? card.querySelector('a').href : '';
@@ -145,35 +153,40 @@ def motor_tramitia_debug(zona, ciudad, area, tipo, hab, ban, park, antiguedad):
                         "Portal": "Metrocuadrado",
                         "Precio": max(precios),
                         "Precio_M2": max(precios) / area,
-                        "Descripcion": txt.split('\n')[0][:60],
                         "URL": href if href else url_mc
                     })
                 except: continue
             page.close()
-        except Exception as e:
-            log_visible += f"⚠️ Error MC: {str(e)}\n"
-            
+        except Exception as e: log_visible += f"⚠️ Error MC: {e}\n"
         browser.close()
 
     if not resultados:
-        return f"{log_visible}\n❌ NO SE ENCONTRARON RESULTADOS.", None, None
+        return f"{log_visible}\n❌ NO HAY DATOS PARA CALCULAR.", None, None, "---"
 
     df = pd.DataFrame(resultados)
     
-    # Generar PDF con las URLs
-    pdf_path = generar_pdf_con_fuentes(zona, df, url_fr, url_mc)
+    # Generar PDF y Cálculos
+    pdf_path, sug, min_p, max_p = generar_pdf_con_estimacion(zona, area, df, url_fr, url_mc)
     
-    return f"{log_visible}\n✅ Éxito: {len(df)} inmuebles encontrados.", df, pdf_path
+    resumen_financiero = (
+        f"💰 **ESTIMACIÓN DE RENTA**\n"
+        f"🔹 **Canon Sugerido:** ${sug:,.0f}\n"
+        f"📉 Mínimo Zona: ${min_p:,.0f}\n"
+        f"📈 Máximo Zona: ${max_p:,.0f}\n"
+        f"📊 Basado en {len(df)} comparables."
+    )
+    
+    return f"{log_visible}\n✅ Cálculo Completado.", df, pdf_path, resumen_financiero
 
 # --- INTERFAZ ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🤖 TramitIA Pro: Auditoría de URLs")
+    gr.Markdown("## 🤖 TramitIA Pro: Calculadora de Renta Justificada")
     
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
             c = gr.Textbox(label="Ciudad", value="Bogota")
             z = gr.Textbox(label="Zona (Ej: Fontibon)", value="Fontibon")
-            a = gr.Number(label="Área M2", value=60)
+            a = gr.Number(label="Área M2 (Tu Cliente)", value=60)
             t = gr.Dropdown(["Apartamento", "Casa"], label="Tipo", value="Apartamento")
             
             with gr.Row():
@@ -185,14 +198,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 ["Menos de 1 año", "1 a 8 años", "9 a 15 años", "16 a 30 años", "Más de 30 años"],
                 label="Antigüedad", value="1 a 8 años"
             )
-            btn = gr.Button("BUSCAR Y MOSTRAR URLS", variant="primary")
+            btn = gr.Button("CALCULAR CANON DE ARRENDAMIENTO", variant="primary")
+            
+        with gr.Column(scale=2):
+            # RESULTADO DESTACADO
+            res_fin = gr.Markdown("### 💰 El resultado aparecerá aquí...")
             
-        with gr.Column():
-            # CAJA DE TEXTO GRANDE PARA VER LAS URLS
-            msg = gr.Textbox(label="🔍 Log de URLs Generadas (Copia y pega para probar)", lines=8)
-            out_df = gr.Dataframe(label="Resultados")
-            out_pdf = gr.File(label="Reporte con Fuentes")
+            with gr.Tabs():
+                with gr.TabItem("🔍 Log URLs"):
+                    msg = gr.Textbox(lines=6, label="Auditoría de Enlaces")
+                with gr.TabItem("📋 Tabla de Comparables"):
+                    out_df = gr.Dataframe()
+                with gr.TabItem("📄 Reporte PDF"):
+                    out_pdf = gr.File()
 
-    btn.click(motor_tramitia_debug, [z, c, a, t, h, b, p, e], [msg, out_df, out_pdf])
+    btn.click(motor_tramitia_estimador, [z, c, a, t, h, b, p, e], [msg, out_df, out_pdf, res_fin])
 
 demo.launch()