WaysAhead Global commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -90,28 +90,6 @@ def get_s3_url_by_filename(file_name: str) -> str:
|
|
| 90 |
|
| 91 |
return ""
|
| 92 |
|
| 93 |
-
from googleapiclient.discovery import build
|
| 94 |
-
|
| 95 |
-
YOUTUBE_API_KEY = "AIzaSyDOTo0hKLMj35_j7gv_uoE3UG1lxdqFAQk"
|
| 96 |
-
|
| 97 |
-
def search_youtube_videos(query: str, max_results: int = 3):
|
| 98 |
-
try:
|
| 99 |
-
youtube = build("youtube", "v3", developerKey=YOUTUBE_API_KEY)
|
| 100 |
-
request = youtube.search().list(
|
| 101 |
-
q=query,
|
| 102 |
-
part="snippet",
|
| 103 |
-
type="video",
|
| 104 |
-
maxResults=max_results
|
| 105 |
-
)
|
| 106 |
-
response = request.execute()
|
| 107 |
-
video_urls = [
|
| 108 |
-
f"https://www.youtube.com/watch?v={item['id']['videoId']}"
|
| 109 |
-
for item in response.get("items", [])
|
| 110 |
-
]
|
| 111 |
-
return video_urls
|
| 112 |
-
except Exception as e:
|
| 113 |
-
logging.warning(f"⚠️ YouTube API error for query '{query}': {e}")
|
| 114 |
-
return []
|
| 115 |
|
| 116 |
|
| 117 |
|
|
@@ -414,16 +392,13 @@ def ask_question(question):
|
|
| 414 |
seen_names.add(key)
|
| 415 |
|
| 416 |
local_photos = get_person_photos_by_name(name)
|
| 417 |
-
web_photos = search_web_images(name)
|
| 418 |
-
|
| 419 |
people_output.append({
|
| 420 |
-
|
| 421 |
-
|
| 422 |
-
|
| 423 |
-
|
| 424 |
-
})
|
| 425 |
-
|
| 426 |
-
|
| 427 |
|
| 428 |
# Get source documents with S3 URLs
|
| 429 |
docs = vector_store.similarity_search(question, k=5)
|
|
@@ -448,18 +423,17 @@ def ask_question(question):
|
|
| 448 |
insights.append(f"The most relevant document is: {sources[0]['source']}")
|
| 449 |
|
| 450 |
# Build the response
|
| 451 |
-
all_videos = [v for person in people_output for v in person.get("videos", [])]
|
| 452 |
response = {
|
| 453 |
"answer": answer,
|
| 454 |
"tabs": {
|
| 455 |
"answer": answer,
|
| 456 |
"images": people_output,
|
| 457 |
-
"videos":
|
| 458 |
"sources": sources,
|
| 459 |
-
"insights":
|
| 460 |
}
|
| 461 |
}
|
| 462 |
-
|
| 463 |
return response
|
| 464 |
except Exception as e:
|
| 465 |
logging.error(f"Failed to process question: {e}")
|
|
|
|
| 90 |
|
| 91 |
return ""
|
| 92 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
|
| 94 |
|
| 95 |
|
|
|
|
| 392 |
seen_names.add(key)
|
| 393 |
|
| 394 |
local_photos = get_person_photos_by_name(name)
|
| 395 |
+
web_photos = search_web_images(name)
|
| 396 |
+
|
| 397 |
people_output.append({
|
| 398 |
+
"name": name,
|
| 399 |
+
"local_photos": local_photos,
|
| 400 |
+
"web_photos": web_photos
|
| 401 |
+
})
|
|
|
|
|
|
|
|
|
|
| 402 |
|
| 403 |
# Get source documents with S3 URLs
|
| 404 |
docs = vector_store.similarity_search(question, k=5)
|
|
|
|
| 423 |
insights.append(f"The most relevant document is: {sources[0]['source']}")
|
| 424 |
|
| 425 |
# Build the response
|
|
|
|
| 426 |
response = {
|
| 427 |
"answer": answer,
|
| 428 |
"tabs": {
|
| 429 |
"answer": answer,
|
| 430 |
"images": people_output,
|
| 431 |
+
"videos": [],
|
| 432 |
"sources": sources,
|
| 433 |
+
"insights": insights
|
| 434 |
}
|
| 435 |
}
|
| 436 |
+
|
| 437 |
return response
|
| 438 |
except Exception as e:
|
| 439 |
logging.error(f"Failed to process question: {e}")
|