Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import requests
+import nbformat
 
 # Ruta del archivo base en el disco C:
 ruta_archivo_base = "blocked_urls.txt"
@@ -8,32 +10,51 @@ salida_coincidencias = ""
 salida_fragmentos = ""
 
 # Función para buscar coincidencias en el archivo de usuario
-def buscar_coincidencias(archivo_usuario):
+def buscar_coincidencias(archivo_usuario, url_colab):
     global salida_coincidencias, salida_fragmentos
     coincidencias = []
     fragmentos_coincidentes = []
     
     # Verificar si archivo_usuario es None
-    if archivo_usuario is None:
-        return "No se ha subido ningún archivo.", ""
-    
-    # Obtener la ruta local del archivo de usuario
-    ruta_archivo_usuario = archivo_usuario.name
+    if archivo_usuario is None and not url_colab:
+        return "No se ha subido ningún archivo ni se ha proporcionado ninguna URL de Colab.", ""
     
     # Leer el contenido del archivo base desde la ruta local (C:)
     with open(ruta_archivo_base, 'r', encoding='utf-8') as base_file:
         lineas_base = base_file.readlines()
     
-    # Leer el contenido del archivo de usuario desde la ruta local temporal
-    with open(ruta_archivo_usuario, 'r', encoding='utf-8') as usuario_file:
-        lineas_usuario = usuario_file.readlines()
+    # Si se proporcionó una URL de Colab, descargar y leer el cuaderno de Colab
+    if url_colab:
+        # Descargar el cuaderno de Colab
+        response = requests.get(url_colab)
+        response.raise_for_status()
+
+        # Leer el cuaderno de Colab
+        notebook = nbformat.reads(response.text, as_version=4)
+
+        # Buscar coincidencias en las celdas de código del cuaderno
+        for num_celda, celda in enumerate(notebook.cells, start=1):
+            if celda.cell_type == "code":
+                for num_linea_base, linea_base in enumerate(lineas_base, start=1):
+                    if linea_base.strip() in celda.source:
+                        coincidencias.append(f"Coincidencia en la Celda {num_celda} del Cuaderno de Colab:\n{celda.source.strip()}")
+                        fragmentos_coincidentes.append(f"Línea {num_linea_base} (Base):\n{linea_base.strip()}")
     
-    # Buscar coincidencias en las líneas
-    for num_linea_usuario, linea_usuario in enumerate(lineas_usuario, start=1):
-        for num_linea_base, linea_base in enumerate(lineas_base, start=1):
-            if linea_base.strip() in linea_usuario:
-                coincidencias.append(f"Coincidencia en Línea {num_linea_usuario}:\n{linea_usuario.strip()}")
-                fragmentos_coincidentes.append(f"Línea {num_linea_base} (Base):\n{linea_base.strip()}")
+    # Si se subió un archivo de usuario, leer y analizar el archivo
+    if archivo_usuario is not None:
+        # Obtener la ruta local del archivo de usuario
+        ruta_archivo_usuario = archivo_usuario.name
+
+        # Leer el contenido del archivo de usuario desde la ruta local temporal
+        with open(ruta_archivo_usuario, 'r', encoding='utf-8') as usuario_file:
+            lineas_usuario = usuario_file.readlines()
+        
+        # Buscar coincidencias en las líneas del archivo
+        for num_linea_usuario, linea_usuario in enumerate(lineas_usuario, start=1):
+            for num_linea_base, linea_base in enumerate(lineas_base, start=1):
+                if linea_base.strip() in linea_usuario:
+                    coincidencias.append(f"Coincidencia en Línea {num_linea_usuario} del Archivo Subido:\n{linea_usuario.strip()}")
+                    fragmentos_coincidentes.append(f"Línea {num_linea_base} (Base):\n{linea_base.strip()}")
     
     if not coincidencias:
         coincidencias.append("No se encontraron coincidencias.")
@@ -48,7 +69,8 @@ def buscar_coincidencias(archivo_usuario):
 interfaz = gr.Interface(
     fn=buscar_coincidencias,
     inputs=[
-        gr.File(label="Archivo de Usuario (cualquier formato o .py .ipynb)", type="file")
+        gr.File(label="Archivo de Usuario (cualquier formato o .py .ipynb)", type="file"),
+        gr.Textbox(label="URL del Cuaderno de Colab", type="url")
     ],
     outputs=[
         gr.Textbox(label="Código Prohibido", type="text"), 
@@ -56,7 +78,7 @@ interfaz = gr.Interface(
     ],
     live=True,
     title="Anti alert Colab SD",
-    description=f"Para usar este espacio, revise el archivo README en Files. Cargue el archivo de su Colab en formato .py o .ipynb para buscar palabras prohibidas por Google en las líneas.",    
+    description=f"Cargue el archivo de su Colab en formato .py o .ipynb o proporcione la URL del cuaderno de Colab para buscar palabras prohibidas por Google en las líneas. Asegúrese de que su cuaderno de Colab esté configurado para ser accesible públicamente.",    
     theme="NoCrypt/miku@>=1.2.2,<2.0.0",
     css=".flag-output {display: none !important;}"
 )