Update app.py

app.py

@@ -1,7 +1,8 @@
+import gradio as gr
+import requests
 from bs4 import BeautifulSoup
 from itertools import combinations
-from IPython.display import display, HTML
-import requests
+from urllib.parse import quote
 
 # Paleta de colores
 color_palette = ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd', '#8c564b', '#e377c2', '#bcbd22', '#17becf', '#ff6666', '#4dff4d', '#6666ff', '#ffcc00', '#993399', '#00cc99', '#ff5050', '#33adff', '#ff9966']
@@ -10,7 +11,7 @@ def buscar_google(query, hl='es', num_results=10):
     all_results = []
 
     for start in range(0, num_results, 10):
-        url = f"https://www.google.com/search?q={query}&hl={hl}&start={start}"
+        url = f"https://www.google.com/search?q={quote(query)}&hl={hl}&start={start}"
         headers = {
             "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.3"
         }
@@ -29,59 +30,40 @@ def buscar_google(query, hl='es', num_results=10):
 
     return serp_data
 
-def calcular_coincidencias(serp1, serp2):
-    urls_serp1 = {entry["URL"] for entry in serp1}
-    urls_serp2 = {entry["URL"] for entry in serp2}
-    common_urls = urls_serp1.intersection(urls_serp2)
-
-    if serp1 and serp2 and serp1[0]["URL"] == serp2[0]["URL"]:
-        return min(100, round(len(common_urls) / len(urls_serp1) * 100 + 5))
-    elif serp1 and serp2 and serp1[1]["URL"] == serp2[1]["URL"]:
-        return min(100, round(len(common_urls) / len(urls_serp1) * 100 + 3))
-    else:
-        return min(100, round(len(common_urls) / len(urls_serp1) * 100))
-
-def generar_html_con_colores(serp_results, color_palette):
-    html_table = "<table border='1'><tr><th>Posición</th>"
+def generar_html_con_colores(keywords):
+    serp_results = {keyword: buscar_google(keyword) for keyword in keywords.split(',')}
+    html_table = "<style>td, th {border: 1px solid #ddd; text-align: left; padding: 8px;} tr:nth-child(even) {background-color: #f2f2f2;} th {padding-top: 11px; padding-bottom: 11px; background-color: #4CAF50; color: white;}</style>"
+    html_table += "<table><tr><th>Posición</th>"
 
     for keyword in serp_results:
-        html_table += f"<th>{keyword}</th>"
+        html_table += f"<th>{keyword.strip()}</th>"
     html_table += "</tr>"
 
     url_colors = {}
-    max_length = max(len(serp_results[keyword]) for keyword in serp_results)
+    max_length = max(len(serp_results[keyword.strip()]) for keyword in keywords.split(','))
+
     for i in range(max_length):
-        row_values = [str(i + 1)]
-        for keyword in serp_results:
+        html_table += "<tr><td>{}</td>".format(i + 1)
+        for keyword in keywords.split(','):
+            keyword = keyword.strip()
             if i < len(serp_results[keyword]):
-                serp_data_entry = serp_results[keyword][i]
-                url = serp_data_entry["URL"]
-                if sum(1 for s in serp_results.values() for entry in s if entry["URL"] == url) > 1:
-                    if url not in url_colors:
-                        url_colors[url] = color_palette[len(url_colors) % len(color_palette)]
-                    color = url_colors[url]
-                    row_values.extend([f"<span style='color: {color}'>{url}</span>"])
-                else:
-                    row_values.extend([url])
+                url = serp_results[keyword][i]["URL"]
+                if url not in url_colors:
+                    url_colors[url] = color_palette[len(url_colors) % len(color_palette)]
+                color = url_colors[url]
+                html_table += "<td><span style='color:{}'>{}</span></td>".format(color, serp_results[keyword][i]["Título"])
             else:
-                row_values.extend([""])
-        html_table += "<tr>" + "".join([f"<td>{value}</td>" for value in row_values]) + "</tr>"
+                html_table += "<td></td>"
+        html_table += "</tr>"
     html_table += "</table>"
     return html_table
 
-# Solicitar al usuario que introduzca las keywords
-keywords_input = input("Introduce las keywords separadas por coma: ")
-keywords = [keyword.strip() for keyword in keywords_input.split(',')]
-
-serp_results = {}
-for keyword in keywords:
-    serp_results[keyword] = buscar_google(keyword)
-
-coincidencia_matrix = {}
-for keyword1, keyword2 in combinations(keywords, 2):
-    coincidencia = calcular_coincidencias(serp_results[keyword1], serp_results[keyword2])
-    coincidencia_matrix[f"{keyword1} & {keyword2}"] = coincidencia
-    print(f"{keyword1} & {keyword2}: {coincidencia}%")
+iface = gr.Interface(
+    fn=generar_html_con_colores,
+    inputs="text",
+    outputs="html",
+    title="Comparador de Keywords en Google",
+    description="Introduce las keywords separadas por comas para encontrar coincidencias en los resultados de búsqueda de Google."
+)
 
-html_table_with_colors = generar_html_con_colores(serp_results, color_palette)
-display(HTML(html_table_with_colors))
+iface.launch()