aiqtest3

Runtime error

aiqtest3 / app.py

Update app.py

49ea093 verified 2 months ago

No virus

1.53 kB


	from googleapiclient.discovery import build
	import gradio as gr

	# API 키와 검색 엔진 ID 설정
	API_KEY = 'AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY'
	CSE_ID = '56b34994f47704ddd'

	def multi_search(query):
	# Google Custom Search JSON API를 사용하여 웹 검색
	service = build("customsearch", "v1", developerKey=API_KEY)
	web_search_result = service.cse().list(q=query, cx=CSE_ID).execute()
	web_results = [item['link'] for item in web_search_result.get('items', [])]

	# YouTube Data API를 사용하여 비디오 검색
	youtube = build('youtube', 'v3', developerKey=API_KEY)
	video_search_result = youtube.search().list(q=query, part='snippet', type='video', maxResults=5).execute()
	video_results = [f"https://www.youtube.com/watch?v={item['id']['videoId']}" for item in video_search_result.get('items', [])]

	# 검색 결과를 튜플의 리스트로 변환
	web_results_tuples = [(url,) for url in web_results]
	video_results_tuples = [(url,) for url in video_results]

	return web_results_tuples, video_results_tuples

	iface = gr.Interface(
	fn=multi_search,
	inputs=gr.Textbox(lines=2, placeholder="검색할 텍스트를 입력하세요..."),
	outputs=[gr.List(label="웹 검색 결과"), gr.List(label="YouTube 비디오 검색 결과")],
	title="멀티 검색 결과 출력",
	description="입력 텍스트를 기준으로 웹 검색 결과와 YouTube 비디오 검색 결과를 구분하여 출력합니다."
	)

	if __name__ == "__main__":
	iface.launch()