Update app.py

app.py

@@ -13,6 +13,7 @@ import os
 import json
 import gradio as gr
 from playwright.async_api import async_playwright
+from urllib.parse import quote_plus
 
 USER_AGENT = (
     "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
@@ -20,6 +21,14 @@ USER_AGENT = (
     "Chrome/91.0.4472.124 Safari/537.36"
 )
 
+def build_url(input_str):
+    input_str = input_str.strip()
+    if input_str.startswith("http://") or input_str.startswith("https://"):
+        return input_str
+    else:
+        categoria = quote_plus(input_str)
+        return f"https://www.deviantart.com/search?q={categoria}"
+
 async def scrape_images(url, max_imgs):
     max_imgs = max(10, min(max_imgs, 300))
     async with async_playwright() as p:
@@ -29,7 +38,6 @@ async def scrape_images(url, max_imgs):
             viewport={"width": 1366, "height": 768},
         )
 
-        # Cargar cookies desde variable de entorno si está definida
         cookies_env = os.getenv("COOKIES_JSON")
         if cookies_env:
             try:
@@ -79,24 +87,28 @@ async def scrape_images(url, max_imgs):
         await browser.close()
         return collected_data[:max_imgs]
 
-def run_scraper(url, max_imgs):
+def run_scraper(user_input, max_imgs):
+    url = build_url(user_input)
+    print(f"Usando URL: {url}")
     return asyncio.run(scrape_images(url, int(max_imgs)))
 
-def interface_fn(url, max_imgs):
-    results = run_scraper(url, max_imgs)
+def interface_fn(user_input, max_imgs):
+    results = run_scraper(user_input, max_imgs)
     images = [(item["img_url"], f"Usuario: {item['user']}") for item in results]
     return images
 
 demo = gr.Interface(
     fn=interface_fn,
     inputs=[
-        gr.Textbox(label="URL de la galería DeviantArt", lines=1, value="https://www.deviantart.com/silkedead/gallery/68498591/screenshots-film-and-movie"),
+        gr.Textbox(label="URL o Categoría DeviantArt", lines=1, 
+                   placeholder="Pega una URL o escribe una categoría o usuario"),
         gr.Slider(minimum=10, maximum=300, step=1, value=30, label="Máximo de imágenes")
     ],
     outputs=gr.Gallery(label="Imágenes recolectadas"),
     title="Scraper de Imágenes - DeviantArt",
-    description="Introduce la URL de la galería DeviantArt y la cantidad máxima de imágenes que quieres recolectar."
+    description="Introduce una URL completa o solo una categoría/usuario para buscar imágenes."
 )
 
 if __name__ == "__main__":
     demo.launch()
+