import os 
api_key = os.environ.get('API KEY 이름')

from openai import OpenAI
client = OpenAI()

# 기본적인 프로그램 틀
system_message = {
  "role": "system",
  "content":"너는 변호사야, 나에게 법률적인 상담을 해줘."
}
# messages =[system_message]
# 메시지를 저장 할 공간
# 사용자 정보 추가 
messages = [system_message, {"role" : "user",
                            "content" : " 내 이름은 Polaris08"},
                            {"role" : "user",
                             "content" : "직업은 학생"},
                             {"role" : "user",
                              "content" : "컴퓨터 공학을 좋아함"}]

# 이전 대화를 바탕으로 대화가 이어지도록 반복문 작성
while True: 
  user_input = input("사용자 전달:")

  # 중단 코드
  if user_input == "exit": 
    print("대답: 즐거운 대화였습니다! 감사합니다!")
    break 

  # 질문 저장
  messages.append({"role" : "user", 
                   "content" : user_input})
  
  # 모델 설정
  completion = client.chat.completions.create(
    model = "gpt-4o",
    messages=messages
  )

  # 대답 생성
  reply = completion.choices[0].message.content
  print("대답: " + reply)

  # 대답 저장 - 이전 대화 내역을 인지하고 추가 답변
  messages.append({"role" : "assistant",
                   "content" : reply})

import os
import requests
from pydub import AudioSegment
from pydub.playback import play
import io

# api key 불러오기
api_key = os.environ.get('Eleven_Labs_API_KEY')

# 결과 파일 설정
output_filename = "output_audio.mp3"

# 음성 모델 불러오기
url = "https://api.elevenlabs.io/v1/text-to-speech/ZJCNdZEjYwkOElxugmW2/stream"
# url = "https://api.elevenlabs.io/v1/text-to-speech/{VOICE_ID}/stream"
headers = {
    "xi-api-key": api_key,
    "Content-Type": "application/json"
}

# 문장을 입력받습니다.
text = input("텍스트를 입력하세요: ")

# 음성 생성 요청을 보냅니다.
data = {
    "text": text,
    "model_id": "eleven_multilingual_v2",
    "voice_settings": {
        "stability": 0.3,
        "similarity_boost": 1,
        "style": 1,
        "use_speaker_boost": True
    }
}

response = requests.post(url, json=data, headers=headers, stream=True)

if response.status_code == 200:
    # api 요청 성공 == 200
    audio_content = b""

    # 청크 단위로 데이터 처리 - 대용량 데이터 처리 시
    for chunk in response.iter_content(chunk_size=1024):
        if chunk:
            audio_content += chunk

    segment = AudioSegment.from_mp3(io.BytesIO(audio_content))
    # 음성 데이터를 mp3 로 변환
    segment.export(output_filename, format="mp3")
    # mp3 파일로 저장
    print(f"Success! Wrote audio to {output_filename}")

    # 오디오를 재생합니다.
    play(segment)
else:
    print(f"Failed to save file: {response.status_code}")