Update app.py

app.py

@@ -8,7 +8,7 @@ from huggingface_hub import HfApi, hf_hub_download
 
 # Load Env
 load_dotenv()
-SAVE_FILE = os.getenv("SAVE_FILE_NAME", "saved_professors.json")
+SAVE_FILE = os.getenv("SAVE_FILE_NAME", "saved_companies.json")
 HF_TOKEN = os.getenv("HF_TOKEN")
 DATASET_REPO_ID = os.getenv("DATASET_REPO_ID")
 
@@ -21,8 +21,9 @@ except Exception as e:
 
 # --- Helper Functions ---
 
-def get_key(p):
-    return f"{p['name']}-{p['university']}"
+def get_key(c):
+    # 使用公司名稱當作 Key
+    return f"{c['name']}"
 
 def load_data():
     data = []
@@ -68,76 +69,75 @@ def save_data(data):
                 path_in_repo=SAVE_FILE,
                 repo_id=DATASET_REPO_ID,
                 repo_type="dataset",
-                commit_message="Sync data from Space"
+                commit_message="Sync company data"
             )
         except Exception as e:
             print(f"Upload Error: {e}")
 
 def format_df(source_list, saved_list):
     if not source_list:
-        return pd.DataFrame(columns=["狀態", "姓名", "大學", "系所", "標籤"])
+        return pd.DataFrame(columns=["狀態", "公司名稱", "產業類別", "標籤"])
     
     if saved_list is None:
         saved_list = []
     
-    saved_map = {get_key(p): p for p in saved_list}
+    saved_map = {get_key(c): c for c in saved_list}
     
     data = []
-    for p in source_list:
-        display_p = saved_map.get(get_key(p), p)
+    for c in source_list:
+        display_c = saved_map.get(get_key(c), c)
         
-        status_map = {'match': '✅', 'mismatch': '❌', 'pending': '❓'}
-        status_icon = status_map.get(display_p.get('status'), '')
-        has_detail = "📄" if display_p.get('details') else ""
+        status_map = {'good': '✅ 優質', 'risk': '⚠️ 風險', 'pending': '❓ 未定'}
+        status_icon = status_map.get(display_c.get('status'), '')
+        has_detail = "📄" if display_c.get('details') else ""
         
-        tags = ", ".join(display_p.get('tags', []))
+        tags = ", ".join(display_c.get('tags', []))
         
         data.append([
             f"{status_icon} {has_detail}",
-            display_p['name'],
-            display_p['university'],
-            display_p['department'],
+            display_c['name'],
+            display_c.get('industry', '未知'),
             tags
         ])
-    return pd.DataFrame(data, columns=["狀態", "姓名", "大學", "系所", "標籤"])
+    return pd.DataFrame(data, columns=["狀態", "公司名稱", "產業類別", "標籤"])
 
-def get_tags_text(prof):
-    if not prof or not prof.get('tags'):
+def get_tags_text(comp):
+    if not comp or not comp.get('tags'):
         return "目前標籤: (無)"
-    return "🏷️ " + ", ".join([f"`{t}`" for t in prof['tags']])
+    return "🏷️ " + ", ".join([f"`{t}`" for t in comp['tags']])
 
-def get_tags_choices(prof):
-    if not prof: return []
-    return prof.get('tags', [])
+def get_tags_choices(comp):
+    if not comp: return []
+    return comp.get('tags', [])
 
 # --- Event Handlers ---
 
-def search_professors(query, current_saved):
+def search_companies(query, current_saved):
     if not query: return gr.update(), current_saved, gr.update()
     
     try:
-        results = gemini_service.search_professors(query)
+        results = gemini_service.search_companies(query)
         return format_df(results, current_saved), results, gr.update(visible=True)
     except Exception as e:
         raise gr.Error(f"搜尋失敗: {e}")
 
-def load_more(query, current_search_results, current_saved):
-    if not query: return gr.update(), current_search_results
+def load_more(query, current_results, current_saved):
+    if not query: return gr.update(), current_results
     
