Update app.py

app.py

@@ -5,6 +5,7 @@ import re
 import xmltodict
 from collections import Counter
 import unicodedata
+import pandas as pd  # Para manejar CSV
 
 # Función para normalizar palabras clave (elimina tildes y convierte a minúsculas)
 def normalize_keyword(keyword):
@@ -165,7 +166,7 @@ def get_top_suggestions(suggestions, top_n=10):
     suggestion_counter = Counter(suggestions)
     return suggestion_counter.most_common(top_n)
 
-# Función principal
+# Función principal que ahora también devuelve un archivo CSV
 def main(keyword):
     expanded_keywords = expand_keyword(keyword)
     all_suggestions = {}
@@ -174,8 +175,8 @@ def main(keyword):
     youtube_suggestions_all = []
     bing_suggestions_all = []
     amazon_suggestions_all = []
-    qwant_suggestions_all = []  # Para almacenar las sugerencias de Qwant
-    brave_suggestions_all = []  # Para almacenar las sugerencias de Brave
+    qwant_suggestions_all = []
+    brave_suggestions_all = []
 
     # Obtener sugerencias de DuckDuckGo
     for exp_keyword in expanded_keywords:
@@ -238,55 +239,34 @@ def main(keyword):
     brave_top_10 = get_top_suggestions(brave_suggestions_all, top_n=10)
 
     # Ordenar las sugerencias combinadas por frecuencia
-    combined_top_10_suggestions = get_top_suggestions(all_suggestions, top_n=10)
-    suggestions_str = ", ".join([f"{sug} ({freq})" for sug, freq in combined_top_10_suggestions])
+    combined_suggestions = sorted(all_suggestions.items(), key=lambda item: item[1], reverse=True)
 
-    # Crear la lista de todas las palabras clave con su número de repeticiones
-    all_suggestions_str = "<ul>"
-    for suggestion, freq in combined_top_10_suggestions:
-        all_suggestions_str += f"<li>{suggestion} - {freq} repeticiones</li>"
-    all_suggestions_str += "</ul>"
+    # Crear un DataFrame de Pandas para las sugerencias y relevancias
+    df = pd.DataFrame(combined_suggestions, columns=["keyword", "relevancia"])
 
-    # Crear el HTML de salida con un botón de copia
+    # Guardar el DataFrame en un archivo CSV
+    csv_file = "/tmp/keywords_relevancia.csv"
+    df.to_csv(csv_file, index=False)
+
+    # Crear el HTML de salida con las sugerencias (ya está hecho en tu código anterior)
     html_output = f"""
     <div>
-        <b>Sugerencias combinadas de Google, DuckDuckGo, YouTube, Bing, Amazon, Qwant y Brave (Top 10 combinadas):</b> <span id='suggestions_text'>{suggestions_str}</span>
+        <b>Sugerencias combinadas de Google, DuckDuckGo, YouTube, Bing, Amazon, Qwant y Brave (Top 10 combinadas):</b>
+        <span id='suggestions_text'>{', '.join([f'{sug} ({freq})' for sug, freq in combined_suggestions[:10]])}</span>
         <button class="lg secondary svelte-cmf5ev" style="font-size: small; padding: 2px; color: #808080ba; border: none; margin-left: 5px;" 
-        onclick='navigator.clipboard.writeText(document.getElementById("suggestions_text").innerText).then(() => alert("Texto copiado al portapapeles"))'>&nbsp;✂&nbsp;</button>
+        onclick='navigator.clipboard.writeText(document.getElementById("suggestions_text").innerText).then(() => alert("Texto copiado al portapapeles"))'>
+        &nbsp;✂&nbsp;</button>
     </div>
-
-    <h4>Top 10 Sugerencias de Google:</h4><ul>
     """
-    for suggestion, freq in google_top_10:
-        html_output += f"<li>{suggestion} ({freq})</li>"
-    html_output += "</ul><h4>Top 10 Sugerencias de DuckDuckGo:</h4><ul>"
-    for suggestion, freq in duckduckgo_top_10:
-        html_output += f"<li>{suggestion} ({freq})</li>"
-    html_output += "</ul><h4>Top 10 Sugerencias de YouTube:</h4><ul>"
-    for suggestion, freq in youtube_top_10:
-        html_output += f"<li>{suggestion} ({freq})</li>"
-    html_output += "</ul><h4>Top 10 Sugerencias de Bing:</h4><ul>"
-    for suggestion, freq in bing_top_10:
-        html_output += f"<li>{suggestion} ({freq})</li>"
-    html_output += "</ul><h4>Top 10 Sugerencias de Amazon:</h4><ul>"
-    for suggestion, freq in amazon_top_10:
-        html_output += f"<li>{suggestion} ({freq})</li>"
-    html_output += "</ul><h4>Top 10 Sugerencias de Qwant:</h4><ul>"
-    for suggestion, freq in qwant_top_10:
-        html_output += f"<li>{suggestion} ({freq})</li>"
-    html_output += "</ul><h4>Top 10 Sugerencias de Brave:</h4><ul>"
-    for suggestion, freq in brave_top_10:
-        html_output += f"<li>{suggestion} ({freq})</li>"
-    html_output += "</ul>"
-
-    return html_output
 
+    # Devolver el HTML y el archivo CSV
+    return html_output, csv_file
 
-# Interfaz de Gradio
+# Interfaz de Gradio con salida HTML y archivo descargable (CSV)
 iface = gr.Interface(
     fn=main,
     inputs="text",
-    outputs="html",
+    outputs=[gr.outputs.HTML(), gr.outputs.File(label="Descargar CSV")],
     title="<div style='margin:0 auto;text-align:center'><div style='margin:0 auto;text-align:center'><img style='width:100px;display: inline-table;margin-bottom:-10px' src='https://artxeweb.com/media/files/search.jpg'><p>Sugerencias Combinadas de Google, DuckDuckGo, YouTube, Bing y Amazon</p></div>",
     description="<p style='margin-bottom:10px;text-align:center;background: #ffffff; padding: 8px; border-radius: 8px; border-width: 1px; border: solid 1px #e5e7eb;'>Ingrese una palabra clave para obtener sugerencias de búsqueda relacionadas de Google, DuckDuckGo, YouTube, Bing y Amazon. Se mostrarán las 10 primeras sugerencias combinadas y también las 10 principales de cada plataforma por separado.</p>",
     article="<div style='margin-top:10px'><p style='text-align: center !important; background: #ffffff; padding: 5px 30px; border-radius: 8px; border-width: 1px; border: solid 1px #e5e7eb; width: fit-content; margin: auto;'>Desarrollada por <a style='text-decoration: none !important; color: #e12a31 !important;' href='https://artxeweb.com'>© Artxe Web</a></p></div>"