Update app.py

app.py

@@ -3,17 +3,20 @@ from google import genai
 from google.genai import types
 import os
 
-# Gemini API 클라이언트 초기화
-client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
-
-def correct_grammar(text):
+def correct_grammar(text, api_key):
     """
     입력된 텍스트의 문법을 자연스럽게 교정하는 함수
     """
+    if not api_key or not api_key.strip():
+        return "❌ API 키를 입력해주세요."
+    
     if not text.strip():
-        return "텍스트를 입력해주세요."
+        return "❌ 텍스트를 입력해주세요."
     
     try:
+        # API 키로 클라이언트 초기화
+        client = genai.Client(api_key=api_key.strip())
+        
         # 시스템 지시문과 함께 API 호출
         response = client.models.generate_content(
             model="gemini-2.5-flash",
@@ -22,8 +25,8 @@ def correct_grammar(text):
                 사용자가 입력한 문장의 문법, 맞춤법, 어순을 자연스럽게 교정해주세요.
                 교정된 문장만 출력하고, 별도의 설명은 하지 마세요.
                 만약 문장이 이미 완벽하다면 '교정할 내용이 없습니다.'라고 답변하세요.""",
-                temperature=0.3,  # 일관된 교정을 위해 낮은 temperature 설정
-                thinking_config=types.ThinkingConfig(thinking_budget=0)  # 빠른 응답을 위해 thinking 비활성화
+                temperature=0.3,
+                thinking_config=types.ThinkingConfig(thinking_budget=0)
             ),
             contents=f"다음 문장을 교정해주세요: {text}"
         )
@@ -31,39 +34,57 @@ def correct_grammar(text):
         return response.text
     
     except Exception as e:
-        return f"오류가 발생했습니다: {str(e)}"
+        return f"❌ 오류가 발생했습니다: {str(e)}"
 
 # Gradio 인터페이스 생성
-with gr.Blocks(title="한국어 문법 교정기") as demo:
+with gr.Blocks(title="한국어 문법 교정기", theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         # 🔤 한국어 문법 교정기
         **Gemini 2.5 Flash** 모델을 사용하여 한국어 문장을 자연스럽게 교정합니다.
+        
+        ### 📌 사용 방법
+        1. [Google AI Studio](https://aistudio.google.com/app/apikey)에서 API 키를 발급받으세요
+        2. 아래에 API 키를 입력하세요
+        3. 교정할 문장을 입력하고 "교정하기" 버튼을 클릭하세요
         """
     )
     
+    with gr.Row():
+        api_key_input = gr.Textbox(
+            label="🔑 Gemini API Key",
+            placeholder="여기에 API 키를 입력하세요...",
+            type="password",
+            info="API 키는 저장되지 않으며 현재 세션에서만 사용됩니다."
+        )
+    
     with gr.Row():
         with gr.Column():
             input_text = gr.Textbox(
                 label="원본 문장",
                 placeholder="교정할 문장을 입력하세요...",
-                lines=5
+                lines=6
             )
-            submit_btn = gr.Button("교정하기", variant="primary")
+            submit_btn = gr.Button("✨ 교정하기", variant="primary", size="lg")
+            clear_btn = gr.Button("🗑️ 지우기", variant="secondary")
             
         with gr.Column():
             output_text = gr.Textbox(
                 label="교정된 문장",
-                lines=5
+                lines=6,
+                show_copy_button=True
             )
     
+    gr.Markdown("### 📝 예시 문장들")
     gr.Examples(
         examples=[
             ["나는 어제 학교에 갔다왔어요"],
             ["저는 밥을 먹고있어요"],
             ["그는 매우 빨리 달릴수있다"],
             ["이것은 정말 좋은 방법인것같아"],
-            ["안녕하세요 저는 AI를 공부하고있는 학생입니다"]
+            ["안녕하세요 저는 AI를 공부하고있는 학생입니다"],
+            ["오늘 날씨가 너무 좋은것 같애요"],
+            ["저희 회사는 AI기술을 개발하고 있습니다"]
         ],
         inputs=input_text
     )
@@ -71,16 +92,32 @@ with gr.Blocks(title="한국어 문법 교정기") as demo:
     # 버튼 클릭 이벤트
     submit_btn.click(
         fn=correct_grammar,
-        inputs=input_text,
+        inputs=[input_text, api_key_input],
         outputs=output_text
     )
     
     # Enter 키로도 제출 가능
     input_text.submit(
         fn=correct_grammar,
-        inputs=input_text,
+        inputs=[input_text, api_key_input],
         outputs=output_text
     )
+    
+    # 지우기 버튼
+    clear_btn.click(
+        fn=lambda: ("", ""),
+        outputs=[input_text, output_text]
+    )
+    
+    gr.Markdown(
+        """
+        ---
+        ### ℹ️ 안내사항
+        - API 키는 서버에 저장되지 않습니다
+        - Gemini API는 무료 tier에서도 사용 가능합니다
+        - 문제가 발생하면 API 키가 올바른지 확인해��세요
+        """
+    )
 
 # 앱 실행
 if __name__ == "__main__":