Update app.py

app.py

@@ -16,7 +16,8 @@ MODELS = {
     "Microsoft": "microsoft/Phi-3-mini-4k-instruct",
     "Mixtral 8x7B": "mistralai/Mistral-7B-Instruct-v0.3",
     "Mixtral Nous-Hermes": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
-    "Aya-23-35B": "CohereForAI/aya-23-35B"
+    "Aya-23-35B": "CohereForAI/aya-23-35B",
+    "DeepSeek-V3": "deepseek/deepseek-chat"
 }
 
 # Cohere Command R+ 모델 ID 정의
@@ -146,6 +147,42 @@ def respond_chatgpt_qna(
         return f"오류가 발생했습니다: {str(e)}"
 
 
+def respond_deepseek_qna(
+    question: str,
+    system_message: str,
+    max_tokens: int,
+    temperature: float,
+    top_p: float,
+    deepseek_token: str
+):
+    """
+    DeepSeek 모델을 이용해 한 번의 질문(question)에 대한 답변을 반환하는 함수.
+    """
+    if not deepseek_token:
+        return "DeepSeek API 토큰이 필요합니다."
+
+    openai.api_key = deepseek_token
+    openai.api_base = "https://api.deepseek.com/v1"
+
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": question}
+    ]
+
+    try:
+        response = openai.ChatCompletion.create(
+            model="deepseek-chat",
+            messages=messages,
+            max_tokens=max_tokens,
+            temperature=temperature,
+            top_p=top_p,
+        )
+        assistant_message = response.choices[0].message['content']
+        return assistant_message
+    except Exception as e:
+        return f"오류가 발생했습니다: {str(e)}"
+
+
 #############################
 # [기본코드] UI 부분 - 수정/삭제 불가
 #############################
@@ -171,6 +208,11 @@ with gr.Blocks() as demo:
             placeholder="Claude API 토큰을 입력하세요...",
             show_copy_button=False
         )
+        deepseek_token_box = gr.Textbox(
+            label="DeepSeek 토큰",
+            type="password",
+            placeholder="DeepSeek API 토큰을 입력하세요..."
+        )
 
     #################
     # 일반 모델 탭
@@ -338,54 +380,6 @@ with gr.Blocks() as demo:
             outputs=chatgpt_answer_output
         )
 
-    #################################################
-    # [클로드 플레이그라운드] - 개선된 코드
-    #################################################
-
-    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)}"
-
     #################
     # Claude 탭
     #################
@@ -454,8 +448,59 @@ with gr.Blocks() as demo:
             outputs=claude_answer_output
         )
 
+    #################
+    # DeepSeek 탭
+    #################
+    with gr.Tab("DeepSeek-V3"):
+        deepseek_input1 = gr.Textbox(label="입력1", lines=1)
+        deepseek_input2 = gr.Textbox(label="입력2", lines=1)
+        deepseek_input3 = gr.Textbox(label="입력3", lines=1)
+        deepseek_input4 = gr.Textbox(label="입력4", lines=1)
+        deepseek_input5 = gr.Textbox(label="입력5", lines=1)
+
+        deepseek_answer_output = gr.Textbox(label="결과", lines=5, interactive=False)
+
+        with gr.Accordion("고급 설정 (DeepSeek)", open=False):
+            deepseek_system_message = gr.Textbox(
+                value="""반드시 한글로 답변할 것.
+너는 DeepSeek-V3, 최고의 언어 모델이다.
+내가 요구하는 것을 최대한 자세하고 정확하게 답변하라.
+""",
+                label="System Message",
+                lines=3
+            )
+            deepseek_max_tokens = gr.Slider(minimum=100, maximum=4000, value=2000, step=100, label="Max Tokens")
+            deepseek_temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
+            deepseek_top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
+
+        deepseek_submit_button = gr.Button("전송")
+
+        def merge_and_call_deepseek(i1, i2, i3, i4, i5, sys_msg, mt, temp, top_p_, deepseek_token):
+            question = " ".join([i1, i2, i3, i4, i5])
+            return respond_deepseek_qna(
+                question=question,
+                system_message=sys_msg,
+                max_tokens=mt,
+                temperature=temp,
+                top_p=top_p_,
+                deepseek_token=deepseek_token
+            )
+
+        deepseek_submit_button.click(
+            fn=merge_and_call_deepseek,
+            inputs=[
+                deepseek_input1, deepseek_input2, deepseek_input3, deepseek_input4, deepseek_input5,
+                deepseek_system_message,
+                deepseek_max_tokens,
+                deepseek_temperature,
+                deepseek_top_p,
+                deepseek_token_box
+            ],
+            outputs=deepseek_answer_output
+        )
+
 #############################
 # 메인 실행부
 #############################
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()