Create app.py

app.py

@@ -0,0 +1,323 @@
+import gradio as gr
+import json
+import os
+import pandas as pd
+from dotenv import load_dotenv
+from services import GeminiService
+
+# Load Env
+load_dotenv()
+SAVE_FILE = os.getenv("SAVE_FILE_NAME", "saved_professors.json")
+
+# Init Service
+try:
+    gemini_service = GeminiService()
+except Exception as e:
+    print(f"Service Error: {e}")
+    gemini_service = None
+
+# --- Helper Functions ---
+
+def get_key(p):
+    return f"{p['name']}-{p['university']}"
+
+def load_data():
+    if os.path.exists(SAVE_FILE):
+        try:
+            with open(SAVE_FILE, 'r', encoding='utf-8') as f:
+                return json.load(f)
+        except:
+            return []
+    return []
+
+def save_data(data):
+    try:
+        with open(SAVE_FILE, 'w', encoding='utf-8') as f:
+            json.dump(data, f, ensure_ascii=False, indent=2)
+    except Exception as e:
+        print(f"Save Error: {e}")
+
+# 將教授列表轉換為 Pandas DataFrame 以供顯示
+def format_df(prof_list):
+    if not prof_list:
+        return pd.DataFrame(columns=["狀態", "姓名", "大學", "系所", "標籤"])
+    
+    data = []
+    for p in prof_list:
+        status_map = {'match': '✅', 'mismatch': '❌', 'pending': '❓'}
+        status_icon = status_map.get(p.get('status'), '')
+        has_detail = "📄" if p.get('details') else ""
+        
+        tags = ", ".join(p.get('tags', []))
+        
+        data.append([
+            f"{status_icon} {has_detail}",
+            p['name'],
+            p['university'],
+            p['department'],
+            tags
+        ])
+    return pd.DataFrame(data, columns=["狀態", "姓名", "大學", "系所", "標籤"])
+
+# --- Event Handlers ---
+
+def search_professors(query, current_saved):
+    if not query: return gr.update(), current_saved, gr.update()
+    
+    try:
+        results = gemini_service.search_professors(query)
+        # 搜尋結果不存檔，只暫存在 State 中
+        # 但為了顯示，我們需要跟 saved 做比對 (merge logic)
+        return format_df(results), results, gr.update(visible=True) # Show Load More
+    except Exception as e:
+        raise gr.Error(f"搜尋失敗: {e}")
+
+def load_more(query, current_search_results):
+    if not query: return gr.update(), current_search_results
+    
+    current_names = [p['name'] for p in current_search_results]
+    try:
+        new_results = gemini_service.search_professors(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)
+                
+        return format_df(current_search_results), current_search_results
+    except Exception as e:
+        raise gr.Error(f"載入失敗: {e}")
+
+def select_professor_from_df(evt: gr.SelectData, search_results, saved_data):
+    # evt.index[0] 是被點擊的行數
+    index = evt.index[0]
+    
+    # 判斷目前是在看「搜尋結果」還是「追蹤清單」
+    # 這裡簡化邏輯：Gradio 比較難做 View Switcher，我們假設列表是混合顯示或單一顯示
+    # 為了簡單，我們這裡假設使用者點的是目前的顯示列表
+    
+    # 我們需要知道目前顯示的是哪一份資料。
+    # 簡單做法：搜尋後，search_results 會更新。
+    # 我們先假設使用者是在點擊 search_results (如果沒搜尋，就顯示 saved)
+    
+    target_list = search_results if search_results else saved_data
+    if index >= len(target_list): return gr.update(), gr.update(), gr.update(), None, None
+    
+    prof = 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
+    
+    # 準備回傳
+    details_md = ""
+    chat_history = []
+    
+    # 檢查是否有 Details
+    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)
+            save_data(saved_data)
+    else:
+        # 需要 Call API
+        gr.Info(f"正在調查 {current_prof['name']}...")
+        try:
+            res = gemini_service.get_professor_details(current_prof)
+            current_prof['details'] = res['text']
+            current_prof['sources'] = res['sources']
+            details_md = res['text']
+            
+            # 更新存檔
+            if saved_prof:
+                saved_prof.update(current_prof)
+            else:
+                saved_data.insert(0, current_prof)
+            save_data(saved_data)
+        except Exception as e:
+            raise gr.Error(f"調查失敗: {e}")
+
+    # 格式化 Sources
+    if current_prof.get('sources'):
+        details_md += "\n\n### 📚 參考來源\n"
+        for s in current_prof['sources']:
+            details_md += f"- [{s['title']}]({s['uri']})\n"
+
+    return (
+        gr.update(visible=True), # Detail Column
+        details_md,
+        [], # Reset Chat
+        current_prof, # Update selected state
+        saved_data # Update saved state
+    )
+
+def chat_response(history, message, selected_prof):
+    if not selected_prof: return history, ""
+    
+    context = selected_prof.get('details', '')
+    if not context: return history, ""
+    
+    # 轉換 history 給 service 用
+    # Gradio history is [[user, bot], [user, bot]]
+    service_history = []
+    for h in history:
+        service_history.append({"role": "user", "content": h[0]})
+        if h[1]:
+            service_history.append({"role": "model", "content": h[1]})
+            
+    try:
+        reply = gemini_service.chat_with_ai(service_history, message, context)
+        history.append((message, reply))
+    except Exception as e:
+        history.append((message, f"Error: {e}"))
+        
+    return history, ""
+
+def update_status(status, selected_prof, saved_data):
+    if not selected_prof: return gr.update(), saved_data
+    
+    selected_prof['status'] = status if selected_prof.get('status') != status else None
+    
+    # Update in saved list
+    key = get_key(selected_prof)
+    for i, p in enumerate(saved_data):
+        if get_key(p) == key:
+            saved_data[i] = selected_prof
+            break
+    save_data(saved_data)
+    
+    return gr.Info(f"狀態已更新: {status}"), saved_data
+
+def remove_prof(selected_prof, saved_data):
+    if not selected_prof: 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]
+    save_data(new_saved)
+    
+    return (
+        gr.Info("已移除"), 
+        format_df(new_saved), # Update DF
+        new_saved, 
+        gr.update(visible=False) # Hide Details
+    )
+
+def toggle_view(mode, search_res, saved_data):
+    if mode == "搜尋結果":
+        return format_df(search_res), gr.update(visible=True)
+    else:
+        return format_df(saved_data), gr.update(visible=False) # Hide load more
+
+# --- UI Layout ---
+
+with gr.Blocks(title="開箱教授去哪兒？", theme=gr.themes.Soft()) as demo:
+    
+    # State
+    saved_state = gr.State(load_data())
+    search_res_state = gr.State([])
+    selected_prof_state = gr.State(None)
+    
+    gr.Markdown("# Prof.404 🎓 開箱教授去哪兒？ (Gradio Edition)")
+
+    with gr.Row():
+        search_input = gr.Textbox(label="搜尋研究領域", placeholder="例如: LLM, 量子計算...", scale=4)
+        search_btn = gr.Button("🔍 搜尋", variant="primary", scale=1)
+    
+    with gr.Row():
+        view_radio = gr.Radio(["搜尋結果", "追蹤清單"], label="顯示模式", value="搜尋結果")
+    
+    with gr.Row():
+        # Left: List
+        with gr.Column(scale=1):
+            prof_df = gr.Dataframe(
+                headers=["狀態", "姓名", "大學", "系所", "標籤"],
+                datatype=["str", "str", "str", "str", "str"],
+                interactive=False,
+                label="教授列表 (點擊查看詳情)"
+            )
+            load_more_btn = gr.Button("載入更多", visible=False)
+        
+        # Right: Details
+        with gr.Column(scale=2, visible=False) as details_col:
+            detail_md = gr.Markdown("詳細資料...")
+            
+            with gr.Row():
+                btn_match = gr.Button("✅ 符合")
+                btn_mismatch = gr.Button("❌ 不符")
+                btn_pending = gr.Button("❓ 待觀察")
+                btn_remove = gr.Button("🗑️ 移除", variant="stop")
+            
+            gr.Markdown("### 💬 AI 助手")
+            chatbot = gr.Chatbot(height=300)
+            msg = gr.Textbox(label="提問")
+            send_btn = gr.Button("送出")
+
+    # --- Wiring ---
+    
+    # Search
+    search_btn.click(
+        search_professors, 
+        inputs=[search_input, saved_state],
+        outputs=[prof_df, search_res_state, load_more_btn]
+    )
+    
+    # Load More
+    load_more_btn.click(
+        load_more,
+        inputs=[search_input, search_res_state],
+        outputs=[prof_df, search_res_state]
+    )
+    
+    # View Toggle
+    view_radio.change(
+        toggle_view,
+        inputs=[view_radio, search_res_state, saved_state],
+        outputs=[prof_df, load_more_btn]
+    )
+    
+    # Select Row
+    prof_df.select(
+        select_professor_from_df,
+        inputs=[search_res_state, saved_state],
+        outputs=[details_col, detail_md, chatbot, selected_prof_state, saved_state]
+    )
+    
+    # Chat
+    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]
+    )
+    
+    # Status Buttons
+    # Note: clicking status needs to refresh the dataframe to show the icon update
+    def refresh_current_view(mode, search_r, saved_d):
+        return toggle_view(mode, search_r, saved_d)[0] # return only df
+
+    for btn, status in [(btn_match, 'match'), (btn_mismatch, 'mismatch'), (btn_pending, 'pending')]:
+        btn.click(
+            update_status,
+            inputs=[gr.State(status), selected_prof_state, saved_state],
+            outputs=[gr.State(None), saved_state] # Info is side effect
+        ).then(
+            refresh_current_view,
+            inputs=[view_radio, search_res_state, saved_state],
+            outputs=[prof_df]
+        )
+
+    # Remove
+    btn_remove.click(
+        remove_prof,
+        inputs=[selected_prof_state, saved_state],
+        outputs=[gr.State(None), prof_df, saved_state, details_col]
+    )
+
+if __name__ == "__main__":
+    demo.launch()