Update app.py

app.py

@@ -3,9 +3,9 @@ import os
 import shutil
 import json
 import torch
-import re
+# import re
 import requests
-import transformers
+# import transformers
 import chardet
 import deepeval
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -540,7 +540,7 @@ def time_tool(query: str) -> str:
         location = "London"
 
     location_key = location.lower()
-    tz_str = zone_map.get(location_key, "Europe/London")
+    tz_str = location_to_timezone(location)
     now = datetime.now(ZoneInfo(tz_str))
 
     # return time or date
@@ -1269,6 +1269,7 @@ def detect_intent_embedding(query, file_names=[]):
                 best_label = label
     return best_label if best_label else "General"
 
+# Router 條件邏輯：你可以換成你自己的邏輯規則
 def decide_next(state):
     query = state.get("query", "")
     file_names = state.get("file_names", [])
@@ -1306,14 +1307,6 @@ def build_langgraph_gpt_like():
     graph.add_node("Summarise", summarizer_run)
 
     graph.set_entry_point("Router")
-
-    # Router 條件邏輯：你可以換成你自己的邏輯規則
-    def decide_next(state):
-        query = state.get("query", "")
-        file_names = state.get("file_names", [])
-        label = detect_intent_embedding(query, file_names)
-        return label
-
     graph.add_conditional_edges("Router", decide_next, {
         "DocQA": "DocQA",
         "General": "General",