Update app.py

app.py

@@ -23,7 +23,7 @@ def generate_prompts(user_input, csv_prompts, prefix, num_examples=3, language='
     system_prompt = (
         f"你是一位自然诗人，请将以下内容转化为{'英文' if language == 'English' else '中文'}的风景描写。"
         "请确保风格与提供的 CSV 中的描述一致，并以统一前缀 '{prefix}' 开头。"
-        "请生成 {num_examples} 条不同风格的结果，每条结果应独立成段。"
+        "请生成 {num_examples} 条不同风格的结果，每条结果不要加编号，且保持简洁自然。"
     )
 
     # 拼接用户输入和 CSV 内容
@@ -43,14 +43,22 @@ def generate_prompts(user_input, csv_prompts, prefix, num_examples=3, language='
     raw_output = response.choices[0].message.content.strip()
     prompts = [line.strip() for line in raw_output.split('\n') if line.strip()]
 
-    # 取前 num_examples 个有效结果
+    # 限制数量
     prompts = prompts[:num_examples]
 
-    # 处理后文本：每行以指定前缀开头
-    processed_prompts = [f"{prefix} {p}" if not p.startswith(prefix) else p for p in prompts]
+    # 处理后文本：每行以指定前缀开头（先判断是否已包含）
+    processed_prompts = []
+    for p in prompts:
+        if prefix not in p:
+            processed_prompts.append(f"{prefix} {p}")
+        else:
+            processed_prompts.append(p)
+
+    # 输出格式
+    output_text = "\n\n".join(prompts)
     processed_text = "\n".join(processed_prompts)
 
-    return "\n\n".join(prompts), processed_text
+    return output_text, processed_text
 
 # Gradio 界面定义
 with gr.Blocks(title="提示词生成器", theme=gr.themes.Soft()) as app: