Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
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 |
-
|
42 |
def translate_text_with_solar(english_text):
|
43 |
-
|
|
|
|
|
|
|
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"
|