o |iQ@sdZddlZddlZddlZddlZddlmZddlmZddlZddl Z ddl m Z m Z m Z mZddlmZddlZddlmZdejd <Gd d d Zd d Zedkr[edSdS)u YouTube 동영상을 AI 기반으로 스마트 요약하는 프로그램 - Whisper로 음성 인식 - 중요 구간 자동 추출 - ChatGPT API로 요약 생성 Author: 최만규 N)Path)datetime) VideoFileClipconcatenate_videoclipsTextClipCompositeVideoClip)OpenAI)CounterdummySDL_AUDIODRIVERc@szeZdZdZdddZdddZd d Zd d d ZddZd!ddZ d"ddZ ddZ ddZ ddZ   d#ddZdS)$SmartShortsConverteru.AI 기반 스마트 동영상 요약 클래스./smart_shorts_outputNcCspt||_|jjdd|jd|_|jjdd|ptd|_|jr,t|jd|_nt dd|_d|_ dS)u 초기화 Args: output_dir (str): 출력 디렉토리 openai_api_key (str): OpenAI API 키 T)exist_oktempOPENAI_API_KEY)api_keyud⚠️ OpenAI API 키가 설정되지 않았습니다. 요약 기능은 사용할 수 없습니다.N) r output_dirmkdirtemp_dirosgetenvopenai_api_keyrclientprint whisper_model)selfrrr2/var/www/html/bicorn/ndegree/smart_shorts_maker.py__init__s   zSmartShortsConverter.__init__basecCs4|jdurtd|dt||_td|jS)u Whisper 모델 로드 Args: model_size (str): 모델 크기 (tiny, base, small, medium, large) Nu 🤖 Whisper 모델 로딩 중 (z)...u ✅ Whisper 모델 로드 완료)rrwhisper load_model)r model_sizerrrload_whisper_model7s  z'SmartShortsConverter.load_whisper_modelc Csrtd|dt|jdddddddgdd d id d d gddgdid }z{t|k}td|j|dd}|d}|d}gd}d}|D]} |j|d| } | r]| }nqI|srt|jd} | rrt | ddd}|rx|s|t dtd|td|t||fWdWS1swYWdSt y} z td t| d} ~ ww)!u 유튜브 동영상 다운로드u+📥 유튜브 동영상 다운로드 중: zbestvideo[ext=mp4][height<=1080]+bestaudio[ext=m4a]/best[ext=mp4][height<=1080]/bestvideo[ext=mp4]+bestaudio/best[ext=mp4]/bestvideo+bestaudio/bestz%(id)s.%(ext)sFmp4FFmpegVideoConvertor)keypreferedformatTz User-Agentzws z=SmartShortsConverter.download_youtube_video..r&u2다운로드된 파일을 찾을 수 없습니다.u✅ 다운로드 완료: u 📁 파일: u❌ 다운로드 실패: ) rstrryt_dlp YoutubeDL extract_infoexistslistglobmax Exception) rurlydl_optsydlinfovideo_id video_titlepossible_extensionsdownloaded_fileext candidatefileserrrdownload_youtube_videoDs^    (z+SmartShortsConverter.download_youtube_videokocCstd|d}td|j||ddd}tdt|dd |jd }t|d d d }tj||dddWdn1sBwYtd||S)u" 동영상 음성을 텍스트로 변환 (Whisper) Args: video_path (str): 동영상 파일 경로 language (str): 언어 코드 (ko, en 등) Returns: dict: 타임스탬프가 포함된 전사 결과 u+ 🎙️ 음성 인식 시작 (Whisper)...ru6📝 전사 중... (시간이 걸릴 수 있습니다) transcribeF)languagetaskverboseu✅ 전사 완료: segments개 세그먼트ztranscript.jsonwutf-8encoding) ensure_asciiindentNu💾 전사 내용 저장: )rr#r\lenropenjsondump)r video_pathr]modelresulttranscript_filefrrrtranscribe_videos   z%SmartShortsConverter.transcribe_videoc s"td|d}g}gd}|D]\}|d|d}|d}||} d} tfdd |D} | | d 7} d tkrCd krInn| d 7} | d krQ| d7} ||} | dks]| dkra| d 7} |||| | dq|jddddtdt|dtd|r|ddnd|S)u8 전사 내용을 분석하여 중요한 구간 추출 Args: transcription (dict): Whisper 전사 결과 video_duration (float): 동영상 총 길이 Returns: list: 중요 구간 리스트 [(start, end, text, score), ...] u! 🧠 중요 구간 분석 중...r`)!u중요u핵심u 포인트u주목u강조u정리u요약u결론u첫째u둘째u셋째u먼저u다음u 마지막u이것u이거u바로u특히u 예를 들어u예시u문제u해결u방법u기법u원리u개념u정의u질문u답u이유u 왜냐하면u 그래서u 따라서textstartendrc3s|] }|vrdVqdS)Nr).0kwrsrr szBSmartShortsConverter.analyze_important_segments.. rfrvg??)rtrursscoredurationcS|dS)NrrrArrrrCzASmartShortsConverter.analyze_important_segments..T)r&reverseu📊 분석 완료: 총 rau 상위 점수: r)rstripsumriappendsort) r transcriptionvideo_durationr`scored_segmentsimportant_keywordssegrtrurr keyword_countposition_ratiorryranalyze_important_segmentss<     z/SmartShortsConverter.analyze_important_segments<cCstd|dg}d}|D]}||d|kr#||||d7}||dkr+nq|jdddtd t|d |d d |S) u! 목표 시간에 맞게 세그먼트 선택 Args: scored_segments (list): 점수가 매겨진 세그먼트 target_duration (int): 목표 시간 (초) Returns: list: 선택된 세그먼트 리스트 u 🎯 u초 분량 선택 중...rrr~cSr)NrtrrArrrrCrzJSmartShortsConverter.select_segments_for_target_duration..rDu✅ u개 구간 선택됨 (총 .1fu초))rrrri)rrtarget_durationselectedtotal_durationrrrr#select_segments_for_target_durations    z8SmartShortsConverter.select_segments_for_target_durationFc Cs.tdt|}g}t|D]\}}td|ddt|d|ddd|d dd ||d|d }|r|d rz7t|d } td td d| dd} t|d dd| dddddddjddd |j } t || g}Wnt y} ztdt | WYd} ~ nd} ~ ww||qtdt|dd} td |j\}}d!}d"}||}|||kr| j|d#}n| j|d$}|j\}}|j|d|d||d%}td&}d'|d(}|j|}td)|j|jt |d*d+d d,d-d.dd/|| ||td0t |S)1uB 선택된 구간으로 쇼츠 동영상 생성 Args: video_path (str): 원본 동영상 경로 selected_segments (list): 선택된 구간 add_subtitles (bool): 자막 추가 여부 Returns: str: 생성된 쇼츠 경로 u$ 🎬 쇼츠 동영상 생성 중...u 구간 rv/z: rtrs ~ rusrs2i,r|NdwhiteArialblackrfcaption)iN)fontsizecolorfont stroke_color stroke_widthmethodsize)centerg333333?T)relativeu" ⚠️ 자막 추가 실패: u 🔗 클립 연결 중...compose)ru&📐 쇼츠 포맷으로 변환 중...i8i)height)width)x_centery_centerrrz %Y%m%d_%H%M%S smart_shorts_.mp4u 💾 인코딩 중: libx264aacmedium5000kF)codec audio_codecfpspresetbitrater_loggeru✅ 쇼츠 생성 완료!)rr enumeraterisubcliprLminr set_position set_durationrrrMrErrrresizecroprnowstrftimernamewrite_videofileclose)rrmselected_segments add_subtitlesvideoclipsirclip text_lengthrsubtitlerY final_videooriginal_widthoriginal_height shorts_width shorts_heightoriginal_ratioresized resized_widthresized_heightcropped timestampoutput_filename output_pathrrrcreate_smart_shorts s 6              z(SmartShortsConverter.create_smart_shortsc Cs|js tddStdddd|dD}t|dkr'|d dd }z.|jjjjd d d ddd|dgddd}|jdjj }tdtd||WSt yw}ztdt |dt |WYd }~Sd }~ww)u ChatGPT API로 동영상 내용 요약 Args: transcription (dict): Whisper 전사 결과 Returns: str: 1-2줄 요약 uF⚠️ OpenAI API 키가 없어 요약을 생성할 수 없습니다.u%요약 생성 불가 (API 키 필요)u% 🤖 ChatGPT로 요약 생성 중... cSsg|]}|dqSryr)rwrrrr szBSmartShortsConverter.generate_summary_with_gpt..r`i Nz...z gpt-4o-minisystemu당신은 강의 동영상을 요약하는 전문가입니다. 주어진 전사 내용을 1-2줄로 핵심만 간결하게 요약해주세요.)rolecontentuseru7다음 강의 내용을 1-2줄로 요약해주세요: r}gffffff?)rnmessages max_tokens temperatureru✅ 요약 생성 완료 📝 요약: u❌ 요약 생성 실패: u요약 생성 실패: ) rrjoinrichat completionscreatechoicesmessagerrrMrE)rr full_textresponsesummaryrYrrrgenerate_summary_with_gptns:    z.SmartShortsConverter.generate_summary_with_gptc Cst|d}t|dddg}|d|d|d|d||d |d|d t|d |dt|d D](\}}||d |ddd|ddd|dd|d|dd qEWdn1sxwYtd|t|S)u 요약 정보를 텍스트 파일로 저장 Args: summary (str): ChatGPT 요약 selected_segments (list): 선택된 구간 output_path (str): 쇼츠 파일 경로 z.txtrbrcrdz============================================================= u$📹 스마트 쇼츠 요약 정보 z>============================================================ u📝 AI 요약 (ChatGPT): z u📊 포함된 구간 (u개): rvz. [rtrrruu s] (점수: rz) z rsNu💾 요약 파일 저장: )r with_suffixrjwriterirrrE)rrrr summary_filerqrrrrrsave_summary_files"       2z&SmartShortsConverter.save_summary_filec Cs|tdz|jdD]}|jdvr|td|jq WdSty=}ztdt|WYd}~dSd}~ww)u임시 파일 정리u! 🧹 임시 파일 정리 중...r>)rz.webmz.mkvu 삭제: u⚠️ 정리 중 오류: N)rrrKsuffixunlinkrrMrE)rfilerYrrrcleanup_temp_filess  z'SmartShortsConverter.cleanup_temp_filesc Cstdtdtdzd||\}}|j||d}t|} | j} | ||| } || |} ||| |} | |}| || | }|sK| tdtdtdtd| td|td|| |fWSt y}z td t |d }~ww) u 유튜브 URL을 AI 기반 스마트 쇼츠로 변환 (전체 프로세스) Args: youtube_url (str): 유튜브 URL target_duration (int): 목표 길이 (초) add_subtitles (bool): 자막 추가 여부 keep_temp (bool): 임시 파일 유지 여부 language (str): 음성 언어 (ko, en 등) Returns: tuple: (쇼츠 경로, 요약 텍스트) z<============================================================u)🧠 AI 스마트 쇼츠 메이커 시작)r]z= ============================================================u🎉 모든 작업 완료!u 📹 쇼츠: ru💬 한줄 요약: u ❌ 작업 실패: N)rrZrrrrrrrrrrrrMrE)r youtube_urlrr keep_tempr]rmrSrrrrr shorts_pathrrrYrrrconvert_url_to_smart_shortss@   z0SmartShortsConverter.convert_url_to_smart_shorts)r N)r)r[)r)F)rFFr[)__name__ __module__ __qualname____doc__rr#rZrrrrrrrrrrrrrr s    A" E c2 r c Cstjdtjdd}|jddd|jddtd d d |jd d ddd|jddd|jdddd|jdddd|jdddd|}t|j|jd}z|j |j |j |j |j |jd\}}tdWd!Sty}ztdt|td WYd!}~d!Sd!}~ww)"u 메인 함수u8AI 기반 스마트 쇼츠 메이커 - Whisper + ChatGPTu 사용 예시: python smart_shorts_maker.py "URL" --api-key "sk-..." python smart_shorts_maker.py "URL" -d 60 --subtitles 환경 변수: export OPENAI_API_KEY="sk-..." ) descriptionformatter_classepilogrNu 유튜브 URL)helpz-dz --durationru"목표 길이 (초, 기본값: 60))typedefaultrz-oz--outputr u출력 디렉토리)rrz --api-keyuOpenAI API 키z --subtitles store_trueu 자막 추가)actionrz --keep-tempu임시 파일 유지z --languager[u음성 언어 (기본값: ko))rr)rrrrr]u ✨ 완료!u 💥 오류: rvN)argparseArgumentParserRawDescriptionHelpFormatter add_argumentint parse_argsr outputrrrNr subtitlesrr]rrMrEsysexit)parserargs converterrrrYrrrmainsR   r__main__)rrrrrkpathlibrrrFr moviepy.editorrrrropenairnumpynp collectionsr environr rrrrrrs,     u4