Update app.py

app.py

@@ -2,6 +2,10 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import openai
 
+#############################
+# [기본코드] - 수정/삭제 불가
+#############################
+
 # 제거할 모델들을 MODELS 사전에서 제외
 MODELS = {
     "Zephyr 7B Beta": "HuggingFaceH4/zephyr-7b-beta",
@@ -142,6 +146,10 @@ def respond_chatgpt_qna(
         return f"오류가 발생했습니다: {str(e)}"
 
 
+#############################
+# [기본코드] UI 부분 - 수정/삭제 불가
+#############################
+
 with gr.Blocks() as demo:
     gr.Markdown("# Prompting AI - 일반 문답형 데모")
     gr.Markdown("언어모델별 문답형 테스트 데모입니다. 한 번에 한 질문씩만 주고받습니다.")
@@ -180,7 +188,6 @@ with gr.Blocks() as demo:
         question_input = gr.Textbox(label="질문을 입력하세요")
         answer_output = gr.Textbox(label="답변", interactive=False)
 
-        # '전송' 버튼 클릭 시 HF 모델 QnA 함수 호출
         submit_button = gr.Button("전송")
 
         submit_button.click(
@@ -261,5 +268,88 @@ with gr.Blocks() as demo:
             outputs=chatgpt_answer_output
         )
 
+    #################################################
+    # [클로드 플레이그라운드] - [참조코드] 일부 발췌
+    #################################################
+    with gr.Tab("클로드 플레이그라운드"):
+        gr.Markdown("클로드 모델과의 간단한 플레이그라운드입니다.")
+
+        # Claude용 토큰
+        claude_token_box = gr.Textbox(
+            label="Claude 토큰 (비공개)",
+            type="password",
+            placeholder="Claude API 토큰을 입력하세요..."
+        )
+
+        claude_system_message = gr.Textbox(
+            label="System Message",
+            value="반드시 한글로 답변할 것.\n너는 Anthropic에서 개발한 클로드이다.\n최대한 정확하고 친절하게 답변하라.\n",
+            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_question_input = gr.Textbox(label="질문을 입력하세요")
+        claude_answer_output = gr.Textbox(label="답변", interactive=False)
+        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=[
+                claude_question_input,
+                claude_system_message,
+                claude_max_tokens,
+                claude_temperature,
+                claude_top_p,
+                claude_token_box
+            ],
+            outputs=claude_answer_output
+        )
+
+#############################
+# 메인 실행부
+#############################
 if __name__ == "__main__":
     demo.launch()