Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 import urllib.parse
 import re
 import xmltodict
-import inflect
 
 # Función para obtener sugerencias de DuckDuckGo
 def fetch_duckduckgo_suggestions(query, lang_code="es"):
@@ -42,9 +41,11 @@ def fetch_youtube_suggestions(query, lang_code="es"):
     
     if response.status_code == 200:
         try:
+            # Extraer las sugerencias del formato window.google.ac.h(["..."])
             match = re.search(r'window\.google\.ac\.h\(\["[^"]*",\[(.*?)\],', response.text)
             if match:
                 suggestions_data = match.group(1)
+                # Extraemos las sugerencias
                 suggestions = re.findall(r'\["([^"]+)"', suggestions_data)
                 return suggestions
             else:
@@ -99,6 +100,7 @@ def fetch_amazon_suggestions(query, market_id="ATVPDKIKX0DER", alias="aps"):
     if response.status_code == 200:
         try:
             data = response.json()
+            # Extraemos las sugerencias desde el JSON bajo la clave 'suggestions'
             return [item['value'] for item in data.get('suggestions', [])]
         except ValueError:
             print("Error decodificando JSON de Amazon")
@@ -106,27 +108,9 @@ def fetch_amazon_suggestions(query, market_id="ATVPDKIKX0DER", alias="aps"):
     else:
         return []
 
-# Función para generar plurales y variaciones
-def generate_plural_variations(keyword):
-    p = inflect.engine()
-    singular = keyword
-    plural = p.plural(keyword)
-    return [singular, plural]
-
-# Función para expandir la palabra clave con stopwords y conjunciones
-STOP_WORDS = ["de", "en", "por", "para", "con", "sin", "y", "o"]
-
-def expand_keyword_with_stopwords(keyword):
-    expanded_keywords = [keyword]
-    for word in STOP_WORDS:
-        expanded_keywords.append(f"{keyword} {word}")
-        expanded_keywords.append(f"{word} {keyword}")
-    return expanded_keywords
-
-# Función para expandir las palabras clave
+# Función para expandir la palabra clave
 def expand_keyword(keyword):
-    expanded_keywords = expand_keyword_with_stopwords(keyword)
-    expanded_keywords += generate_plural_variations(keyword)
+    expanded_keywords = [keyword]
     for letter in 'abcdefghijklmnopqrstuvwxyz*_':
         expanded_keywords.append(keyword + " " + letter)
         expanded_keywords.append(letter + " " + keyword)
@@ -142,32 +126,38 @@ def main(keyword):
     bing_suggestions_all = []
     amazon_suggestions_all = []
 
-    # Obtener sugerencias de cada fuente
+    # Obtener sugerencias de DuckDuckGo
     for exp_keyword in expanded_keywords:
         suggestions = fetch_duckduckgo_suggestions(exp_keyword)
-        duckduckgo_suggestions_all.extend(suggestions)
+        duckduckgo_suggestions_all.extend(suggestions)  # Agregar todas las sugerencias
         for suggestion in suggestions:
             if suggestion in all_suggestions:
                 all_suggestions[suggestion] += 1
             else:
                 all_suggestions[suggestion] = 1
 
+    # Obtener sugerencias de Google
+    for exp_keyword in expanded_keywords:
         suggestions = fetch_google_suggestions(exp_keyword)
-        google_suggestions_all.extend(suggestions)
+        google_suggestions_all.extend(suggestions)  # Agregar todas las sugerencias
         for suggestion in suggestions:
             if suggestion in all_suggestions:
                 all_suggestions[suggestion] += 1
             else:
                 all_suggestions[suggestion] = 1
 
+    # Obtener sugerencias de YouTube
+    for exp_keyword in expanded_keywords:
         suggestions = fetch_youtube_suggestions(exp_keyword)
-        youtube_suggestions_all.extend(suggestions)
+        youtube_suggestions_all.extend(suggestions)  # Agregar todas las sugerencias
         for suggestion in suggestions:
             if suggestion in all_suggestions:
                 all_suggestions[suggestion] += 1
             else:
                 all_suggestions[suggestion] = 1
 
+    # Obtener sugerencias de Bing
+    for exp_keyword in expanded_keywords:
         suggestions = fetch_bing_suggestions(exp_keyword)
         bing_suggestions_all.extend(suggestions)
         for suggestion in suggestions:
@@ -176,6 +166,8 @@ def main(keyword):
             else:
                 all_suggestions[suggestion] = 1
 
+    # Obtener sugerencias de Amazon
+    for exp_keyword in expanded_keywords:
         suggestions = fetch_amazon_suggestions(exp_keyword)
         amazon_suggestions_all.extend(suggestions)
         for suggestion in suggestions:
@@ -184,14 +176,14 @@ def main(keyword):
             else:
                 all_suggestions[suggestion] = 1
 
-    # Filtrar las top 10 sugerencias de cada plataforma
+    # Filtrar las top 10 de cada plataforma con su número de repeticiones
     google_top_10 = list(set(google_suggestions_all))[:10]
     duckduckgo_top_10 = list(set(duckduckgo_suggestions_all))[:10]
     youtube_top_10 = list(set(youtube_suggestions_all))[:10]
     bing_top_10 = list(set(bing_suggestions_all))[:10]
     amazon_top_10 = list(set(amazon_suggestions_all))[:10]
 
-    # Ordenar y filtrar las sugerencias combinadas
+    # Ordenar y filtrar las sugerencias más frecuentes combinadas
     sorted_suggestions = sorted(all_suggestions.items(), key=lambda item: item[1], reverse=True)
     combined_top_10_suggestions = [sug for sug, freq in sorted_suggestions if freq >= 2][:10]
     suggestions_str = ", ".join(combined_top_10_suggestions)
@@ -272,4 +264,4 @@ iface = gr.Interface(
     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>"
 )
 
-iface.launch()
+iface.launch()