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()