Update app.py

app.py

@@ -1,15 +1,10 @@
 # 필요한 라이브러리 불러오기
 import gradio as gr  # Gradio: 웹 인터페이스를 만들기 위한 라이브러리
 import requests       # API 요청을 보내기 위한 라이브러리
-import os             # 환경 변수에서 API 키를 불러오기 위해 사용
 from openai import OpenAI  # Upstage Solar LLM 사용을 위한 OpenAI 호환 클라이언트
 from io import BytesIO      # 이미지 데이터를 메모리 상에서 처리하기 위한 도구
 
-# 환경 변수에서 Upstage API 키 불러오기
-UPSTAGE_API_KEY = os.getenv("UPSTAGE_API_KEY")
-
-
-def extract_text_from_image(image):
+def extract_text_from_image(image, api_key):
     """
     이미지에서 텍스트를 추출하는 함수 (Upstage Document OCR API 사용)
     """
@@ -17,7 +12,7 @@ def extract_text_from_image(image):
     url = "https://api.upstage.ai/v1/document-digitization"  
 
     # API Key 인증을 위한 헤더 설정
-    headers = {'Authorization': f'Bearer {UPSTAGE_API_KEY}'}  
+    headers = {'Authorization': f'Bearer {api_key}'}  
 
     # 이미지를 메모리 버퍼에 저장 (JPEG 형식)
     buffer = BytesIO()
@@ -41,13 +36,13 @@ def extract_text_from_image(image):
 
 
 
-def translate_text_with_solar(english_text):
+def translate_text_with_solar(english_text, api_key):
     """
     영어 텍스트를 한국어로 번역하는 함수 (Upstage Solar Pro API 사용)
     """
     # Solar LLM 호출을 위한 OpenAI 클라이언트 초기화
     client = OpenAI(
-        api_key=UPSTAGE_API_KEY,
+        api_key=api_key,
         base_url="https://api.upstage.ai/v1"
     )
 
@@ -82,6 +77,9 @@ with gr.Blocks() as demo:
     gr.Markdown("편지 이미지를 업로드하면 Upstage Docuemnt OCR이 영어 텍스트를 추출하고,\n🌐 번역하기 버튼을 누르면 Solar LLM을 호출하여 한국어로 번역합니다!")
     gr.Markdown("예제 이미지는 GenAI를 통해 생성된 이미지이며, Files 버튼을 클릭하면 확인 및 다운로드 가능합니다.")
 
+    # ✅ API Key 입력창 추가
+    api_key_input = gr.Textbox(label="🔑 Upstage API Key", type="password", placeholder="Paste your API key here")
+
     # 레이아웃: 좌우 2단 구성
     with gr.Row():
         # 왼쪽 열: 이미지 업로드
@@ -95,10 +93,10 @@ with gr.Blocks() as demo:
             korean_box = gr.Textbox(label="🇰🇷 번역된 한국어 텍스트", lines=10)
 
     # Step 1: 이미지 업로드 시 OCR 함수 실행 → 추출된 텍스트를 영어 텍스트 박스에 표시
-    image_input.change(fn=extract_text_from_image, inputs=image_input, outputs=english_box)
+    image_input.change(fn=extract_text_from_image, inputs=[image_input, api_key_input], outputs=english_box)
 
     # Step 2: 버튼 클릭 시 번역 함수 실행 → 번역된 결과를 한국어 텍스트 박스에 표시
-    translate_button.click(fn=translate_text_with_solar, inputs=english_box, outputs=korean_box)
+    translate_button.click(fn=translate_text_with_solar, inputs=[english_box, api_key_input], outputs=korean_box)
 
 # 앱 실행
 if __name__ == "__main__":