Spaces:
Building
Building
Update app.py
Browse files
app.py
CHANGED
@@ -38,37 +38,38 @@ def cleanup_temp_files():
|
|
38 |
print(f"Error deleting {file_path}: {e}")
|
39 |
|
40 |
def search_google_images(query, num_images=1):
|
41 |
-
"""Search for
|
42 |
try:
|
43 |
api_key = os.getenv('GOOGLE_API_KEY')
|
44 |
cse_id = os.getenv('GOOGLE_CSE_ID')
|
45 |
|
46 |
-
|
|
|
47 |
print(f"CSE ID configurado: {cse_id}")
|
48 |
print(f"API Key configurada: {api_key[:5]}...{api_key[-5:] if api_key else ''}")
|
49 |
|
50 |
service = build("customsearch", "v1", developerKey=api_key)
|
51 |
|
52 |
-
# Realizar la búsqueda
|
53 |
result = service.cse().list(
|
54 |
q=query,
|
55 |
cx=cse_id,
|
56 |
searchType="image",
|
57 |
num=num_images,
|
58 |
-
imgSize="XLARGE", # Buscar imágenes de tamaño extra grande
|
59 |
safe='off' # Permitir todos los resultados
|
60 |
).execute()
|
61 |
|
|
|
62 |
if 'items' in result:
|
63 |
image_urls = [item['link'] for item in result.get('items', [])]
|
64 |
-
print(f"Encontradas {len(image_urls)} imágenes
|
65 |
return image_urls
|
66 |
else:
|
67 |
-
print("No se encontraron imágenes
|
68 |
return []
|
69 |
|
70 |
except Exception as e:
|
71 |
-
print(f"Error detallado en la búsqueda de imágenes
|
72 |
return []
|
73 |
|
74 |
def download_image(url):
|
@@ -94,7 +95,7 @@ def concatenate_google_images(keywords, clip_duration=5, num_images_per_keyword=
|
|
94 |
video_clips = []
|
95 |
for keyword in keyword_list:
|
96 |
try:
|
97 |
-
print(f"Searching
|
98 |
image_urls = search_google_images(keyword, num_images=num_images_per_keyword)
|
99 |
for url in image_urls:
|
100 |
image = download_image(url)
|
|
|
38 |
print(f"Error deleting {file_path}: {e}")
|
39 |
|
40 |
def search_google_images(query, num_images=1):
|
41 |
+
"""Search for images using Google Custom Search API."""
|
42 |
try:
|
43 |
api_key = os.getenv('GOOGLE_API_KEY')
|
44 |
cse_id = os.getenv('GOOGLE_CSE_ID')
|
45 |
|
46 |
+
# Imprimir información de depuración (los ... son para no mostrar la key completa)
|
47 |
+
print(f"Buscando imágenes para: {query}")
|
48 |
print(f"CSE ID configurado: {cse_id}")
|
49 |
print(f"API Key configurada: {api_key[:5]}...{api_key[-5:] if api_key else ''}")
|
50 |
|
51 |
service = build("customsearch", "v1", developerKey=api_key)
|
52 |
|
53 |
+
# Realizar la búsqueda
|
54 |
result = service.cse().list(
|
55 |
q=query,
|
56 |
cx=cse_id,
|
57 |
searchType="image",
|
58 |
num=num_images,
|
|
|
59 |
safe='off' # Permitir todos los resultados
|
60 |
).execute()
|
61 |
|
62 |
+
# Verificar si hay resultados
|
63 |
if 'items' in result:
|
64 |
image_urls = [item['link'] for item in result.get('items', [])]
|
65 |
+
print(f"Encontradas {len(image_urls)} imágenes")
|
66 |
return image_urls
|
67 |
else:
|
68 |
+
print("No se encontraron imágenes")
|
69 |
return []
|
70 |
|
71 |
except Exception as e:
|
72 |
+
print(f"Error detallado en la búsqueda de imágenes: {str(e)}")
|
73 |
return []
|
74 |
|
75 |
def download_image(url):
|
|
|
95 |
video_clips = []
|
96 |
for keyword in keyword_list:
|
97 |
try:
|
98 |
+
print(f"Searching images for keyword '{keyword}'...")
|
99 |
image_urls = search_google_images(keyword, num_images=num_images_per_keyword)
|
100 |
for url in image_urls:
|
101 |
image = download_image(url)
|