o \h@sddlmZmZmZmZddlmZddlZddlmZddl Z ddl Z ddl Z ddl Z ddl mZmZmZmZmZmZddlmZmZddlZddlmZddlZddlZddlmZddlZdd l m!Z!dd l"m#Z#dd l$m%Z%m&Z&ddl Z ddl'Z'dd l(m(Z(ddl)Z)dd l*m+Z+e)j,dde+dddZ-e-.dZ/eZ0ee_1eedZ2e 2dZ3dZ4ddZ5ddZ6ddZ7ddZ8e09dd d!Z:e0;d"d#efd$d%Z/szask_openai..gpt-4omodelmessagesr)openaichat completionscreatechoicesmessager!strip)r+valid_messagesresponseanswerr%r%r& ask_openai.sr6cstjjjdd|d}d|dtd}t|d}|D]}||q Wdn1s2wY|dtd}t |t |t |dt d |S) Nztts-1nova)r*voiceinputz tmp/audio--z.mp3wbzhttps://z.s3.amazonaws.com/) clientaudiospeechr/timeopen iter_byteswrite s3_client upload_file bucket_nameosremove)textuser_idr4 audio_filefchunks3_keyr%r%r&synthesize_speech8s     rNcsHtjjd|d}|jdj}tj|gdd}|dd}d|}|S)Nztext-embedding-3-small)r*r9r)query_embeddings n_results documentsz )r< embeddingsr/data embeddingcollection_profilequeryjoin)questionr4query_embeddingresultsretrieved_docsprofile_contextr%r%r&search_profile_contextIs   r^c s|ds d|}ttdt|IdH}td|ttdddddd ddd ddd ddd ddd ddddg}|rP|dd|d|d|dttdtjjj d|d}ttd|j dj j }td||S)uwGPT-4o를 사용해 벡터DB의 웹사이트 데이터, 회사제공 엑셀 데이터 정보를 포함한 답변 생성u 御社のu크롤링시작Nzprofile_data : u 크롤링끝systemuuあなたは「セグエグループ」の会社紹介および就職希望者向けのチャットボットです。)roler!uK以下のルールに従って、ユーザーと会話してください。u1.会話スタイル：SNS上のやり取りのように、親しみやすく、1回の返信は300文字以内に収めてください。u2.トーン：丁寧で共感的な言葉遣いを心がけ、ユーザーとの会話を楽しむように、適度に質問を返してください。u3.情報の制限：返答に使用できるのは、提供されたURLとExcelファイルに含まれる情報のみとします。その他の情報は使用しないでください。uq4.数値データ形式：数値がある場合は、マークダウン形式の表で表示してください。u]この制約のもと、ユーザーに寄り添いながら会話を進めてください。u企業サイト情報 useru대답생성 시작r(r)u대답생성 종료ru 대답 : ) startswithprintrnowr^appendr<r-r.r/r0r1r!)rY profile_datar+r4r5r%r%r&generate_gpt_answerks6    rgz/healthcs ddiS)Nstatushealthyr%r%r%r%r& health_checksrjz/apige2/speech speech_textcs6|j}|j}|stdddt||IdH}d|iS)NzText is required status_codedetailrJ)rH chat_tokenrrN)rkrHrprJr%r%r&r>s r>z/apige2/ask_question user_questioncs2|j}|stdddt|IdH}d|iS)NrlzQuestion is requiredrmr5)rYr2rrg)rq question_textgenerated_answerr%r%r&gemini_questions   rt)>fastapirrrrfastapi.responsesrr,rreboto3rFr?schemasrr r r r r configrrjsonsklearn.feature_extraction.textrloggingrequestsbs4rnumpynpseleniumr!selenium.webdriver.chrome.optionsr urllib.parserrastrchromadbchromadb.configr HttpClient chroma_clientget_collectionrVrouterrr<rCrEr6rNr^rgr"rjpostr>rtr%r%r%r&sZ             "1