WaysAhead Global commited on
Commit
ab272ec
·
verified ·
1 Parent(s): 52e81cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -35
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
- video_links = search_youtube_videos(name)
419
  people_output.append({
420
- "name": name,
421
- "local_photos": local_photos,
422
- "web_photos": web_photos,
423
- "videos": video_links
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": all_videos,
458
  "sources": sources,
459
- "insights": [f"{name} was mentioned in the answer." for name in seen]
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}")