Update app.py

change api key input

app.py

@@ -7,20 +7,9 @@ import tempfile
 import os
 import io
 
-# API 키 불러오기 
-UPSTAGE_API_KEY = os.getenv("UPSTAGE_API_KEY")
 
-# Upstage API 클라이언트 설정 (스키마 생성용)
-schema_client = OpenAI(
-    api_key=UPSTAGE_API_KEY,
-    base_url="https://api.upstage.ai/v1/information-extraction/schema-generation"
-)
 
-# Upstage API 클라이언트 설정 (정보 추출용)
-extraction_client = OpenAI(
-    api_key=UPSTAGE_API_KEY,
-    base_url="https://api.upstage.ai/v1/information-extraction"
-)
+UPSTAGE_API_KEY = ""
 
 # 이미지 파일을 base64 문자열로 인코딩하는 함수
 def encode_img_to_base64(img_path):
@@ -46,7 +35,21 @@ def flatten_to_rows(data):
     return flat
 
 # 명함 이미지를 받아 정보 추출 및 CSV 저장을 처리하는 핵심 함수
-def extract_info(image_path):
+def extract_info(image_path, UPSTAGE_API_KEY):
+
+    
+    # Upstage API 클라이언트 설정 (스키마 생성용)
+    schema_client = OpenAI(
+        api_key=UPSTAGE_API_KEY,
+        base_url="https://api.upstage.ai/v1/information-extraction/schema-generation"
+    )
+    
+    # Upstage API 클라이언트 설정 (정보 추출용)
+    extraction_client = OpenAI(
+        api_key=UPSTAGE_API_KEY,
+        base_url="https://api.upstage.ai/v1/information-extraction"
+    )
+
     base64_data = encode_img_to_base64(image_path)
 
     # 스키마 생성
@@ -93,10 +96,14 @@ def extract_info(image_path):
     return json_result, df_result
 
 
-# 🔧 Gradio 인터페이스 구성
+
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("## 📇 명함 정보 자동 추출기")
-    gr.Markdown("명함 이미지를 업로드하면 정보를 JSON과 테이블 형식으로 추출합니다.")
+    gr.Markdown("명함 이미지를 업로드하면 정보를 추출하고, 다운로드 버튼을 눌러 CSV로 저장할 수 있습니다.")
+
+    with gr.Row():
+        api_key_input = gr.Textbox(label="🔑 Upstage API Key", type="password", placeholder="Paste your API key here")
 
     with gr.Row():
         with gr.Column():
@@ -104,16 +111,22 @@ with gr.Blocks() as demo:
             extract_btn = gr.Button("📤 정보 추출하기")
 
         with gr.Column():
-            json_output = gr.Textbox(label="📄 추출된 JSON", lines=8)
+            json_output = gr.Textbox(label="📄 추출된 JSON", lines=20)
             table_output = gr.Dataframe(label="📋 추출된 정보 (표 형태)", interactive=False)
+            download_btn = gr.Button("📥 CSV 다운로드 받기")
+            file_output = gr.File(label="", visible=False)
 
-    # 버튼 클릭 시 정보 추출 함수 실행
     extract_btn.click(
-        fn=extract_info, 
-        inputs=image_input,
-        outputs=[json_output, table_output]
+        fn=extract_info,
+        inputs=[image_input, api_key_input],
+        outputs=[json_output, table_output, file_output]
     )
 
+    download_btn.click(
+        fn=lambda x: x,
+        inputs=file_output,
+        outputs=file_output
+    )
 
 
 # 앱 실행