Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 import math
+import html as py_html  # ✅ 为了转义全文文本
 import gradio as gr
 from gradio_client import Client
 
@@ -139,7 +140,6 @@ body, .gradio-container {
     justify-content: center !important;
     transition: all 0.2s ease !important;
 }
-/* Hover */
 .arrow-btn:hover {
     color: var(--primary-color) !important; 
     transform: scale(1.1);
@@ -162,6 +162,32 @@ body, .gradio-container {
     border-color: #cbd5e0 !important;
 }
 
+/* 全文导出区域 */
+.fulltext-wrapper { padding: 10px 0; }
+.fulltext-copy-btn {
+    background: #4f46e5;
+    color: #fff;
+    border: none;
+    border-radius: 8px;
+    padding: 6px 12px;
+    font-size: 0.9rem;
+    cursor: pointer;
+    margin-bottom: 10px;
+}
+.fulltext-copy-btn:hover {
+    filter: brightness(1.08);
+}
+.fulltext-area {
+    width: 100%;
+    min-height: 500px;
+    border-radius: 8px;
+    border: 1px solid #e2e8f0;
+    padding: 10px;
+    font-family: 'JetBrains Mono', monospace;
+    font-size: 0.9rem;
+    resize: vertical;
+}
+
 /* 移动端适配 */
 @media (max-width: 768px) {
     .book-page-container { padding: 0; height: auto; min-height: 500px; }
@@ -174,17 +200,18 @@ body, .gradio-container {
 # 2. 逻辑处理
 # ==========================================
 
-def paginate_story(story_data, chars_per_page=600):
+def paginate_story(story_data, chars_per_page=1200, first_page_ratio=0.8):
     """
-    分章 -> 分页。同时记录每一章的起始页码，方便章节跳转。
+    分章 -> 分页。
+    - 首章首页可用字符数 = chars_per_page * first_page_ratio（因为有大标题和空白）
+    - 之后的每一页都用 chars_per_page，尽量“填满”固定高度。
     返回: (flat_pages, chapter_start_indices)
     """
     if not story_data:
         return [], []
 
     flat_pages = []
-    chapter_start_indices = [] # 记录 Chapter i 的起始页 index
-    
+    chapter_start_indices = []
     current_global_page_index = 0
 
     for chapter in story_data:
@@ -193,17 +220,20 @@ def paginate_story(story_data, chars_per_page=600):
         title = chapter.get("title", "")
         content = chapter.get("content", "")
         paragraphs = content.split('\n')
-        
+
         current_page_content = []
         current_char_count = 0
-        limit = chars_per_page // 1.5 
         is_start = True
-        
-        # --- 分页算法 ---
+
+        # 首页稍微少一点，为标题留空间
+        limit = int(chars_per_page * first_page_ratio)
+
         for p in paragraphs:
             p = p.strip()
-            if not p: continue
-            
+            if not p:
+                continue
+
+            # 如果当前段落加进去会超过限制，并且当前页已有内容 -> 换页
             if current_char_count + len(p) > limit and current_page_content:
                 flat_pages.append({
                     "title": title if is_start else "",
@@ -211,16 +241,18 @@ def paginate_story(story_data, chars_per_page=600):
                     "is_chapter_start": is_start,
                     "chapter_title": title
                 })
-                current_global_page_index += 1 # 全局页码 +1
+                current_global_page_index += 1
 
+                # 新的一页
                 current_page_content = [p]
                 current_char_count = len(p)
                 is_start = False
-                limit = chars_per_page
+                limit = chars_per_page  # 之后的页按照标准容量
             else:
                 current_page_content.append(p)
                 current_char_count += len(p)
-        
+
+        # 收尾
         if current_page_content:
             flat_pages.append({
                 "title": title if is_start else "",
@@ -232,6 +264,7 @@ def paginate_story(story_data, chars_per_page=600):
 
     return flat_pages, chapter_start_indices
 
+
 def render_book_page(flat_pages, page_index):
     # 1. 空状态
     if not flat_pages or len(flat_pages) == 0:
@@ -245,20 +278,20 @@ def render_book_page(flat_pages, page_index):
             </div>
         </div>
         """
-    
-    # 2. 渲染
+
+    # 2. 渲染正常页面
     total_pages = len(flat_pages)
     page_index = max(0, min(page_index, total_pages - 1))
     page_data = flat_pages[page_index]
-    
+
     is_start = page_data.get("is_chapter_start", False)
     display_title = page_data.get("title", "")
     chapter_ref = page_data.get("chapter_title", "")
     content = page_data.get("content", "")
-    
+
     paragraphs = [f"<p>{p}</p>" for p in content.split('\n') if p.strip()]
     content_html = "".join(paragraphs)
-    
+
     if is_start:
         header_html = f"""
         <div class="page-top-spacer"></div>
@@ -290,47 +323,105 @@ def render_book_page(flat_pages, page_index):
     """
     return html
 
+
+def build_full_text(story_data):
+    """
+    把章节列表拼成一整本小说文本，供导出/复制。
+    story_data: List[{"title": ..., "content": ...}, ...]
+    """
+    if not story_data:
+        return ""
+
+    blocks = []
+    for ch in story_data:
+        title = ch.get("title", "").strip()
+        content = ch.get("content", "").strip()
+        if title:
+            blocks.append(title)
+        if content:
+            blocks.append(content)
+    return "\n\n".join(blocks).strip()
+
+
+def build_full_text_html(full_text: str) -> str:
+    """
+    生成带「一键复制」按钮的 HTML。
+    使用 textarea + JS 调用 navigator.clipboard.writeText。
+    """
+    escaped = py_html.escape(full_text or "")
+    return f"""
+    <div class="fulltext-wrapper">
+        <button class="fulltext-copy-btn" onclick="
+            const ta = document.getElementById('full-story-area');
+            if (ta) {{
+                ta.select();
+                ta.setSelectionRange(0, 999999);
+                navigator.clipboard && navigator.clipboard.writeText(ta.value);
+            }}
+        ">
+            📋 一键复制全文
+        </button>
+        <textarea id="full-story-area" class="fulltext-area">{escaped}</textarea>
+    </div>
+    """
+
 # ==========================================
 # 3. 后端连接
 # ==========================================
 def bridge_to_backend(premise):
     if not premise.strip():
-        yield "⚠️ 请输入故事梗概...", None, None, None, [], [], render_book_page([], 0)
+        empty_full_html = build_full_text_html("")
+        yield "⚠️ 请输入故事梗概...", None, None, None, [], [], empty_full_html, render_book_page([], 0)
         return
 
     log_buffer = "🚀 初始化前端连接...\n"
     initial_html = render_book_page([], 0)
-    
-    # 初始状态 [log, outline, plan, personas, pages, chap_indices, html]
-    yield log_buffer, None, None, None, [], [], initial_html
+    empty_full_html = build_full_text_html("")
+
+    # 初始状态 [log, outline, plan, personas, pages, chap_indices, full_text_html, book_html]
+    yield log_buffer, None, None, None, [], [], empty_full_html, initial_html
 
     try:
         log_buffer += f"🔗 连接后端 Space: {PRIVATE_SPACE_ID}...\n"
         client = Client(PRIVATE_SPACE_ID, hf_token=HF_TOKEN)
         job = client.submit(premise, api_name="/generate_novel")
-        
+
         for result in job:
-            # result: [0]Log, [1]Outline, [2]Plan, [3]Personas, [4]StoryList
+            # result: [0]Log, [1]Outline, [2]Plan, [3]Personas, [4]StoryList(章节列表)
             backend_log = result[0]
             outline = result[1]
             plan = result[2]
             personas_html = result[3]
-            raw_story_list = result[4] 
-            
-            # 分页 + 计算章节索引
-            flat_pages, chap_indices = paginate_story(raw_story_list, chars_per_page=600)
+            raw_story_list = result[4]
+
+            # 分页 + 计算章节索引（让页面尽量填满）
+            flat_pages, chap_indices = paginate_story(
+                raw_story_list,
+                chars_per_page=1200,
+                first_page_ratio=0.8,
+            )
+
+            # 一键复制全文的文本 + HTML
+            full_text = build_full_text(raw_story_list)
+            full_text_html = build_full_text_html(full_text)
+
             book_html = render_book_page(flat_pages, 0)
-            
-            yield backend_log, outline, plan, personas_html, flat_pages, chap_indices, book_html
+
+            yield backend_log, outline, plan, personas_html, flat_pages, chap_indices, full_text_html, book_html
 
     except Exception as e:
         error_msg = f"❌ 前端连接错误: {str(e)}"
-        yield error_msg, None, None, None, [], [], render_book_page([], 0)
+        error_full_html = build_full_text_html(str(e))
+        yield error_msg, None, None, None, [], [], error_full_html, render_book_page([], 0)
 
 # ==========================================
 # 4. 前端 UI 布局
 # ==========================================
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"), css=custom_css, title="LongStory Agent") as demo:
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"),
+    css=custom_css,
+    title="LongStory Agent"
+) as demo:
     
     # --- 状态管理 ---
     story_pages_state = gr.State([])      # 所有的页面 List[Dict]
@@ -359,7 +450,13 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"),
                     Story Premise
                 </div>
                 """)
-                premise_input = gr.Textbox(label="Premise", show_label=False, lines=5, elem_classes=["input-box"], placeholder="输入故事创意...")
+                premise_input = gr.Textbox(
+                    label="Premise",
+                    show_label=False,
+                    lines=5,
+                    elem_classes=["input-box"],
+                    placeholder="输入故事创意..."
+                )
                 
                 # Examples
                 gr.HTML('<div class="examples-container"><div class="examples-label">⚡ 快速开始 (Quick Inspirations)</div></div>')
@@ -379,8 +476,20 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"),
                 
                 submit_btn = gr.Button("✨ 开始生成 (GENERATE)", elem_classes=["generate-btn"])
                 
-                gr.HTML('<div class="terminal-wrapper"><div class="terminal-header"><div class="dot dot-red"></div><div class="dot dot-yellow"></div><div class="dot dot-green"></div><div class="terminal-title">system.log</div></div>')
-                log_output = gr.Textbox(label="Log", lines=10, interactive=False, elem_classes=["terminal-log"], show_label=False, value="> System initialized...")
+                gr.HTML(
+                    '<div class="terminal-wrapper"><div class="terminal-header">'
+                    '<div class="dot dot-red"></div><div class="dot dot-yellow"></div>'
+                    '<div class="dot dot-green"></div><div class="terminal-title">system.log</div>'
+                    '</div>'
+                )
+                log_output = gr.Textbox(
+                    label="Log",
+                    lines=10,
+                    interactive=False,
+                    elem_classes=["terminal-log"],
+                    show_label=False,
+                    value="> System initialized..."
+                )
                 gr.HTML("</div>")
 
         # === 右侧：内容展示 ===
@@ -397,17 +506,18 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"),
                         
                         # 核心内容
                         with gr.Column(scale=15):
-                            story_display = gr.HTML(label="Book View", value=render_book_page([], 0))
+                            story_display = gr.HTML(
+                                label="Book View",
+                                value=render_book_page([], 0)
+                            )
                         
                         # 右侧箭头
                         with gr.Column(scale=1, min_width=40, elem_classes=["arrow-col"]):
                             btn_next_page = gr.Button("›", elem_classes=["arrow-btn"])
                     
-                    # 2. 章节导航
+                    # 章节导航
                     with gr.Row(elem_classes=["chapter-nav-row"]):
                         btn_prev_chap = gr.Button("⏮️ 上一章", elem_classes=["chapter-nav-btn"])
-                        # 占位符，把按钮挤到两边或中间
-                        # gr.Spacer() 
                         btn_next_chap = gr.Button("⏭️ 下一章", elem_classes=["chapter-nav-btn"])
 
                 # Tab 2: 大纲
@@ -422,6 +532,10 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"),
                 with gr.TabItem("👥 人物档案", id="tab-persona"):
                     persona_output = gr.HTML(label="Character Cards")
 
+                # Tab 5: 全文导出
+                with gr.TabItem("📄 全文导出", id="tab-fulltext"):
+                    full_text_html = gr.HTML(label="Full Story Export")
+
     # ==========================================
     # 5. 事件交互
     # ==========================================
@@ -430,56 +544,78 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"),
     submit_btn.click(
         fn=bridge_to_backend,
         inputs=[premise_input],
-        outputs=[log_output, outline_output, plan_output, persona_output, story_pages_state, chapter_indices_state, story_display],
+        outputs=[
+            log_output,
+            outline_output,
+            plan_output,
+            persona_output,
+            story_pages_state,
+            chapter_indices_state,
+            full_text_html,      # ✅ 新增：全文导出 HTML
+            story_display
+        ],
         concurrency_limit=1
     )
 
     # B. 翻页 (上一页/下一页)
     def on_prev_page(pages, current_idx):
-        if not pages: return 0, render_book_page([], 0)
+        if not pages:
+            return 0, render_book_page([], 0)
         new_idx = max(0, current_idx - 1)
         return new_idx, render_book_page(pages, new_idx)
 
     def on_next_page(pages, current_idx):
-        if not pages: return 0, render_book_page([], 0)
+        if not pages:
+            return 0, render_book_page([], 0)
         new_idx = min(len(pages) - 1, current_idx + 1)
         return new_idx, render_book_page(pages, new_idx)
 
-    btn_prev_page.click(fn=on_prev_page, inputs=[story_pages_state, current_page_state], outputs=[current_page_state, story_display])
-    btn_next_page.click(fn=on_next_page, inputs=[story_pages_state, current_page_state], outputs=[current_page_state, story_display])
+    btn_prev_page.click(
+        fn=on_prev_page,
+        inputs=[story_pages_state, current_page_state],
+        outputs=[current_page_state, story_display]
+    )
+    btn_next_page.click(
+        fn=on_next_page,
+        inputs=[story_pages_state, current_page_state],
+        outputs=[current_page_state, story_display]
+    )
 
     # C. 章节跳转 (上��章/下一章)
     def on_prev_chap(pages, current_idx, chap_indices):
-        if not pages or not chap_indices: return current_idx, render_book_page(pages, current_idx)
-        
-        # 找到当前页属于第几章 (找到最后一个 <= current_idx 的 start_index)
-        # 例如 chap_indices = [0, 5, 12], current = 6. 属于索引为1的章节(index 5)
-        # 我们要跳到索引为0的章节(index 0)
-        
+        if not pages or not chap_indices:
+            return current_idx, render_book_page(pages, current_idx)
+
         target_idx = 0
-        # 简单遍历寻找目标
         for start_idx in reversed(chap_indices):
             if start_idx < current_idx:
                 target_idx = start_idx
                 break
-        
+
         return target_idx, render_book_page(pages, target_idx)
 
     def on_next_chap(pages, current_idx, chap_indices):
-        if not pages or not chap_indices: return current_idx, render_book_page(pages, current_idx)
-        
+        if not pages or not chap_indices:
+            return current_idx, render_book_page(pages, current_idx)
+
         target_idx = current_idx
-        # 找到第一个 > current_idx 的 start_index
         for start_idx in chap_indices:
             if start_idx > current_idx:
                 target_idx = start_idx
                 break
-        
-        # 如果没找到(已经是最后一章)，保持不变或跳到最后一页，这里保持不变
+
         return target_idx, render_book_page(pages, target_idx)
 
-    btn_prev_chap.click(fn=on_prev_chap, inputs=[story_pages_state, current_page_state, chapter_indices_state], outputs=[current_page_state, story_display])
-    btn_next_chap.click(fn=on_next_chap, inputs=[story_pages_state, current_page_state, chapter_indices_state], outputs=[current_page_state, story_display])
+    btn_prev_chap.click(
+        fn=on_prev_chap,
+        inputs=[story_pages_state, current_page_state, chapter_indices_state],
+        outputs=[current_page_state, story_display]
+    )
+    btn_next_chap.click(
+        fn=on_next_chap,
+        inputs=[story_pages_state, current_page_state, chapter_indices_state],
+        outputs=[current_page_state, story_display]
+    )
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch()