Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import requests
 import urllib.parse
-import re
 
 # Función para obtener sugerencias de DuckDuckGo
 def fetch_duckduckgo_suggestions(query, lang_code="es"):
@@ -56,7 +55,7 @@ def fetch_youtube_suggestions(query, lang_code="es"):
     else:
         return []
 
-# Función para obtener la primera sugerencia relacionada de Yahoo (sin expansiones)
+# Función para obtener solo las sugerencias correctas de Yahoo
 def fetch_yahoo_suggestion(query):
     encoded_query = urllib.parse.quote(query)
     url = f"https://search.yahoo.com/sugg/gossip/gossip-us-ura/?output=sd1&command={encoded_query}"
@@ -64,9 +63,9 @@ def fetch_yahoo_suggestion(query):
     if response.status_code == 200:
         try:
             data = response.json()
-            # Extraemos solo la primera sugerencia relacionada de la clave 'r'
-            if 'r' in data and len(data['r']) > 0:
-                return [data['r'][0]['k']]  # Solo la primera sugerencia
+            # Solo tomamos las sugerencias bajo la clave 'r' y retornamos
+            if 'r' in data:
+                return [item['k'] for item in data['r']]  # Devolvemos todas las sugerencias bajo 'r'
             else:
                 return []
         except ValueError:
@@ -122,11 +121,11 @@ def main(keyword):
             else:
                 all_suggestions[suggestion] = 1
 
-    # Obtener la primera sugerencia relacionada de Yahoo
+    # Obtener solo las sugerencias correctas de Yahoo
     for exp_keyword in expanded_keywords:
-        suggestion = fetch_yahoo_suggestion(exp_keyword)
-        yahoo_suggestions_all.extend(suggestion)
-        for suggestion in suggestion:
+        suggestions = fetch_yahoo_suggestion(exp_keyword)
+        yahoo_suggestions_all.extend(suggestions)
+        for suggestion in suggestions:
             if suggestion in all_suggestions:
                 all_suggestions[suggestion] += 1
             else: