Update services.py

services.py

@@ -15,7 +15,6 @@ class GeminiService:
             print("警告：找不到 GEMINI_API_KEY")
         
         self.client = genai.Client(api_key=api_key) if api_key else None
-        # 建議使用最新模型以獲得最佳分析能力
         self.model_id = os.getenv("GEMINI_MODEL_ID", "gemini-2.0-flash")
 
     def _check_client(self):
@@ -24,23 +23,25 @@ class GeminiService:
 
     def search_companies(self, query: str, exclude_names: List[str] = []) -> List[Dict]:
         """
-        Step 1: 搜尋台灣公司
+        Step 1: 領域探索 -> 公司列表
         """
         self._check_client()
         exclusion_prompt = ""
         if exclude_names:
             exclusion_prompt = f"IMPORTANT: Do not include: {', '.join(exclude_names)}."
 
-        # Phase 1: Google Search (廣泛搜尋)
+        # Phase 1: Google Search (廣泛探索)
+        # 這裡的 Prompt 強調：如果使用者輸入的是「領域(如: AI)」，請列出該領域的台灣代表性公司。
         search_prompt = f"""
         Using Google Search, find 5 to 10 prominent companies in Taiwan related to the query: "{query}".
         
-        **CRITICAL INSTRUCTIONS:**
-        1. **TARGET:** Focus on Taiwanese companies (or global companies with a major branch in Taiwan).
-        2. **IDENTIFIERS:** Try to find their distinct "Company Name" (e.g., 台積電 / 台灣積體電路製造股份有限公司).
+        **Instructions:**
+        1. **Domain Search:** If "{query}" is an industry or technology (e.g., "AI", "Green Energy"), list the top representative Taiwanese companies in this field.
+        2. **Company Search:** If "{query}" is a specific name, list that company and its direct competitors.
+        3. **Target:** Focus on Taiwanese companies (or global companies with major R&D in Taiwan).
         {exclusion_prompt}
         
-        List them (Full Name - Industry/Sector) in Traditional Chinese.
+        List them (Full Name - Industry/Main Product) in Traditional Chinese.
         """
 
         search_response = self.client.models.generate_content(
@@ -54,7 +55,7 @@ class GeminiService:
 
         # Phase 2: Extract JSON (結構化)
         extract_prompt = f"""
-        From the text below, extract company names and their industry.
+        From the text below, extract company names and their industry/main product.
         Calculate a Relevance Score (0-100) based on query: "{query}".
         
         Return ONLY a JSON array: [{{"name": "...", "industry": "...", "relevanceScore": 85}}]
@@ -87,37 +88,29 @@ class GeminiService:
         name = company.get('name')
         
         prompt = f"""
-        Act as a professional "Business Due Diligence Analyst" (商業徵信分析師). 
+        Act as a professional "Business Analyst & Investigator". 
         Conduct a comprehensive investigation on the Taiwanese company: "{name}".
 
-        **Investigation Targets (Must search for these specifically):**
+        **Investigation Targets:**
 
-        1.  **Corporate Identity (基本資料)**:
-            - Find the **Tax ID (統一編號)**.
-            - **Registered Capital (資本額)**.
+        1.  **Overview (基本盤)**:
+            - **Tax ID (統編)** & **Capital (資本額)**. (Try to find specific numbers)
             - **Representative (代表人)**.
-            - **Establishment Date (設立日期)**.
-            - *Source Hint: Ministry of Economic Affairs (經濟部商業司), Datagovtw.*
-
-        2.  **Scale & Business (規模與業務)**:
-            - **Employee Count**: Estimated number of employees.
-            - **Core Products/Services**: What do they actually sell or do?
-            - *Source Hint: 104 Job Bank, Company Website, LinkedIn.*
-
-        3.  **Market Reputation & Culture (評價與文化 - KEY PART)**:
-            - Search for employee reviews on **PTT (Tech_Job, Soft_Job, Salary)**, **Dcard (Work board)**, **Qollie (求職天眼通)**, or **Google Maps**.
-            - Summarize the **Pros** (e.g., high pay, free snacks) and **Cons** (e.g., toxic management, forced overtime, family business style).
-            - *Tone:* Be objective but highlight recurring complaints.
-
-        4.  **Legal & Risk Assessment (法律與風險 - CRITICAL)**:
-            - Search for keywords: "{name} 判決", "{name} 勞資糾紛", "{name} 違反勞基法", "{name} 詐騙", "{name} 吸金", "{name} 罰款".
-            - List any major lawsuits, fines, or controversies found in news or government records.
-            - If clean, state "No major public legal disputes found."
-
-        **Format Requirements**:
-        - Structure the output as a clean, readable report using Markdown.
-        - Use clear headings.
-        - **Language**: Traditional Chinese (繁體中文).
+            - **Core Business**: What specific problem do they solve? What is their "Ace" product?
+
+        2.  **Workforce & Culture (內部情報)**:
+            - **Employee Count**.
+            - **Reviews/Gossip**: Search **PTT (Tech_Job, Soft_Job)**, **Dcard**, **Qollie**.
+            - Summarize the *REAL* work vibe (e.g., "Good for juniors but low ceiling", "Free snacks but forced overtime").
+
+        3.  **Legal & Risks (排雷專區)**:
+            - Search: "{name} 勞資糾紛", "{name} 違反勞基法", "{name} 判決", "{name} 罰款".
+            - List any red flags found in government records or news.
+
+        **Format**:
+        - Use Markdown.
+        - Language: Traditional Chinese (繁體中文).
+        - Be objective but don't sugarcoat potential risks.
         """
 
         response = self.client.models.generate_content(
@@ -135,7 +128,6 @@ class GeminiService:
                 if chunk.web and chunk.web.uri and chunk.web.title:
                     sources.append({"title": chunk.web.title, "uri": chunk.web.uri})
         
-        # Deduplicate
         unique_sources = {v['uri']: v for v in sources}.values()
 
         return {
@@ -145,7 +137,7 @@ class GeminiService:
 
     def chat_with_ai(self, history: List[Dict], new_message: str, context: str) -> str:
         self._check_client()
-        system_instruction = f"You are a sharp Business Analyst. Answer based on this due diligence report:\n{context}"
+        system_instruction = f"You are an expert Business Consultant. Answer based on this company report:\n{context}"
         
         chat_history = []
         for h in history: