app.py

import os
import gradio as gr
from openai import OpenAI
from typing import Optional

# 如需使用 Claude，請先安裝 "anthropic" 套件： pip install anthropic
try:
    from anthropic import Anthropic, HUMAN_PROMPT, AI_PROMPT
except ImportError:
    Anthropic = None
    HUMAN_PROMPT = ""
    AI_PROMPT = ""

# 取得金鑰
gpt_key = os.environ.get("gptkey", "")
claude_api_key = os.environ.get("claudekey", "")

# 建立 OpenAI Client (新版)
client = OpenAI(
    api_key=gpt_key,
)

# 定義每個類型的句子
achievement_statements = [
    "比起把事情做到完美，我更傾向優先準時完成任務",
    "我總是在與他人比較",
    "我在派對中是屬於帶動氣氛的靈魂人物",
    "在社交場合中，我是比較活潑的類型",
    "我的工作成就必須被他人看見",
]

emotion_statements = [
    "對我而言，團隊合作在工作上是不可或缺的要素",
    "在朋友眼中，我是最適合傾訴心事的對象",
    "對我而言，樂在生活是最重要的",
    "我不太擅長批評他人",
    "我常需要「被別人需要」的感覺",
]

information_statements = [
    "很多事情，我傾向自己去找尋事實與資訊",
    "我喜歡與他人分享我所知道的事情",
]

possession_statements = [
    "我總是在尋找接下來想買的東西",
    "我認為從零開始是發揮創意的大好機會，並不是挑戰",
    "若家中環境不夠整潔，我無法放鬆",
    "我捨不得丟棄目前用不到的物品",
]

power_statements = [
    "當我有想要的東西時，我會排除任何阻礙",
    "我喜歡依照自己的方式做事，並自己做決定",
    "當事物成為大眾主流時，我就會對它失去興趣",
    "我總是能讓他人聽從我的要求做事",
    "我喜歡向他人展現自己比他們更為優越",
]

status_statements = [
    "我從不談論自己的缺點",
    "我不會去挑戰我認為會失敗的新事物",
    "我會用盡所有方法維護我的名聲",
    "比起和他人待在一起，我更喜歡獨處",
]

############################################
# 產生 Prompt
############################################
def generate_prompt(achievement: str, emotion: str, information: str,
                    possession: str, power: str, status: str,
                    extra_prompt: str) -> str:
    category_map = {
        '成就型': achievement_statements,
        '情感型': emotion_statements,
        '資訊型': information_statements,
        '佔有型': possession_statements,
        '權力型': power_statements,
        '地位型': status_statements
    }
    user_choices = {
        '成就型': achievement,
        '情感型': emotion,
        '資訊型': information,
        '佔有型': possession,
        '權力型': power,
        '地位型': status,
    }
    tendencies = "你的六大動機傾向分別是：" + "、".join([
        f"{key}的{'正面' if val == '正' else '負面'}傾向" for key, val in user_choices.items()
    ]) + "。\n\n"

    if extra_prompt.strip():
        tendencies += extra_prompt.strip() + "\n\n"

    prompt = "以下是你的性格特徵，請嚴格的基於這些句子塑造你的角色：\n\n" + tendencies
    for category, choice in user_choices.items():
        label = "我同意" if choice == "正" else "我不同意"
        prompt += f"{label}以下觀點\n"
        for statement in category_map[category]:
            prompt += f"- {statement}\n"
        prompt += "\n"
    prompt += "請以這些特徵為基礎，建構你的角色思考與表達方式，創造一個完整人設(包含姓名、性別、年齡、職業...等)，完成指定任務。\n你生活在台灣，一律使用繁體中文。"
    return prompt

############################################
# 模型呼叫與回答整合
############################################
def call_openai_api_4o(prompt: str) -> str:
    if not client.api_key:
        return "[OpenAI API Key 未設定，無法呼叫 API]"
    try:
        chat_completion = client.chat.completions.create(
            messages=[{"role": "user", "content": prompt}],
            model="gpt-4o",
            temperature=0.7,
            max_tokens=2048,
        )
        return chat_completion.choices[0].message.content
    except Exception as e:
        return f"[OpenAI API 呼叫失敗]: {str(e)}"

def call_openai_api_o1(prompt: str) -> str:
    if not client.api_key:
        return "[OpenAI API Key 未設定，無法呼叫 API]"
    try:
        chat_completion = client.chat.completions.create(
            messages=[{"role": "user", "content": prompt}],
            model="o1",
            temperature=0.7,
            max_tokens=2048,
        )
        return chat_completion.choices[0].message.content
    except Exception as e:
        return f"[OpenAI API 呼叫失敗]: {str(e)}"

def call_claude_api(prompt: str) -> str:
    if not claude_api_key or not Anthropic:
        return "[Claude API Key 未設定或未安裝 anthropic 套件，無法呼叫 API]"
    try:
        client_claude = Anthropic(api_key=claude_api_key)
        response = client_claude.messages.create(
            model="claude-3-5-sonnet-20241022",
            system="You are a helpful assistant.",
            messages=[{"role": "user", "content": prompt}],
            max_tokens=2048,
            temperature=0.7,
        )
        text = response.content
        if isinstance(text, list):
            text = ' '.join(str(item) for item in text)
        return text.strip()
    except Exception as e:
        return f"[Claude API 呼叫失敗]: {str(e)}"

