Update app.py

app.py

@@ -1,23 +1,18 @@
-import gradio as gr
-from datetime import datetime, timedelta
-import requests
-import json
+def wayback(website, limit=None):
+    limit = 3000 if limit is None else limit
 
-def wayback(website, days=365):
     if not website:
         return '😭 Error: introduce una url correcta'
 
-    # Calcular la fecha de inicio basada en el número de días proporcionados
     end_date = datetime.now()
-    start_date = end_date - timedelta(days=days)
+    start_date = end_date - timedelta(days=365)
     datefrom = start_date.strftime('%Y%m%d')
     dateto = end_date.strftime('%Y%m%d')
 
-    # Construir la consulta a la Wayback Machine
-    query = f"?url={website}&output=json&from={datefrom}&to={dateto}&limit=3000"
+    query = f"?url={website}&output=json&from={datefrom}&to={dateto}"
 
     try:
-        response = requests.get(f"http://web.archive.org/cdx/search/cdx{query}")
+        response = requests.get(f"http://web.archive.org/cdx/search/cdx{query}&limit={limit}")
         if response.status_code != 200:
             return f'😭 Error: Respuesta no exitosa con código de estado {response.status_code}'
         content = json.loads(response.text)
@@ -28,7 +23,6 @@ def wayback(website, days=365):
     except Exception as e:
         return f"😭 Error: {e}"
 
-    # Procesar los resultados y devolverlos
     results = []
     for row in content[1:]:
         date, page, status = [row[i] for i in [1, 2, 4]]
@@ -37,14 +31,10 @@ def wayback(website, days=365):
         results.append(f"👓 {formatted_date} {formatted_wayback_url}")
     return "\n".join(results)
 
-# Crear la interfaz de Gradio
-iface = gr.Interface(
-    fn=wayback, 
-    inputs=["text", "number"], 
-    outputs="text",
-    title="Wayback Machine Lookup",
-    description="Busca instantáneas de una página web en la Wayback Machine. Introduce la URL y el número de días hacia atrás."
-)
+iface = gr.Interface(fn=wayback, 
+                     inputs=["text", gr.Number(label="Límite", value=3000)], 
+                     outputs="text",
+                     title="Wayback Machine Lookup",
+                     description="Busca instantáneas de una página web en la Wayback Machine.")
 
-# Lanzar la aplicación
-iface.launch()
+iface.launch()