Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
 import requests
+import pandas as pd
+from io import BytesIO
 
-# Función para realizar la solicitud a la API
-def obtener_keywords(dominio, pais, idioma):
+# Función para consultar la API y procesar datos
+def obtener_datos_y_transformar(dominio, idioma="es", pais="es"):
     url = f"https://v7.authoritas.com/api/v3/visibility-explorer/ranking/keywords/{idioma}_{pais}?domains[]={dominio}&pageSize=5000"
     headers = {
         "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/117.0.0.0 Safari/537.36",
@@ -11,26 +13,69 @@ def obtener_keywords(dominio, pais, idioma):
     }
 
     try:
-        response = requests.get(url, headers=headers, cookies={})
+        response = requests.get(url, headers=headers)
         if response.status_code == 200:
-            return response.json()  # Devuelve el resultado en formato JSON
+            json_data = response.json()
+
+            # Mapeo de columnas
+            column_mapping = {
+                "Palabra_clave": "keyword",
+                "Búsquedas_mensuales": "searchVolume",
+                "CPC": "cpc",
+                "URL": "fullUrl",
+                "Visibilidad_1": "potentialVisibilityScoreInc",
+                "Visibilidad_2": "visibilityIndex",
+                "Visibilidad_3": "visibilityIndexShare"
+            }
+
+            # Procesar datos
+            items = json_data.get("items", [])
+            if not items:
+                return None, "No se encontraron datos para el dominio proporcionado."
+
+            # Crear DataFrame con las columnas en orden
+            data = []
+            for item in items:
+                row = {key: item.get(value, 0) for key, value in column_mapping.items()}
+                data.append(row)
+
+            df = pd.DataFrame(data, columns=column_mapping.keys())
+
+            # Aplicar formato adicional (por ejemplo, palabras clave con más de 4 palabras)
+            df["Palabra_clave"] = df["Palabra_clave"].apply(
+                lambda x: f"{x} (¡Clave larga!)" if isinstance(x, str) and len(x.split()) > 4 else x
+            )
+
+            # Generar archivo Excel
+            output = BytesIO()
+            with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
+                df.to_excel(writer, index=False, sheet_name="Datos")
+                sheet = writer.sheets["Datos"]
+                sheet.set_column(0, len(column_mapping) - 1, 20)  # Ajustar ancho de columnas
+
+            output.seek(0)
+            return output, "Archivo generado exitosamente. Descárgalo abajo."
         else:
-            return {"error": f"HTTP code {response.status_code}"}
-    except requests.exceptions.RequestException as e:
-        return {"error": str(e)}
+            return None, f"Error al consultar la API: HTTP {response.status_code}"
+    except requests.RequestException as e:
+        return None, f"Error en la conexión: {str(e)}"
 
 # Interfaz de Gradio
 def interfaz():
     with gr.Blocks() as app:
-        gr.Markdown("## Consulta de Keywords en Authoritas")
-        dominio_input = gr.Textbox(label="Dominio (ejemplo.com)")
-        pais_input = gr.Textbox(label="País (ej: ES)")
-        idioma_input = gr.Textbox(label="Idioma (ej: es)")
-        output = gr.JSON(label="Resultados")
-
-        btn = gr.Button("Obtener Keywords")
-        btn.click(fn=obtener_keywords, inputs=[dominio_input, pais_input, idioma_input], outputs=output)
-    
+        gr.Markdown("## Consulta de Keywords y Exportación a Excel")
+        dominio_input = gr.Textbox(label="Dominio (ejemplo.com)", placeholder="psicologiaymente.com")
+        idioma_input = gr.Textbox(label="Idioma (ej: es)", value="es")
+        pais_input = gr.Textbox(label="País (ej: es)", value="es")
+        archivo_salida = gr.File(label="Archivo Excel", file_types=[".xlsx"])
+        mensaje_salida = gr.Textbox(label="Estado", interactive=False)
+
+        btn = gr.Button("Obtener Datos y Transformar")
+        btn.click(
+            fn=obtener_datos_y_transformar,
+            inputs=[dominio_input, idioma_input, pais_input],
+            outputs=[archivo_salida, mensaje_salida]
+        )
     return app
 
 # Ejecutar la aplicación