Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import requests
 from bs4 import BeautifulSoup
-from itertools import combinations
 from urllib.parse import quote
 
 # Paleta de colores
@@ -31,27 +30,30 @@ def buscar_google(query, hl='es', num_results=10):
     return serp_data
 
 def generar_html_con_colores(keywords):
-    serp_results = {keyword: buscar_google(keyword) for keyword in keywords.split(',')}
+    keywords_list = keywords.split(',')
+    serp_results = {keyword.strip(): buscar_google(keyword.strip()) for keyword in keywords_list}
+    
+    max_length = max(len(results) for results in serp_results.values()) if serp_results else 0
+
     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:
+    for keyword in keywords_list:
         html_table += f"<th>{keyword.strip()}</th>"
     html_table += "</tr>"
 
     url_colors = {}
-    max_length = max(len(serp_results[keyword.strip()]) for keyword in keywords.split(','))
-
     for i in range(max_length):
-        html_table += "<tr><td>{}</td>".format(i + 1)
-        for keyword in keywords.split(','):
+        html_table += f"<tr><td>{i + 1}</td>"
+        for keyword in keywords_list:
             keyword = keyword.strip()
-            if i < len(serp_results[keyword]):
-                url = serp_results[keyword][i]["URL"]
+            results = serp_results.get(keyword, [])
+            if i < len(results):
+                url = results[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"])
+                html_table += f"<td><span style='color:{color}'>{results[i]['Título']}</span></td>"
             else:
                 html_table += "<td></td>"
         html_table += "</tr>"
@@ -66,4 +68,4 @@ iface = gr.Interface(
     description="Introduce las keywords separadas por comas para encontrar coincidencias en los resultados de búsqueda de Google."
 )
 
-iface.launch()
+iface.launch()