Update app.py

app.py

@@ -63,14 +63,35 @@ def archive_now(website):
 
     archive_url = f"https://web.archive.org/save/{website}"
     try:
-        response = requests.get(archive_url, headers=headers)
+        response = requests.get(archive_url, headers=headers, timeout=60)  # Establece un tiempo límite de 60 segundos
         if response.status_code == 200:
             return f"<div>👌 URL archivada con éxito en: <a href='https://web.archive.org/save/{website}' target='_blank'>https://web.archive.org/save/{website}</a></div>"
         else:
-            return f"<div>😭 Error al archivar la URL: Código de estado {response.status_code}</div>"
+            # Si la respuesta no es exitosa, busca la última instantánea
+            return check_last_snapshot(website)
+    except requests.exceptions.Timeout:
+        # Si se supera el tiempo de espera, busca la última instantánea
+        return check_last_snapshot(website)
     except Exception as e:
         return f"<div>Error al archivar la URL: {e}</div>"
 
+def check_last_snapshot(website):
+    availability_api_url = f"http://archive.org/wayback/available?url={website}"
+    try:
+        response = requests.get(availability_api_url, headers=headers)
+        if response.status_code == 200:
+            data = json.loads(response.text)
+            if data["archived_snapshots"]:
+                closest_snapshot = data["archived_snapshots"]["closest"]
+                if closest_snapshot and closest_snapshot["available"]:
+                    snapshot_url = closest_snapshot["url"]
+                    timestamp = closest_snapshot["timestamp"]
+                    formatted_date = datetime.strptime(timestamp, '%Y%m%d%H%M%S').strftime('%d/%m/%Y')
+                    return f"<div>Última instantánea disponible: <a href='{snapshot_url}' target='_blank'>{formatted_date}</a></div>"
+    except Exception as e:
+        return f"<div>Error al buscar la última instantánea: {e}</div>"
+    return "<div>😭 No se encontraron datos archivados para esta URL.</div>"
+
 
 # Crear la interfaz de Gradio para la función wayback
 wayback_interface = gr.Interface(