Update app.py

app.py

@@ -0,0 +1,163 @@
+
+PRIVATE_SPACE_ID = "Yoyo2004/Longstory-backend" 
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
+custom_css = """
+@import url('https://fonts.googleapis.com/css2?family=Noto+Sans+SC:wght@300;400;700&family=Noto+Serif+SC:wght@400;700&display=swap');
+
+body { font-family: 'Noto Sans SC', sans-serif !important; }
+
+.container { max-width: 1200px; margin: 0 auto; }
+
+/* 标题样式 */
+.header-title {
+    font-family: 'Noto Serif SC', serif;
+    font-weight: 700;
+    font-size: 2.5rem;
+    color: #2d3436;
+    margin-bottom: 0.5rem;
+}
+.header-subtitle {
+    color: #636e72;
+    font-size: 1.1rem;
+    margin-bottom: 2rem;
+}
+
+/* 输入框美化 */
+textarea {
+    border: 2px solid #dfe6e9 !important;
+    border-radius: 12px !important;
+    transition: all 0.3s ease;
+    font-size: 16px !important;
+}
+textarea:focus {
+    border-color: #74b9ff !important;
+    box-shadow: 0 0 10px rgba(116, 185, 255, 0.2) !important;
+}
+
+/* 按钮美化 */
+#generate-btn {
+    background: linear-gradient(135deg, #6c5ce7, #a29bfe) !important;
+    border: none;
+    color: white;
+    font-weight: bold;
+    font-size: 18px;
+    padding: 12px 24px;
+    border-radius: 30px;
+    box-shadow: 0 4px 15px rgba(108, 92, 231, 0.4);
+    transition: transform 0.2s;
+}
+#generate-btn:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 6px 20px rgba(108, 92, 231, 0.6);
+}
+
+/* 输出区域卡片化 */
+.output-card {
+    background: white;
+    border: 1px solid #f1f2f6;
+    border-radius: 16px;
+    padding: 20px;
+    box-shadow: 0 4px 20px rgba(0,0,0,0.03);
+}
+
+/* 小说正文阅读模式 */
+.story-content {
+    font-family: 'Noto Serif SC', serif;
+    line-height: 1.8;
+    font-size: 18px;
+    color: #2d3436;
+    background-color: #fafafa;
+    padding: 40px;
+    border-radius: 8px;
+    border-left: 5px solid #6c5ce7;
+}
+"""
+
+# ==========================================
+# 客户端连接逻辑
+# ==========================================
+def bridge_to_backend(premise):
+    """
+    连接到私有 Space 并流式传输结果
+    """
+    if not HF_TOKEN:
+        yield "⚠️ Error: HF_TOKEN not found in Secrets.", None, None, None, None
+        return
+
+    try:
+        # 连接私有 Space
+        client = Client(PRIVATE_SPACE_ID, hf_token=HF_TOKEN)
+        
+        # 提交任务 (使用 generate_novel 函数)
+        # 注意：这里的 fn_index 或 api_name 取决于后端 Gradio 的自动命名，
+        # 通常如果后端只有一个函数，可以直接用 submit
+        job = client.submit(premise, api_name="/generate_novel")
+        
+        # 迭代获取生成器产生的数据
+        for result in job:
+            # result 是一个元组，对应后端的 outputs
+            # (log, outline, plan, persona, story)
+            yield result
+
+    except Exception as e:
+        yield f"❌ Connection Error: {str(e)}", None, None, None, None
+
+
+# ==========================================
+# 构建前端界面
+# ==========================================
+with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="LongStory AI") as demo:
+    
+    with gr.Row(elem_classes=["container"]):
+        with gr.Column(scale=1):
+            gr.Markdown("# 🖊️ LongStory AI", elem_classes=["header-title"])
+            gr.Markdown("智能长篇小说辅助创作系统 · Deep Layout & Persona Driven", elem_classes=["header-subtitle"])
+    
+    with gr.Row(elem_classes=["container"]):
+        # 左侧控制栏
+        with gr.Column(scale=1, min_width=300):
+            with gr.Group():
+                premise_input = gr.Textbox(
+                    label="💡 小说核心梗概 (Premise)", 
+                    placeholder="例如：构造一个长篇青春校园小说, 事件覆盖从刚刚入学报道到高考之前...",
+                    lines=6
+                )
+                submit_btn = gr.Button("✨ 开始创作 (Generate)", elem_id="generate-btn")
+            
+            # 日志区
+            with gr.Accordion("构建日志 (System Logs)", open=True):
+                log_output = gr.Textbox(label="实时进度", lines=10, interactive=False, show_label=False)
+
+        # 右侧展示栏
+        with gr.Column(scale=2, min_width=500):
+            with gr.Tabs():
+                # Tab 1: 最终正文
+                with gr.TabItem("📖 正文草稿 (Draft)"):
+                    story_output = gr.Markdown(
+                        value="等待生成...", 
+                        elem_classes=["story-content"]
+                    )
+                
+                # Tab 2: 大纲
+                with gr.TabItem("🗺️ 故事大纲 (Outline)"):
+                    outline_output = gr.JSON(label="事件流 (Event Flow)")
+                
+                # Tab 3: 细纲
+                with gr.TabItem("📅 详细规划 (Planning)"):
+                    plan_output = gr.JSON(label="子事件 (Sub-events)")
+                
+                # Tab 4: 人物
+                with gr.TabItem("👥 人物设定 (Personas)"):
+                    persona_output = gr.JSON(label="人物弧光 (Character Arc)")
+
+    # 事件绑定
+    submit_btn.click(
+        bridge_to_backend,
+        inputs=[premise_input],
+        outputs=[log_output, outline_output, plan_output, persona_output, story_output]
+    )
+
+if __name__ == "__main__":
+    demo.queue().launch()