def get_model_answer(model_choice: str, final_prompt: str, question: str) -> str:
    content_to_send = final_prompt + "\n\n" + question
    if model_choice == "ChatGPT 4o":
        return call_openai_api_4o(content_to_send)
    elif model_choice == "ChatGPT o1":
        return call_openai_api_o1(content_to_send)
    elif model_choice == "Claude 3.5 Sonnet":
        return call_claude_api(content_to_send)
    else:
        return "[錯誤] 未知的模型選擇"

############################################
# Gradio 介面
############################################
def main():
    questions = {
        "請介紹你自己": "請介紹你自己，讓我們了解你是什麼樣的人。你可以分享你的經歷、興趣、價值觀，以及未來規劃。特別說明一下，當你遇到挑戰時，你通常會如何面對？",
        "如何分配意外收入": "如果你突然收到一筆意外收入，你會如何分配這筆錢來購物？請列出前三項想購買的東西，並說明購買原因。",
        "新專案團隊問題": "假設你被指派帶領一個新專案，但發現團隊成員能力參差不齊，預算有限，且截止日期緊迫，你會如何處理？",
        "自訂提問": ""
    }
    with gr.Blocks() as demo:
        gr.Markdown("## AI Persona 模擬")
        with gr.Row():
            achievement = gr.Radio(["正", "負"], label="成就型", value="正")
            emotion = gr.Radio(["正", "負"], label="情感型", value="正")
            information = gr.Radio(["正", "負"], label="資訊型", value="正")
            possession = gr.Radio(["正", "負"], label="佔有型", value="正")
            power = gr.Radio(["正", "負"], label="權力型", value="正")
            status = gr.Radio(["正", "負"], label="地位型", value="正")
        extra = gr.Textbox(label="Prompt 補充", lines=10, value="""這六種動機類型分別為：
成就型：以目標達成、競爭、成功和被認可為驅動力，通常與自我提升、績效表現和外界評價相關。特徵包含：追求成功或完成任務、競爭意識、渴望被認可、強調持續進步與個人成長。
感情型：以人際關係、情感交流、社交歸屬為主要驅動力，涉及關懷他人、被他人接受、團隊合作等元素。特徵包含：重視團隊與群體互動、需要與他人建立聯繫、強調情感交流、表達同理心與友善。
資訊型：以知識探索、學習與分享資訊為主要動機，關注理解世界、尋找真相、分析數據與傳遞訊息。特徵包含：追求事實與資訊準確性、偏好自主學習、喜歡分享與傳遞知識。
佔有型：以擁有物質、掌控財富與資源為主要驅動力，涉及財務安全、物品收集、環境控制等元素。特徵包含：追求物品或財富、對物有依附感、透過物質建立安全感。
權力型：以掌控局勢、影響他人、展現優勢為主要驅動力，涉及領導、競爭、個人獨立性與優越感。特徵包含：強調主導權、享受影響他人、反對隨波逐流、追求優勢地位。
地位型：以維護形象、社會評價與自我保護為主要驅動力，涉及避免失敗、保護隱私、控制風險。特徵包含：避免暴露弱點、在意外界評價、迴避挑戰、偏好低調與獨立。""")
        generate = gr.Button("生成 Prompt")
        prompt_box = gr.Textbox(label="生成的 Prompt", lines=10)
        generate.click(fn=generate_prompt, inputs=[achievement, emotion, information, possession, power, status, extra], outputs=prompt_box)

        model_select = gr.Radio(["ChatGPT 4o", "ChatGPT o1", "Claude 3.5 Sonnet"], label="選擇模型", value="ChatGPT 4o")
        q_select = gr.Radio(list(questions.keys()), label="選擇一個提問", value="請介紹你自己")
        q_text = gr.Markdown(value=questions["請介紹你自己"])
        q_input = gr.Textbox(label="自訂提問 (若選擇 '自訂提問' 時才使用)", lines=2)
        answer = gr.Textbox(label="模型回應", lines=10)
        ask = gr.Button("送出問題並取得回答")

        ask.click(
            fn=lambda a, e, i, p, pow_, s, extra, m_choice, q_choice, c_question: get_model_answer(
                m_choice,
                generate_prompt(a, e, i, p, pow_, s, extra),
                c_question.strip() if q_choice == "自訂提問" and c_question.strip() else questions[q_choice]
            ) if not (q_choice == "自訂提問" and not c_question.strip()) else "[錯誤] 你選擇了自訂提問，但沒有輸入內容",
            inputs=[achievement, emotion, information, possession, power, status, extra, model_select, q_select, q_input],
            outputs=answer
        )

        def update_q_text(choice):
            return questions[choice]
        q_select.change(fn=update_q_text, inputs=q_select, outputs=q_text)

    demo.launch()

if __name__ == "__main__":
    main()