Update app.py

app.py

@@ -130,26 +130,82 @@ async def search_arxiv(
             'error': str(e)
         }
 
-
-
-def letter_counter(word, letter):
-    """Count the occurrences of a specific letter in a word.
+async def list_retrievers(source: str = None) -> dict:
+    """
+    Devuelve la lista de retrievers disponibles. 
+    Si se especifica una source y existe, filtra por ella; si no existe, devuelve todas.
     
     Args:
-        word: The word or phrase to analyze
-        letter: The letter to count occurrences of
-        
+        source (str, optional): Fuente para filtrar. Si no existe, se ignorará. Defaults to None.
+    
     Returns:
-        The number of times the letter appears in the word
+        dict: {
+            "retrievers": Lista de retrievers (filtrados o completos),
+            "count": Número total,
+            "status": "success"|"error",
+            "source_requested": source,  # Muestra lo que se solicitó
+            "source_used": "all"|source  # Muestra lo que realmente se usó
+        }
     """
-    return word.lower().count(letter.lower())
-
-demo = gr.Interface(
-    fn=letter_counter,
-    inputs=["text", "text"],
-    outputs="number",
-    title="Letter Counter",
-    description="Count how many times a letter appears in a word"
+    try:
+        available = []
+        source_exists = source in retrievers_metadata if source else False
+        
+        for current_source, indexes in retrievers_metadata.items():
+            # Solo filtrar si el source existe, sino mostrar todo
+            if source_exists and current_source != source:
+                continue
+                
+            for index_name, metadata in indexes.items():
+                available.append({
+                    "name": index_name,
+                    "source": current_source,
+                    "title": metadata.get("title", ""),
+                    "description": metadata.get("description", "")
+                })
+        
+        return {
+            "retrievers": available,
+            "count": len(available),
+            "status": "success",
+            "source_requested": source,
+            "source_used": source if source_exists else "all"
+        }
+    except Exception as e:
+        return {
+            "retrievers": [],
+            "count": 0,
+            "status": "error",
+            "error": str(e),
+            "source_requested": source,
+            "source_used": "none"
+        }
+
+# Configuración de la interfaz Gradio
+demo = gr.TabbedInterface(
+    [
+        gr.Interface(
+            fn=search_arxiv,
+            inputs=[
+                gr.Textbox(label="Términos de búsqueda", placeholder="Ej: deep learning"),
+                gr.Slider(1, 10, value=5, step=1, label="Número máximo de resultados")
+            ],
+            outputs=gr.JSON(label="Resultados de búsqueda"),
+            title="Búsqueda en ArXiv",
+            description="Busca artículos académicos en ArXiv por palabras clave."
+        ),
+        gr.Interface(
+            fn=list_retrievers,
+            inputs=gr.Textbox(label="Fuente (opcional)", placeholder="Dejar vacío para listar todos"),
+            outputs=gr.JSON(label="Lista de retrievers"),
+            title="Lista de Retrievers",
+            description="Muestra los retrievers disponibles, opcionalmente filtrados por fuente."
+        )
+    ],
+    [
+        "Búsqueda ArXiv",
+        "Lista de Retrievers"
+    ]
 )
 
 demo.launch(mcp_server=True)