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/e-.dZ0eZ1ee_2eedZ3e 3dZ4dZ5ddZ6ddZ7ddZ8ddZ9d d!Z:e1;d"d#d$Ze1=d)d*efd+d,Z?dS)-) APIRouter HTTPExceptionRequestDepends)RedirectResponseN)OpenAI) ChatMessage SpeechText QuizMessageLineUser MedicineText UserQuestion)loggeropenai_api_key)TfidfVectorizer) BeautifulSoup) webdriver)Options)urljoinurlparse)datetime)Settings localhostiArest)chroma_api_impl)hostportsettings sanwa_profile excel_honbun)api_keys3z$shanri-ai-chatbot-for-text-to-speechcs8dd|D}tjjjd|d}|jdjj}|S)NcSsg|] }|dr|qS)content)get).0msgr&//home/air/sanwanet/backup_V2/chatbot_router4.py /szask_openai..gpt-4omodelmessagesr)openaichat completionscreatechoicesmessager"strip)r,valid_messagesresponseanswerr&r&r' ask_openai.sr7cstjjjdd|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/) clientaudiospeechr0timeopen iter_byteswrite s3_client upload_file bucket_nameosremove)textuser_idr5 audio_filefchunks3_keyr&r&r'synthesize_speech8s     rOcHtjjd|d}|jdj}tj|gdd}|dd}d|}|SNztext-embedding-3-small)r+r:r)query_embeddings n_results documentsz )r= embeddingsr0data embeddingcollection_profilequeryjoin)questionr5query_embeddingresultsretrieved_docsprofile_contextr&r&r'search_profile_contextI   racrPrQ)r=rVr0rWrXcollection_excelrZr[)r\r5r]r^r_ excel_contextr&r&r'search_excel_contextZrbrecsD|ds d|}ttdt|IdH}td|ttdttdt|IdH}td|ttdd d d d d d d d d d dd d dd d dd g}|rg|d d|d |rt|d d|d |d|d ttdtjj j d|d}ttd|j dj j }td||S)uwGPT-4o를 사용해 벡터DB의 웹사이트 데이터, 회사제공 엑셀 데이터 정보를 포함한 답변 생성u 御社のu크롤링시작Nzprofile_data : u 크롤링끝uExcel 검색 시작z excel_data : uExcel 검색 끝systemuxあなたは「三和電業グループ」の会社紹介および就職希望者向けのチャットボットです。)roler"uK以下のルールに従って、ユーザーと会話してください。u1.会話スタイル：SNS上のやり取りのように、親しみやすく、1回の返信は300文字以内に収めてください。u2.トーン：丁寧で共感的な言葉遣いを心がけ、ユーザーとの会話を楽しむように、適度に質問を返してください。u3.情報の制限：返答に使用できるのは、提供されたURLとExcelファイルに含まれる情報のみとします。その他の情報は使用しないでください。u]この制約のもと、ユーザーに寄り添いながら会話を進めてください。u企業サイト情報 u0Excelファイルに記載されている情報 useru대답생성 시작r)r*u대답생성 종료ru 대답 : ) startswithprintrnowrareappendr=r.r/r0r1r2r")r\ profile_data excel_datar,r5r6r&r&r'generate_gpt_answerks@     roz/healthcs ddiS)Nstatushealthyr&r&r&r&r' health_checksrrz/sanwa/gpt-v2-backup/speech speech_textcs6|j}|j}|stdddt||IdH}d|iS)NuText is required。 status_codedetailrK)rI chat_tokenrrO)rsrIrxrKr&r&r'r?s r?z!/sanwa/gpt-v2-backup/ask_question user_questioncs>|j}|stdddttt|IdH}d|iS)NrtzQuestion is requiredrur6)r\r3rrj chroma_clientlist_collectionsro)ry question_textgenerated_answerr&r&r' ask_questions   r~)@fastapirrrrfastapi.responsesrr-rreboto3rGr@schemasrr r r r r configrrjsonsklearn.feature_extraction.textrloggingrequestsbs4rnumpynpseleniumr!selenium.webdriver.chrome.optionsr urllib.parserrastrchromadbchromadb.configr HttpClientrzget_collectionrYrcrouterr r=rDrFr7rOrareror#rrpostr?r~r&r&r&r's^              1