-    current_names = [p['name'] for p in current_search_results]
+    current_names = [c['name'] for c in current_results]
     try:
-        new_results = gemini_service.search_professors(query, exclude_names=current_names)
+        new_results = gemini_service.search_companies(query, exclude_names=current_names)
         
-        existing_keys = set(get_key(p) for p in current_search_results)
-        for p in new_results:
-            if get_key(p) not in existing_keys:
-                current_search_results.append(p)
+        existing_keys = set(get_key(c) for c in current_results)
+        for c in new_results:
+            if get_key(c) not in existing_keys:
+                current_results.append(c)
                 
-        return format_df(current_search_results, current_saved), current_search_results
+        return format_df(current_results, current_saved), current_results
     except Exception as e:
         raise gr.Error(f"載入失敗: {e}")
 
-def select_professor_from_df(evt: gr.SelectData, search_results, saved_data, view_mode):
+def select_company(evt: gr.SelectData, search_results, saved_data, view_mode):
     if not evt: return [gr.update()] * 8
     index = evt.index[0]
     
@@ -145,69 +145,71 @@ def select_professor_from_df(evt: gr.SelectData, search_results, saved_data, vie
     if not target_list or index >= len(target_list): 
         return gr.update(), gr.update(), gr.update(), None, None, gr.update(), gr.update(), gr.update()
     
-    prof = target_list[index]
+    comp = target_list[index]
     
-    key = get_key(prof)
-    saved_prof = next((p for p in saved_data if get_key(p) == key), None)
-    current_prof = saved_prof if saved_prof else prof
+    key = get_key(comp)
+    saved_comp = next((c for c in saved_data if get_key(c) == key), None)
+    current_comp = saved_comp if saved_comp else comp
     
     details_md = ""
     
-    if current_prof.get('details') and len(current_prof.get('details')) > 10:
-        details_md = current_prof['details']
-        if not saved_prof: 
-            saved_data.insert(0, current_prof)
+    # Check Cache
+    if current_comp.get('details') and len(current_comp.get('details')) > 10:
+        details_md = current_comp['details']
+        if not saved_comp: 
+            saved_data.insert(0, current_comp)
             save_data(saved_data)
     else:
-        gr.Info(f"正在調查 {current_prof['name']}...")
+        # Call API
+        gr.Info(f"正在調查 {current_comp['name']} (查詢統編、PTT評價)...")
         try:
-            res = gemini_service.get_professor_details(current_prof)
-            current_prof['details'] = res['text']
-            current_prof['sources'] = res['sources']
+            res = gemini_service.get_company_details(current_comp)
+            current_comp['details'] = res['text']
+            current_comp['sources'] = res['sources']
             details_md = res['text']
             
-            if saved_prof:
-                saved_prof.update(current_prof)
+            if saved_comp:
+                saved_comp.update(current_comp)
             else:
-                saved_data.insert(0, current_prof)
+                saved_data.insert(0, current_comp)
             save_data(saved_data)
         except Exception as e:
             raise gr.Error(f"調查失敗: {e}")
 
-    if current_prof.get('sources'):
-        details_md += "\n\n### 📚 參考來源\n"
-        for s in current_prof['sources']:
+    if current_comp.get('sources'):
+        details_md += "\n\n### 📚 資料來源\n"
+        for s in current_comp['sources']:
             details_md += f"- [{s['title']}]({s['uri']})\n"
 
     return (
         gr.update(visible=True),    
         details_md,                 
         [],                         
-        current_prof,               
+        current_comp,               
         saved_data,                 
-        get_tags_text(current_prof),           
-        gr.update(choices=get_tags_choices(current_prof), value=None), 
+        get_tags_text(current_comp),           
+        gr.update(choices=get_tags_choices(current_comp), value=None), 
         gr.update(visible=True)     
     )
 
-def add_tag(new_tag, selected_prof, saved_data, view_mode, search_results):
-    if not selected_prof or not new_tag: 
+def add_tag(new_tag, selected_comp, saved_data, view_mode, search_results):
+    if not selected_comp or not new_tag: 
         return gr.update(), gr.update(), gr.update(), saved_data, gr.update()
 
-    if 'tags' not in selected_prof: selected_prof['tags'] = []
+    if 'tags' not in selected_comp: selected_comp['tags'] = []
     
-    if new_tag not in selected_prof['tags']:
-        selected_prof['tags'].append(new_tag)
+    if new_tag not in selected_comp['tags']:
+        selected_comp['tags'].append(new_tag)
         
-        key = get_key(selected_prof)
+        key = get_key(selected_comp)
         found = False
-        for i, p in enumerate(saved_data):
-            if get_key(p) == key:
-                saved_data[i] = selected_prof
+        for i, c in enumerate(saved_data):
+            if get_key(c) == key:
+                saved_data[i] = selected_comp
                 found = True
                 break
         if not found:
-            saved_data.insert(0, selected_prof)
+            saved_data.insert(0, selected_comp)
         
         save_data(saved_data)
         gr.Info(f"已新增標籤: {new_tag}")
@@ -217,23 +219,23 @@ def add_tag(new_tag, selected_prof, saved_data, view_mode, search_results):
 
     return (
         gr.update(value=""), 
-        get_tags_text(selected_prof),
-        gr.update(choices=selected_prof['tags']),
+        get_tags_text(selected_comp),
+        gr.update(choices=selected_comp['tags']),
         saved_data,
         new_df
     )
 
-def remove_tag(tag_to_remove, selected_prof, saved_data, view_mode, search_results):
-    if not selected_prof or not tag_to_remove: 
+def remove_tag(tag_to_remove, selected_comp, saved_data, view_mode, search_results):
+    if not selected_comp or not tag_to_remove: 
         return gr.update(), gr.update(), saved_data, gr.update()
     
-    if 'tags' in selected_prof and tag_to_remove in selected_prof['tags']:
-        selected_prof['tags'].remove(tag_to_remove)
+    if 'tags' in selected_comp and tag_to_remove in selected_comp['tags']:
+        selected_comp['tags'].remove(tag_to_remove)
         
-        key = get_key(selected_prof)
-        for i, p in enumerate(saved_data):
-            if get_key(p) == key:
-                saved_data[i] = selected_prof
+        key = get_key(selected_comp)
+        for i, c in enumerate(saved_data):
+            if get_key(c) == key:
+                saved_data[i] = selected_comp
                 break
         save_data(saved_data)
         gr.Info(f"已移除標籤: {tag_to_remove}")
@@ -242,15 +244,15 @@ def remove_tag(tag_to_remove, selected_prof, saved_data, view_mode, search_resul
     new_df = format_df(target_list, saved_data)
     
     return (
-        get_tags_text(selected_prof),
-        gr.update(choices=selected_prof['tags'], value=None),
+        get_tags_text(selected_comp),
+        gr.update(choices=selected_comp['tags'], value=None),
         saved_data,
         new_df
     )
 
-def chat_response(history, message, selected_prof):
-    if not selected_prof: return history, ""
-    context = selected_prof.get('details', '')
+def chat_response(history, message, selected_comp):
+    if not selected_comp: return history, ""
+    context = selected_comp.get('details', '')
     if not context: return history, ""
     
     service_history = []
@@ -265,26 +267,26 @@ def chat_response(history, message, selected_prof):
         history.append((message, f"Error: {e}"))
     return history, ""
 
-def update_status(status, selected_prof, saved_data, view_mode, search_results):
-    if not selected_prof: return gr.update(), saved_data
+def update_status(status, selected_comp, saved_data, view_mode, search_results):
+    if not selected_comp: return gr.update(), saved_data
     
-    selected_prof['status'] = status if selected_prof.get('status') != status else None
+    selected_comp['status'] = status if selected_comp.get('status') != status else None
     
-    key = get_key(selected_prof)
-    for i, p in enumerate(saved_data):
-        if get_key(p) == key:
-            saved_data[i] = selected_prof
+    key = get_key(selected_comp)
+    for i, c in enumerate(saved_data):
+        if get_key(c) == key:
+            saved_data[i] = selected_comp
             break
     save_data(saved_data)
     
     target_list = saved_data if view_mode == "追蹤清單" else search_results
     return format_df(target_list, saved_data), saved_data
 
-def remove_prof(selected_prof, saved_data, view_mode, search_results):
-    if not selected_prof: return gr.update(), gr.update(value=None), saved_data, gr.update(visible=False)
+def remove_comp(selected_comp, saved_data, view_mode, search_results):
+    if not selected_comp: return gr.update(), gr.update(value=None), saved_data, gr.update(visible=False)
     
-    key = get_key(selected_prof)
-    new_saved = [p for p in saved_data if get_key(p) != key]
+    key = get_key(selected_comp)
+    new_saved = [c for c in saved_data if get_key(c) != key]
     save_data(new_saved)
     
     target_list = new_saved if view_mode == "追蹤清單" else search_results
@@ -308,21 +310,33 @@ def init_on_load():
 
 # --- UI Layout ---
 
-with gr.Blocks(title="Prof.404 開箱教授去哪兒？", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Com.404 台企天眼通", theme=gr.themes.Soft()) as demo:
     
     saved_state = gr.State([])
     search_res_state = gr.State([])
-    selected_prof_state = gr.State(None)
+    selected_comp_state = gr.State(None)
 
-    # 🌟 這裡插入了您要求的徽章與文字，使用 HTML 置中
+    # 🌟 Com.404 專屬 Header
     gr.Markdown("""
     <div align="center">
     
-    # 🎓 Prof.404 - 開箱教授去哪兒？    
-    **學術研究啟程的導航系統，拒絕當科研路上的無頭蒼蠅** | **API Rate limits 是 RPD 20，故建議自行 Fork使用**</span>    
+    # 🏢 Com.404 - 台企天眼通 (Company Scout)
+    
+    [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/DeepLearning101/Com.404) &nbsp;
+    [![GitHub](https://img.shields.io/badge/GitHub-Repo-black)](https://github.com/Deep-Learning-101/prof-404) &nbsp;
+    [![Powered by](https://img.shields.io/badge/Powered%20by-Gemini%202.0%20Flash-4285F4?logo=google)](https://deepmind.google/technologies/gemini/)
+    
+    **查統編、看資本額、搜 PTT/Dcard 評價、掃描勞資糾紛，一鍵完成。**
+    <span style="font-size: 0.9em; color: gray;">(支援雲端同步！Space 重啟資料不遺失 🔄 | API KEY RPD，建議自行 Fork)</span>
+    
     👉 歡迎 Star [GitHub](https://github.com/Deep-Learning-101/prof-404) ⭐ 覺得不錯 👈
-
+    </div>
+    
+    ---
+    
+    <div align="center">
     <h3>🧠 補腦專區：<a href="https://deep-learning-101.github.io/" target="_blank">Deep Learning 101</a></h3>
+    </div>
     
     | 🔥 技術傳送門 (Tech Stack) | 📚 必讀心法 (Must Read) |
     | :--- | :--- |
@@ -330,11 +344,10 @@ with gr.Blocks(title="Prof.404 開箱教授去哪兒？", theme=gr.themes.Soft()
     | 📝 [**自然語言處理 (NLP)**](https://deep-learning-101.github.io/Natural-Language-Processing) | 📊 [**評測篇：臺灣 LLM 分析**](https://deep-learning-101.github.io/Blog/TW-LLM-Benchmark) |
     | 👁️ [**電腦視覺 (CV)**](https://deep-learning-101.github.io//Computer-Vision) | 🛠️ [**實戰篇：打造高精準 RAG**](https://deep-learning-101.github.io/RAG) |
     | 🎤 [**語音處理 (Speech)**](https://deep-learning-101.github.io/Speech-Processing) | 🕳️ [**避坑篇：AI Agent 開發陷阱**](https://deep-learning-101.github.io/agent) |
-    </div>
     """)
     
     with gr.Row():
-        search_input = gr.Textbox(label="搜尋研究領域", placeholder="例如: 大型語言模型, 後量子密碼遷移...", scale=4)
+        search_input = gr.Textbox(label="輸入公司名稱或統編", placeholder="例如: 台積電, 2330, 八方雲集...", scale=4)
         search_btn = gr.Button("🔍 搜尋", variant="primary", scale=1)
     
     with gr.Row():
@@ -343,11 +356,11 @@ with gr.Blocks(title="Prof.404 開箱教授去哪兒？", theme=gr.themes.Soft()
     with gr.Row():
         # Left: List
         with gr.Column(scale=1):
-            prof_df = gr.Dataframe(
-                headers=["狀態", "姓名", "大學", "系所", "標籤"],
-                datatype=["str", "str", "str", "str", "str"],
+            comp_df = gr.Dataframe(
+                headers=["狀態", "公司名稱", "產業類別", "標籤"],
+                datatype=["str", "str", "str", "str"],
                 interactive=False,
-                label="教授列表 (點擊查看詳情)"
+                label="公司列表 (點擊查看詳情)"
             )
             load_more_btn = gr.Button("載入更多", visible=False)
         
@@ -355,92 +368,98 @@ with gr.Blocks(title="Prof.404 開箱教授去哪兒？", theme=gr.themes.Soft()
         with gr.Column(scale=2, visible=False) as details_col:
             detail_md = gr.Markdown("詳細資料...")
             
-            # Status Buttons
-            with gr.Row():
-                btn_match = gr.Button("✅ 符合")
-                btn_mismatch = gr.Button("❌ 不符")
-                btn_pending = gr.Button("❓ 待觀察")
-                btn_remove = gr.Button("🗑️ 移除", variant="stop")
-            
+            # 🌟 Chat Section (位於中間)
+            with gr.Column(elem_classes="chat-section"):
+                gr.Markdown("### 🤖 商業分析師 (已閱讀下方報告)")
+                chatbot = gr.Chatbot(height=250, type="messages")
+                with gr.Row():
+                    msg = gr.Textbox(label="提問", placeholder="例如：這間公司有勞資糾紛嗎？薪資結構如何？", scale=4)
+                    send_btn = gr.Button("送出", scale=1)
+
             gr.Markdown("---")
-            
-            # Tags Management
+
+            # Tags & Status
             with gr.Column(visible=False) as tags_row:
                 tags_display = gr.Markdown("目前標籤: (無)")
                 with gr.Row():
-                    tag_input = gr.Textbox(label="新增標籤", placeholder="輸入後按新增...", scale=3)
+                    tag_input = gr.Textbox(label="新增標籤", placeholder="例如: 面試過, 薪水高...", scale=3)
                     tag_add_btn = gr.Button("➕ 新增", scale=1)
                 
                 with gr.Accordion("刪除標籤", open=False):
                     with gr.Row():
                         tag_dropdown = gr.Dropdown(label="選擇標籤", choices=[], scale=3)
                         tag_del_btn = gr.Button("🗑️ 刪除", scale=1, variant="secondary")
-
-            gr.Markdown("---")
-            gr.Markdown("### 💬 AI 助手")
-            chatbot = gr.Chatbot(height=300)
-            msg = gr.Textbox(label="提問")
-            send_btn = gr.Button("送出")
+            
+            with gr.Row():
+                btn_good = gr.Button("✅ 優質")
+                btn_risk = gr.Button("⚠️ 風險")
+                btn_pending = gr.Button("❓ 未定")
+                btn_remove = gr.Button("🗑️ 移除", variant="stop")
 
     # --- Wiring ---
     
-    demo.load(init_on_load, inputs=None, outputs=[saved_state, prof_df])
+    # Init
+    demo.load(init_on_load, inputs=None, outputs=[saved_state, comp_df])
 
+    # Search & Load More
     search_btn.click(
-        search_professors, 
-        inputs=[search_input, saved_state],
-        outputs=[prof_df, search_res_state, load_more_btn]
+        search_companies, 
+        inputs=[search_input, saved_state], 
+        outputs=[comp_df, search_res_state, load_more_btn]
     ).then(
         lambda: gr.update(value="搜尋結果"), outputs=[view_radio]
     )
     
     load_more_btn.click(
-        load_more,
-        inputs=[search_input, search_res_state, saved_state],
-        outputs=[prof_df, search_res_state]
+        load_more, 
+        inputs=[search_input, search_res_state, saved_state], 
+        outputs=[comp_df, search_res_state]
     )
     
     view_radio.change(
-        toggle_view,
-        inputs=[view_radio, search_res_state, saved_state],
-        outputs=[prof_df, load_more_btn]
+        toggle_view, 
+        inputs=[view_radio, search_res_state, saved_state], 
+        outputs=[comp_df, load_more_btn]
     )
     
-    prof_df.select(
-        select_professor_from_df,
-        inputs=[search_res_state, saved_state, view_radio],
+    # Selection
+    comp_df.select(
+        select_company, 
+        inputs=[search_res_state, saved_state, view_radio], 
         outputs=[
-            details_col, detail_md, chatbot, selected_prof_state, saved_state, 
+            details_col, detail_md, chatbot, selected_comp_state, saved_state, 
             tags_display, tag_dropdown, tags_row
         ]
     )
     
-    send_btn.click(chat_response, inputs=[chatbot, msg, selected_prof_state], outputs=[chatbot, msg])
-    msg.submit(chat_response, inputs=[chatbot, msg, selected_prof_state], outputs=[chatbot, msg])
+    # Chat
+    send_btn.click(chat_response, inputs=[chatbot, msg, selected_comp_state], outputs=[chatbot, msg])
+    msg.submit(chat_response, inputs=[chatbot, msg, selected_comp_state], outputs=[chatbot, msg])
     
+    # Tags
     tag_add_btn.click(
-        add_tag,
-        inputs=[tag_input, selected_prof_state, saved_state, view_radio, search_res_state],
-        outputs=[tag_input, tags_display, tag_dropdown, saved_state, prof_df]
+        add_tag, 
+        inputs=[tag_input, selected_comp_state, saved_state, view_radio, search_res_state], 
+        outputs=[tag_input, tags_display, tag_dropdown, saved_state, comp_df]
     )
-    
     tag_del_btn.click(
-        remove_tag,
-        inputs=[tag_dropdown, selected_prof_state, saved_state, view_radio, search_res_state],
-        outputs=[tags_display, tag_dropdown, saved_state, prof_df]
+        remove_tag, 
+        inputs=[tag_dropdown, selected_comp_state, saved_state, view_radio, search_res_state], 
+        outputs=[tags_display, tag_dropdown, saved_state, comp_df]
     )
-
-    for btn, status in [(btn_match, 'match'), (btn_mismatch, 'mismatch'), (btn_pending, 'pending')]:
+    
+    # Status & Remove
+    for btn, status in [(btn_good, 'good'), (btn_risk, 'risk'), (btn_pending, 'pending')]:
         btn.click(
-            update_status,
-            inputs=[gr.State(status), selected_prof_state, saved_state, view_radio, search_res_state],
-            outputs=[prof_df, saved_state]
+            update_status, 
+            inputs=[gr.State(status), selected_comp_state, saved_state, view_radio, search_res_state], 
+            outputs=[comp_df, saved_state]
         )
-
+    
     btn_remove.click(
-        remove_prof,
-        inputs=[selected_prof_state, saved_state, view_radio, search_res_state],
-        outputs=[gr.State(None), prof_df, saved_state, details_col]
+        remove_comp, 
+        inputs=[selected_comp_state, saved_state, view_radio, search_res_state], 
+        outputs=[gr.State(None), comp_df, saved_state, details_col]
     )
 
 if __name__ == "__main__":