wanda222 commited on
Commit
e611d39
·
verified ·
1 Parent(s): f5a3b55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -9,12 +9,14 @@ from io import BytesIO # 이미지 데이터를 메모리 상에서 처리
9
  UPSTAGE_API_KEY = os.getenv("UPSTAGE_API_KEY")
10
 
11
 
12
- # 이미지에서 텍스트를 추출하는 함수 (Upstage Document OCR API 사용)
13
  def extract_text_from_image(image):
14
- # Upstage Docuemnt Digitization Endpoint 주소
 
 
 
15
  url = "https://api.upstage.ai/v1/document-digitization"
16
 
17
- # 인증을 위한 헤더 설정
18
  headers = {'Authorization': f'Bearer {UPSTAGE_API_KEY}'}
19
 
20
  # 이미지를 메모리 버퍼에 저장 (JPEG 형식)
@@ -26,7 +28,7 @@ def extract_text_from_image(image):
26
  files = {"document": ("image.jpg", buffer, "image/jpeg")}
27
  data = {"model": "ocr"} # 사용할 모델: OCR
28
 
29
- # POST 요청 보내기 (이미지와 함께)
30
  response = requests.post(url, headers=headers, files=files, data=data)
31
 
32
  # 요청 성공 시 텍스트 추출
@@ -38,9 +40,12 @@ def extract_text_from_image(image):
38
  return f"OCR 실패: {response.status_code} - {response.text}"
39
 
40
 
41
- # 영어 텍스트를 한국어로 번역하는 함수 (Upstage Solar Pro API 사용)
42
  def translate_text_with_solar(english_text):
43
- # Solar LLM을 위한 OpenAI 클라이언트 초기화
 
 
 
44
  client = OpenAI(
45
  api_key=UPSTAGE_API_KEY,
46
  base_url="https://api.upstage.ai/v1"
 
9
  UPSTAGE_API_KEY = os.getenv("UPSTAGE_API_KEY")
10
 
11
 
 
12
  def extract_text_from_image(image):
13
+ """
14
+ 이미지에서 텍스트를 추출하는 함수 (Upstage Document OCR API 사용)
15
+ """
16
+ # Upstage API Endpoint 주소
17
  url = "https://api.upstage.ai/v1/document-digitization"
18
 
19
+ # API Key 인증을 위한 헤더 설정
20
  headers = {'Authorization': f'Bearer {UPSTAGE_API_KEY}'}
21
 
22
  # 이미지를 메모리 버퍼에 저장 (JPEG 형식)
 
28
  files = {"document": ("image.jpg", buffer, "image/jpeg")}
29
  data = {"model": "ocr"} # 사용할 모델: OCR
30
 
31
+ # POST 요청 보내기
32
  response = requests.post(url, headers=headers, files=files, data=data)
33
 
34
  # 요청 성공 시 텍스트 추출
 
40
  return f"OCR 실패: {response.status_code} - {response.text}"
41
 
42
 
43
+
44
  def translate_text_with_solar(english_text):
45
+ """
46
+ 영어 텍스트를 한국어로 번역하는 함수 (Upstage Solar Pro API 사용)
47
+ """
48
+ # Solar LLM 호출을 위한 OpenAI 클라이언트 초기화
49
  client = OpenAI(
50
  api_key=UPSTAGE_API_KEY,
51
  base_url="https://api.upstage.ai/v1"