Update app.py

app.py

@@ -1,14 +1,39 @@
 import gradio as gr
 import google.generativeai as genai
 from typing import Dict, List
+import json
 
 # Cấu hình API key cho Gemini
-GOOGLE_API_KEY = "AIzaSyAnZ-wajwMF6Uy66FmkC4MOugA2fA66kMA"
+GOOGLE_API_KEY = "AIzaSyAnZ-wajwMF6Uy66FmkC4MOugA2fA66kMA"  # Thay thế bằng API key của bạn
 genai.configure(api_key=GOOGLE_API_KEY)
 
 # Khởi tạo model
 model = genai.GenerativeModel('gemini-1.5-flash-latest')
 
+# Định nghĩa câu hỏi MBTI test
+MBTI_QUESTIONS = [
+    {
+        "question": "Bạn thích làm việc theo cách nào hơn?",
+        "options": ["Một mình (I)", "Trong nhóm (E)"],
+        "dimension": "IE"
+    },
+    {
+        "question": "Khi tiếp nhận thông tin, bạn thiên về:",
+        "options": ["Các sự kiện và chi tiết cụ thể (S)", "Ý tưởng và khái niệm trừu tượng (N)"],
+        "dimension": "SN"
+    },
+    {
+        "question": "Khi đưa ra quyết định, bạn dựa vào:",
+        "options": ["Logic và phân tích khách quan (T)", "Cảm xúc và giá trị cá nhân (F)"],
+        "dimension": "TF"
+    },
+    {
+        "question": "Trong cuộc sống, bạn thích:",
+        "options": ["Lập kế hoạch và tổ chức (J)", "Linh hoạt và thích nghi (P)"],
+        "dimension": "JP"
+    }
+]
+
 # Định nghĩa đầy đủ các kiểu MBTI và đặc điểm nghề nghiệp
 MBTI_CAREER_TRAITS = {
     "ISTJ": {
@@ -113,71 +138,125 @@ class CareerAdvisor:
     def __init__(self):
         self.chat_history = []
         self.system_prompt = """Bạn là một chuyên gia tư vấn nghề nghiệp có nhiều năm kinh nghiệm. 
-        Nhiệm vụ của bạn là giúp người dùng định hướng nghề nghiệp, đưa ra lời khuyên phù hợp và thiết thực.
-        Hãy trả lời một cách chuyên nghiệp, đồng cảm và chi tiết."""
+        Nhiệm vụ của bạn là giúp người dùng định hướng nghề nghiệp, đưa ra lời khuyên phù hợp và thiết thực."""
     
-    def get_career_advice(self, user_input: str, mbti_type: str) -> str:
-        mbti_info = MBTI_CAREER_TRAITS[mbti_type]
+    def calculate_mbti(self, answers):
+        """Tính toán kiểu MBTI dựa trên câu trả lời của người dùng"""
+        result = ""
+        dimensions = {"IE": [], "SN": [], "TF": [], "JP": []}
         
-        prompt = f"""
-        Dựa trên thông tin người dùng có:
-        - Kiểu tính cách MBTI: {mbti_type}
-        - Đặc điểm tính cách: {mbti_info['traits']}
-        - Điểm mạnh: {mbti_info['strengths']}
-        - Điểm cần cải thiện: {mbti_info['weaknesses']}
-        - Ngành nghề phù hợp: {mbti_info['careers']}
+        for answer, question in zip(answers, MBTI_QUESTIONS):
+            dim = question["dimension"]
+            dimensions[dim].append(0 if answer == 0 else 1)
         
-        Thông tin bổ sung từ người dùng: {user_input}
+        result += "I" if sum(dimensions["IE"]) < len(dimensions["IE"])/2 else "E"
+        result += "S" if sum(dimensions["SN"]) < len(dimensions["SN"])/2 else "N"
+        result += "T" if sum(dimensions["TF"]) < len(dimensions["TF"])/2 else "F"
+        result += "J" if sum(dimensions["JP"]) < len(dimensions["JP"])/2 else "P"
         
-        Hãy đưa ra phân tích chi tiết và tư vấn về:
-        1. Phân tích điểm mạnh và cách phát huy trong công việc
-        2. Điểm cần cải thiện và cách khắc phục
-        3. Top 3 ngành nghề phù hợp nhất và lý do
-        4. Lộ trình phát triển cụ thể cho từng ngành (3-5 năm)
-        5. Các kỹ năng cần phát triển và cách thức phát triển
-        
-        Hãy trả lời một cách có cấu trúc, chi tiết và thực tế.
-        """
-        
-        response = model.generate_content(prompt)
-        return response.result.text
+        return result
+
+    def get_career_advice(self, user_input: str, mbti_type: str) -> str:
+        try:
+            mbti_info = MBTI_CAREER_TRAITS[mbti_type]
+            
+            prompt = f"""
+            Dựa trên thông tin người dùng có:
+            - Kiểu tính cách MBTI: {mbti_type}
+            - Đặc điểm tính cách: {mbti_info['traits']}
+            - Điểm mạnh: {mbti_info['strengths']}
+            - Điểm cần cải thiện: {mbti_info['weaknesses']}
+            - Ngành nghề phù hợp: {mbti_info['careers']}
+            
+            Thông tin bổ sung từ người dùng: {user_input}
+            
+            Hãy đưa ra phân tích chi tiết và tư vấn về:
+            1. Phân tích điểm mạnh và cách phát huy trong công việc
+            2. Điểm cần cải thiện và cách khắc phục
+            3. Top 3 ngành nghề phù hợp nhất và lý do
+            4. Lộ trình phát triển cụ thể cho từng ngành (3-5 năm)
+            5. Các kỹ năng cần phát triển và cách thức phát triển
+            
+            Hãy trả lời một cách có cấu trúc, chi tiết và thực tế.
+            """
+            
+            response = model.generate_content(prompt)
+            return response.result.text
+        except Exception as e:
+            return f"Đã xảy ra lỗi khi tạo tư vấn: {str(e)}"
 
     def chat_with_advisor(self, message: str, history: List[Dict[str, str]]) -> List[Dict[str, str]]:
-        # Tạo prompt cho cuộc trò chuyện
-        context = f"""
-        {self.system_prompt}
-        
-        Dựa trên cuộc trò chuyện trước đó và câu hỏi mới của người dùng:
-        "{message}"
-        
-        Hãy đưa ra lời khuyên chuyên nghiệp, cụ thể và có tính thực tiễn cao.
-        Nếu cần thêm thông tin để tư vấn tốt hơn, hãy đặt câu hỏi phù hợp.
-        """
-        
-        response = model.generate_content(context)
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": response.result.text})
-        return history
+        try:
+            context = f"""
+            {self.system_prompt}
+            
+            Dựa trên cuộc trò chuyện trước đó và câu hỏi mới của người dùng:
+            "{message}"
+            
+            Hãy đưa ra lời khuyên chuyên nghiệp, cụ thể và có tính thực tiễn cao.
+            Nếu cần thêm thông tin để tư vấn tốt hơn, hãy đặt câu hỏi phù hợp.
+            """
+            
+            response = model.generate_content(context)
+            history.append({"role": "user", "content": message})
+            history.append({"role": "assistant", "content": response.result.text})
+            return history
+        except Exception as e:
+            history.append({"role": "assistant", "content": f"Đã xảy ra lỗi: {str(e)}"})
+            return history
 
 def create_interface():
     advisor = CareerAdvisor()
     
-    with gr.Blocks(title="AI Career Advisor") as interface:
+    with gr.Blocks(title="AI Career Advisor", theme=gr.themes.Soft()) as interface:
         gr.Markdown("""
-        # AI Career Advisor - Tư vấn nghề nghiệp thông minh
+        # 🎯 AI Career Advisor - Tư vấn nghề nghiệp thông minh
         
-        Chào mừng bạn đến với hệ thống tư vấn nghề nghiệp AI. Hệ thống có hai chức năng chính:
-        1. Tư vấn dựa trên tính cách MBTI
-        2. Tư vấn 1-1 với chuyên gia AI
+        Chào mừng bạn đến với hệ thống tư vấn nghề nghiệp AI. Hãy bắt đầu bằng việc cho chúng tôi biết về tính cách MBTI của bạn.
         """)
         
-        with gr.Tab("Tư vấn theo MBTI"):
+        # Tab chọn trạng thái biết/không biết MBTI
+        with gr.Tab("🎯 Bắt đầu"):
+            know_mbti = gr.Radio(
+                choices=["Tôi đã biết kiểu tính cách MBTI của mình", "Tôi chưa biết kiểu tính cách MBTI của mình"],
+                label="Trạng thái",
+                type="index"
+            )
+        
+        # Tab làm bài test MBTI
+        with gr.Tab("📝 Khám phá tính cách MBTI") as mbti_test_tab:
+            gr.Markdown("""### Hãy trả lời các câu hỏi sau để xác định tính cách MBTI của bạn""")
+            
+            questions = []
+            for q in MBTI_QUESTIONS:
+                questions.append(
+                    gr.Radio(
+                        choices=q["options"],
+                        label=q["question"],
+                        type="index"
+                    )
+                )
+            
+            test_submit = gr.Button("Xác định tính cách MBTI", variant="primary")
+            mbti_result = gr.Textbox(label="Kết quả tính cách MBTI của bạn", interactive=False)
+            
+            def calculate_mbti_result(*answers):
+                return advisor.calculate_mbti(answers)
+            
+            test_submit.click(
+                fn=calculate_mbti_result,
+                inputs=questions,
+                outputs=mbti_result
+            )
+        
+        # Tab tư vấn nghề nghiệp
+        with gr.Tab("💼 Tư vấn nghề nghiệp"):
             mbti_input = gr.Dropdown(
                 choices=list(MBTI_CAREER_TRAITS.keys()),
-                label="Chọn kiểu tính cách MBTI của bạn"
+                label="Kiểu tính cách MBTI của bạn",
+                info="Chọn kiểu tính cách MBTI đã xác định"
             )
             
-            # Hiển thị thông tin MBTI
             mbti_info = gr.Markdown(visible=False)
             
             def update_mbti_info(mbti_type):
@@ -186,27 +265,34 @@ def create_interface():
                     return gr.Markdown.update(
                         visible=True,
                         value=f"""
-                        ### Thông tin về kiểu tính cách {mbti_type}:
-                        - **Đặc điểm**: {info['traits']}
-                        - **Điểm mạnh**: {info['strengths']}
-                        - **Điểm cần cải thiện**: {info['weaknesses']}
-                        - **Ngành nghề phù hợp**: {info['careers']}
+                        ### 📊 Thông tin về kiểu tính cách {mbti_type}:
+                        
+                        #### 🎯 Đặc điểm chính:
+                        {info['traits']}
+                        
+                        #### 💪 Điểm mạnh:
+                        {info['strengths']}
+                        
+                        #### 🔄 Điểm cần cải thiện:
+                        {info['weaknesses']}
+                        
+                        #### 👔 Ngành nghề phù hợp:
+                        {info['careers']}
                         """
                     )
                 return gr.Markdown.update(visible=False)
             
             user_info = gr.Textbox(
-                label="Chia sẻ thêm về bản thân (sở thích, kỹ năng, mong muốn...)",
+                label="Chia sẻ thêm về bản thân",
                 lines=5,
                 placeholder="Ví dụ: Tôi thích làm việc với số liệu, có kỹ năng phân tích tốt, mong muốn một công việc ổn định..."
             )
             
-            get_advice_btn = gr.Button("Nhận tư vấn", variant="primary")
+            get_advice_btn = gr.Button("📋 Nhận tư vấn chi tiết", variant="primary")
             
-            advice_output = gr.Textbox(
+            advice_output = gr.Markdown(
                 label="Kết quả tư vấn",
-                lines=15,
-                interactive=False
+                value="Kết quả tư vấn sẽ hiển thị ở đây..."
             )
             
             # Kết nối các components
@@ -222,24 +308,24 @@ def create_interface():
                 outputs=advice_output
             )
         
-        with gr.Tab("Tư vấn 1-1 với Chuyên gia"):
+        # Tab chat với chuyên gia
+        with gr.Tab("💬 Tư vấn 1-1 với Chuyên gia"):
             gr.Markdown("""
-            ### Trò chuyện với Chuyên gia Tư vấn AI
+            ### 💬 Trò chuyện với Chuyên gia Tư vấn AI
             
             Bạn có thể đặt câu hỏi về:
-            - Định hướng nghề nghiệp
-            - Phát triển kỹ năng
-            - Chuyển đổi công việc
-            - Lộ trình sự nghiệp
-            - Và các vấn đề nghề nghiệp khác
+            - 🎯 Định hướng nghề nghiệp
+            - 📚 Phát triển kỹ năng
+            - 🔄 Chuyển đổi công việc
+            - 📈 Lộ trình sự nghiệp
+            - 💼 Và các vấn đề nghề nghiệp khác
             """)
             
             chatbot = gr.Chatbot(
                 label="Cuộc trò chuyện",
                 height=400,
                 bubble_full_width=False,
-                avatar_images=(None, "🧑‍💼"),
-                type='messages'  # Chỉ định kiểu 'messages'
+                avatar_images=(None, "🧑‍💼")
             )
             
             msg = gr.Textbox(
@@ -248,7 +334,7 @@ def create_interface():
                 lines=3
             )
             
-            clear = gr.Button("Xóa lịch sử")
+            clear = gr.Button("🗑️ Xóa lịch sử")
             
             def respond(message, chat_history):
                 bot_message = advisor.chat_with_advisor(message, chat_history)
@@ -260,54 +346,27 @@ def create_interface():
             msg.submit(respond, [msg, chatbot], [msg, chatbot])
             clear.click(clear_history, None, chatbot)
         
-        # Thêm phần footer với thông tin và hướng dẫn
-        with gr.Accordion("Hướng dẫn sử dụng", open=False):
-            gr.Markdown("""
-            ### Cách sử dụng AI Career Advisor:
-            
-            1. **Tư vấn theo MBTI**:
-               - Chọn kiểu tính cách MBTI của bạn
-               - Xem thông tin chi tiết về kiểu tính cách
-               - Chia sẻ thêm thông tin về bản thân
-               - Nhấn "Nhận tư vấn" để nhận phân tích chi tiết
-            
-            2. **Tư vấn 1-1 với Chuyên gia**:
-               - Đặt câu hỏi cụ thể về nghề nghiệp
-               - Trò chuyện tự nhiên với AI
-               - Có thể hỏi thêm để làm rõ các vấn đề
-               - Xóa lịch sử để bắt đầu cuộc trò chuyện mới
-            
-            3. **Lưu ý**:
-               - Càng cung cấp nhiều thông tin, tư vấn càng chính xác
-               - Có thể kết hợp cả hai hình thức tư vấn
-               - Lưu lại kết quả tư vấn để tham khảo sau
-            """)
-            
-        with gr.Accordion("Thông tin thêm", open=False):
-            gr.Markdown("""
-            ### Về AI Career Advisor:
-            
-            Hệ thống sử dụng AI tiên tiến để cung cấp tư vấn nghề nghiệp cá nhân hóa. 
-            Kết hợp giữa phân tích tính cách MBTI và tư vấn chuyên sâu để giúp bạn:
-            
-            - Hiểu rõ điểm mạnh và điểm cần phát triển
-            - Khám phá các ngành nghề phù hợp
-            - Xây dựng lộ trình phát triển sự nghiệp
-            - Đưa ra quyết định nghề nghiệp đúng đắn
-            
-            *Lưu ý: Đây là công cụ hỗ trợ tham khảo. Quyết định cuối cùng vẫn nên dựa trên đánh giá cá nhân và tham khảo ý kiến của các chuyên gia thực tế.*
-            """)
+        # Show/hide tabs based on MBTI knowledge
+        def update_tabs(know_mbti):
+            if know_mbti == 0:  # Đã biết MBTI
+                return gr.Tabs.update(selected="💼 Tư vấn nghề nghiệp")
+            else:  # Chưa biết MBTI
+                return gr.Tabs.update(selected="📝 Khám phá tính cách MBTI")
+        
+        know_mbti.change(
+            fn=update_tabs,
+            inputs=[know_mbti],
+            outputs=[interface]
+        )
 
     return interface
 
 def main():
-    # Khởi chạy ứng dụng
     interface = create_interface()
     interface.launch(
         share=True,
         server_name="0.0.0.0",
         server_port=7860
-        # Loại bỏ tham số 'enable_queue'
     )
 
 if __name__ == "__main__":