Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import openai
+import anthropic
+from typing import Optional
 
 #############################
 # [기본코드] - 수정/삭제 불가
@@ -237,7 +239,7 @@ with gr.Blocks() as demo:
         )
 
     # --- Tab: ChatGPT ---
-    with gr.Tab("gpt-4o-mini"):
+    with gr.Tab("ChatGPT"):
         chatgpt_system_message = gr.Textbox(
             value="""반드시 한글로 답변할 것.
 너는 ChatGPT, OpenAI에서 개발한 언어 모델이다.
@@ -269,72 +271,109 @@ with gr.Blocks() as demo:
         )
 
     #################################################
-    # [클로드 플레이그라운드] - [참조코드] 일부 발췌
+    # [클로드 플레이그라운드] - 개선된 코드
     #################################################
-    with gr.Tab("claude-3-haiku"):
+
+    def validate_claude_token(token: str) -> bool:
+        """Claude API 토큰 검증"""
+        return bool(token and len(token.strip()) >= 10)
+
+    def respond_claude_qna(
+        question: str,
+        system_message: str,
+        max_tokens: int,
+        temperature: float,
+        top_p: float,
+        claude_api_key: str
+    ) -> str:
+        """
+        Claude API를 사용한 개선된 응답 생성 함수
+        """
+        if not validate_claude_token(claude_api_key):
+            return "유효한 Claude API 토큰이 필요합니다."
+    
+        try:
+            client = anthropic.Anthropic(api_key=claude_api_key)
+            
+            # 메시지 생성
+            message = client.messages.create(
+                model="claude-3-haiku-20240307",
+                max_tokens=max_tokens,
+                temperature=temperature,
+                system=system_message,
+                messages=[
+                    {
+                        "role": "user",
+                        "content": question
+                    }
+                ]
+            )
+            
+            return message.content[0].text
+                
+        except anthropic.APIError as ae:
+            return f"Claude API 오류: {str(ae)}"
+        except anthropic.RateLimitError:
+            return "요청 한도를 초과했습니다. 잠시 후 다시 시도해주세요."
+        except Exception as e:
+            return f"예상치 못한 오류가 발생했습니다: {str(e)}"
+
+    with gr.Tab("클로드 플레이그라운드"):
         gr.Markdown("클로드 모델과의 간단한 플레이그라운드입니다.")
 
-        # Claude용 토큰
         claude_token_box = gr.Textbox(
             label="Claude 토큰 (비공개)",
             type="password",
-            placeholder="Claude API 토큰을 입력하세요..."
+            placeholder="Claude API 토큰을 입력하세요...",
+            show_copy_button=False
         )
 
         claude_system_message = gr.Textbox(
             label="System Message",
-            value="반드시 한글로 답변할 것.\n너는 Anthropic에서 개발한 클로드이다.\n최대한 정확하고 친절하게 답변하라.\n",
+            value="""반드시 한글로 답변할 것.
+너는 Anthropic에서 개발한 클로드이다.
+최대한 정확하고 친절하게 답변하라.""",
             lines=3
         )
 
-        claude_max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens")
-        claude_temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
-        claude_top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p")
+        claude_max_tokens = gr.Slider(
+            minimum=1,
+            maximum=4096,
+            value=512,
+            step=1,
+            label="Max Tokens"
+        )
+        
+        claude_temperature = gr.Slider(
+            minimum=0.1,
+            maximum=2.0,
+            value=0.7,
+            step=0.05,
+            label="Temperature"
+        )
+        
+        claude_top_p = gr.Slider(
+            minimum=0.1,
+            maximum=1.0,
+            value=0.95,
+            step=0.05,
+            label="Top-p"
+        )
 
-        claude_question_input = gr.Textbox(label="질문을 입력하세요")
-        claude_answer_output = gr.Textbox(label="답변", interactive=False)
+        claude_question_input = gr.Textbox(
+            label="질문을 입력하세요",
+            lines=3,
+            placeholder="질문을 입력하세요..."
+        )
+            
+        claude_answer_output = gr.Textbox(
+            label="답변",
+            interactive=False,
+            lines=10
+        )
+        
         claude_submit_button = gr.Button("전송")
 
-        # Claude 응답 함수 (최소 기능만 사용)
-        def respond_claude_qna(
-            question: str,
-            system_message: str,
-            max_tokens: int,
-            temperature: float,
-            top_p: float,
-            claude_api_key: str
-        ):
-            """
-            [참조코드]에서 최소한의 클로드 호출 ���분만 발췌한 함수.
-            플레이그라운드 형태로 한 번의 질문에 대한 답변을 반환.
-            """
-            import anthropic
-
-            # 토큰이 없으면 에러
-            if not claude_api_key:
-                return "Claude API 토큰이 필요합니다."
-
-            try:
-                claude_client = anthropic.Anthropic(api_key=claude_api_key)
-                # 클로드에게 보낼 프롬프트 구성
-                # Anthropic 공식 예시를 참고하여 HUMAN_PROMPT / AI_PROMPT 사용
-                prompt = (
-                    f"{anthropic.HUMAN_PROMPT}\n"
-                    f"{system_message}\n\n"
-                    f"질문: {question}\n"
-                    f"{anthropic.AI_PROMPT}"
-                )
-                response = claude_client.completions.create(
-                    model="claude-3-haiku-20240307",
-                    max_tokens_to_sample=max_tokens,
-                    temperature=temperature,
-                    top_p=top_p,
-                    prompt=prompt
-                )
-                return response.completion.strip()
-            except Exception as e:
-                return f"오류가 발생했습니다: {str(e)}"
-
         claude_submit_button.click(
             fn=respond_claude_qna,
             inputs=[
@@ -353,3 +392,4 @@ with gr.Blocks() as demo:
 #############################
 if __name__ == "__main__":
     demo.launch()
+