diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py
index 71f751e..a264b3c 100644
--- a/frontend/dataset_generate_page.py
+++ b/frontend/dataset_generate_page.py
@@ -1,6 +1,7 @@
 import gradio as gr
 import sys
 from pathlib import Path
+from langchain.prompts import PromptTemplate
 
 sys.path.append(str(Path(__file__).resolve().parent.parent))
 from global_var import get_docs, get_prompt_store
@@ -45,6 +46,11 @@ def dataset_generate_page():
 
         def on_prompt_change(selected_prompt):
             print(f"模板选择已更改为: {selected_prompt}")
+            selected_prompt_id = int(selected_prompt.split(" ")[0])
+            selected_prompt = get_prompt_store().get(doc_id=selected_prompt_id)
+            selected_prompt = selected_prompt["content"]
+            selected_prompt = PromptTemplate.from_template(selected_prompt)
+            input_variables_list = selected_prompt.input_variables
             return selected_prompt
 
         doc_dropdown.change(on_doc_change, inputs=doc_dropdown, outputs=doc_state)