Update app.py

app.py

@@ -101,6 +101,10 @@ custom_css = """
     padding: 2px 5px;
     border-radius: 3px;
     background-color: #fff;
+    cursor: pointer;
+}
+.tag-item:hover {
+    background-color: #f0f0f0;
 }
 .tag-en {
     font-weight: bold;
@@ -117,16 +121,90 @@ custom_css = """
 .btn-container {
     margin-top: 20px;
 }
+.copy-btn {
+    margin-top: 10px;
+    padding: 5px 10px;
+    background-color: #f0f0f0;
+    border: 1px solid #ddd;
+    border-radius: 4px;
+    cursor: pointer;
+    display: inline-flex;
+    align-items: center;
+    font-size: 0.9em;
+}
+.copy-btn:hover {
+    background-color: #e0e0e0;
+}
+.copy-icon {
+    margin-right: 5px;
+    width: 16px;
+    height: 16px;
+}
+.copied-message {
+    display: none;
+    color: #4CAF50;
+    margin-left: 10px;
+    font-size: 0.9em;
+}
+.note-text {
+    color: #ff6b6b;
+    font-size: 0.9em;
+    padding: 5px;
+    border-left: 3px solid #ff6b6b;
+    margin-top: 15px;
+    background-color: #fff5f5;
+}
 """
 
-with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=custom_css) as demo:
+js_code = """
+function setupCopyFunctions() {
+    // 为标签项设置点击复制
+    document.querySelectorAll('.tag-item').forEach(item => {
+        item.addEventListener('click', function() {
+            const tagText = this.querySelector('.tag-en').textContent;
+            navigator.clipboard.writeText(tagText).then(() => {
+                // 显示临时复制成功提示
+                const msg = document.createElement('span');
+                msg.textContent = '已复制!';
+                msg.style.color = '#4CAF50';
+                msg.style.marginLeft = '5px';
+                msg.style.fontSize = '0.8em';
+                this.appendChild(msg);
+                setTimeout(() => msg.remove(), 1000);
+            });
+        });
+    });
+
+    // 为汇总区域的复制按钮设置功能
+    document.getElementById('copy-tags-btn').addEventListener('click', function() {
+        const tagsText = document.getElementById('summary-text').value;
+        navigator.clipboard.writeText(tagsText).then(() => {
+            const copiedMsg = document.getElementById('copied-message');
+            copiedMsg.style.display = 'inline';
+            setTimeout(() => {
+                copiedMsg.style.display = 'none';
+            }, 2000);
+        });
+    });
+}
+
+// 在DOM加载完成或更新后调用设置函数
+function onUiUpdate() {
+    setupCopyFunctions();
+}
+
+document.addEventListener('DOMContentLoaded', onUiUpdate);
+"""
+
+with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=custom_css, js=js_code) as demo:
     gr.Markdown("# 🖼️ AI 图像标签分析器")
     gr.Markdown("上传图片自动识别标签，并可一键翻译成中文")
+    gr.Markdown("<div class='note-text'>⚠️ 注意：角色识别仅支持推测2024年2月以前的角色</div>", elem_id="character-notice")
 
     with gr.Row():
         with gr.Column(scale=1):
             img_in = gr.Image(type="pil", label="上传图片")
-            with gr.Accordion("⚙️ 高级设置", open=True):
+            with gr.Accordion("⚙️ 高级设置", open=False):
                 gen_slider  = gr.Slider(0, 1, 0.35,
                                         label="通用标签阈值", info="越高→标签更少更准")
                 char_slider = gr.Slider(0, 1, 0.85,
@@ -153,26 +231,42 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
                     out_rating = gr.HTML(label="Rating Tags")
             
             gr.Markdown("### 标签汇总")
-            out_summary = gr.Textbox(label="标签汇总", 
-                                     placeholder="选择需要汇总的标签类别...",
-                                     lines=3)
+            with gr.Row():
+                out_summary = gr.Textbox(label="标签汇总", 
+                                         placeholder="选择需要汇总的标签类别...",
+                                         lines=3,
+                                         elem_id="summary-text")
+            
+            # 添加复制按钮的HTML
+            copy_btn_html = gr.HTML("""
+            <div style="display: flex; align-items: center;">
+                <button id="copy-tags-btn" class="copy-btn">
+                    <svg class="copy-icon" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+                        <rect x="9" y="9" width="13" height="13" rx="2" ry="2"></rect>
+                        <path d="M5 15H4a2 2 0 01-2-2V4a2 2 0 012-2h9a2 2 0 012 2v1"></path>
+                    </svg>
+                    复制标签
+                </button>
+                <span id="copied-message" class="copied-message">已复制！</span>
+            </div>
+            """)
 
     # ----------------- 处理回调 -----------------
     def format_tags_html(tags_dict, translations, show_translation=True):
-        """格式化标签为HTML格式"""
+        """格式化标签为HTML格式，添加点击复制功能"""
         if not tags_dict:
             return "<p>暂无标签</p>"
         
         html = '<div class="label-container">'
         for i, (tag, score) in enumerate(tags_dict.items()):
-            tag_html = f'<div class="tag-item">'
-            tag_html += f'<div><span class="tag-en">{tag}</span>'
+            # 添加可点击复制的标签项
+            html += f'<div class="tag-item" title="点击复制标签">'
+            html += f'<div><span class="tag-en">{tag}</span>'
             if show_translation and i < len(translations):
-                tag_html += f'<span class="tag-zh">({translations[i]})</span>'
-            tag_html += '</div>'
-            tag_html += f'<span class="tag-score">{score:.3f}</span>'
-            tag_html += '</div>'
-            html += tag_html
+                html += f'<span class="tag-zh">({translations[i]})</span>'
+            html += '</div>'
+            html += f'<span class="tag-score">{score:.3f}</span>'
+            html += '</div>'
         html += '</div>'
         return html
 
@@ -223,33 +317,22 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             char_html = format_tags_html(res["characters"], translations_dict["characters"], show_zh)
             rating_html = format_tags_html(res["ratings"], translations_dict["ratings"], show_zh)
 
-            # 生成汇总文本
+            # 生成汇总文本 - 修改为仅显示英文标签，无注释
             summary_parts = []
             separators = {"逗号": ", ", "换行": "\n", "空格": " "}
             separator = separators[sep_type]
             
+            all_tags = []
             if sum_gen and res["general"]:
-                if show_zh and translations_dict["general"]:
-                    gen_tags = [f"{en}({zh})" for en, zh in zip(res["general"].keys(), translations_dict["general"])]
-                else:
-                    gen_tags = list(res["general"].keys())
-                summary_parts.append("通用标签: " + separator.join(gen_tags))
+                all_tags.extend(list(res["general"].keys()))
             
             if sum_char and res["characters"]:
-                if show_zh and translations_dict["characters"]:
-                    char_tags = [f"{en}({zh})" for en, zh in zip(res["characters"].keys(), translations_dict["characters"])]
-                else:
-                    char_tags = list(res["characters"].keys())
-                summary_parts.append("角色标签: " + separator.join(char_tags))
+                all_tags.extend(list(res["characters"].keys()))
             
             if sum_rat and res["ratings"]:
-                if show_zh and translations_dict["ratings"]:
-                    rat_tags = [f"{en}({zh})" for en, zh in zip(res["ratings"].keys(), translations_dict["ratings"])]
-                else:
-                    rat_tags = list(res["ratings"].keys())
-                summary_parts.append("评分标签: " + separator.join(rat_tags))
+                all_tags.extend(list(res["ratings"].keys()))
             
-            summary_text = "\n\n".join(summary_parts) if summary_parts else "请选择要汇总的标签类别"
+            summary_text = separator.join(all_tags) if all_tags else "请选择要汇总的标签类别"
 
             # 完成处理，返回最终结果
             yield (