Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-import webbrowser # 用於開啟網頁地圖連結
 
 # --- 模型配置 (CPU TinyLlama) ---
+# 這部分保持不變
 MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 print("Loading TinyLlama model and tokenizer...")
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
@@ -11,6 +11,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 print("TinyLlama model and tokenizer loaded.")
 
 # --- 文字生成函數 ---
+# 這部分保持不變
 def generate_text(prompt, max_new_tokens, temperature, top_p):
     print(f"Generating TinyLlama response for prompt: '{prompt}'")
     chat = [{"role": "user", "content": prompt}]
@@ -25,30 +26,39 @@ def generate_text(prompt, max_new_tokens, temperature, top_p):
             top_p=float(top_p),
             eos_token_id=tokenizer.eos_token_id
         )
-    decoded_output = tokenizer.decode(outputs[:, inputs["input_ids"].shape[-1]:][0], skip_special_tokens=True)
-    print(f"Generated response: {decoded_output}")
-    return decoded_output.strip()
+    # 修正解碼方式以只回傳模型生成的部分
+    response = tokenizer.decode(outputs[0, inputs["input_ids"].shape[-1]:], skip_special_tokens=True)
+    print(f"Generated response: {response}")
+    return response.strip()
 
-# --- 地圖生成函數 (開啟 Google 地圖網頁) ---
+# --- 地圖生成函數 (回傳 Google 地圖網址) ---
+# 我們將修改此函數，使其回傳一個可點擊的連結，而不是試圖開啟新分頁
 def show_map(location_query):
-    """開啟包含指定地點的 Google 地圖網頁。"""
+    """根據地點名稱生成一個 Google 地圖的 URL"""
+    if not location_query:
+        return "請輸入一個地點！"
     base_url = "https://www.google.com/maps/search/?api=1&query="
-    map_url = base_url + location_query.replace(" ", "+")
-    webbrowser.open_new_tab(map_url)
-    return f"已在您的預設瀏覽器中開啟 {location_query} 的地圖。"
+    # 將地點名稱進行 URL 編碼，避免特殊字元問題
+    from urllib.parse import quote
+    map_url = base_url + quote(location_query)
+    # 使用 Markdown 格式回傳一個可點擊的連結
+    return f"點擊這裡查看 **{location_query}** 的地圖：\n[在 Google 地圖中開啟]({map_url})"
 
-# --- Gradio 介面 ---
+# --- Gradio 介面 (修正版) ---
+# 使用 gr.Blocks 和 gr.Tabs 來建立介面
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 多功能應用：文字生成與地圖顯示")
+    gr.Markdown("# ✨ 多功能應用：文字生成與地圖顯示")
     gr.Markdown("這個應用程式可以生成文字，也可以根據您輸入的地點顯示地圖。")
 
-    with gr.TabbedInterface(["文字生成", "顯示地圖"]) as tabbed:
+    # 使用 gr.Tabs 建立分頁容器
+    with gr.Tabs():
+        # 第一個分頁：文字生成
         with gr.TabItem("文字生成"):
             gr.Markdown("## 使用 TinyLlama 生成文字")
             with gr.Column():
                 prompt_input = gr.Textbox(
                     label="您的提示詞",
-                    placeholder="請輸入您想讓模型接續的文字..."
+                    placeholder="例如：請用繁體中文解釋什麼是光合作用？"
                 )
                 with gr.Accordion("生成參數", open=False):
                     max_new_tokens_slider = gr.Slider(minimum=50, maximum=512, value=256, step=1, label="最大生成 Token 數")
@@ -56,25 +66,29 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     top_p_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-P (Nucleus Sampling)")
                 generate_button = gr.Button("生成文字 ✨", variant="primary")
                 output_text = gr.Textbox(label="模型的回應", lines=10, interactive=False)
-                generate_button.click(
-                    fn=generate_text,
-                    inputs=[prompt_input, max_new_tokens_slider, temperature_slider, top_p_slider],
-                    outputs=output_text
-                )
 
+        # 第二個分頁：顯示地圖
         with gr.TabItem("顯示地圖"):
             gr.Markdown("## 顯示指定地點的地圖")
             location_input = gr.Textbox(
                 label="輸入地點名稱",
                 placeholder="例如：台北 101, New York Central Park, 東京鐵塔"
             )
-            show_map_button = gr.Button("顯示地圖 🗺️", variant="primary")
-            map_output = gr.Textbox(label="地圖顯示訊息", interactive=False)
-            show_map_button.click(
-                fn=show_map,
-                inputs=location_input,
-                outputs=map_output
-            )
+            show_map_button = gr.Button("取得地圖連結 🗺️", variant="primary")
+            # 輸出元件改為 gr.Markdown 以便顯示可點擊的連結
+            map_output = gr.Markdown(label="地圖連結")
+
+    # 將元件的事件監聽器放在 gr.Tabs 之外
+    generate_button.click(
+        fn=generate_text,
+        inputs=[prompt_input, max_new_tokens_slider, temperature_slider, top_p_slider],
+        outputs=output_text
+    )
+    show_map_button.click(
+        fn=show_map,
+        inputs=location_input,
+        outputs=map_output
+    )
 
 # --- 啟動 Gradio 應用 ---
 demo.launch()