Update app.py

app.py

@@ -559,9 +559,40 @@ def get_demo_path(filename):
     return filename if os.path.exists(filename) else None
 
 
-# 构建 Gradio 界面
+# # 构建 Gradio 界面
+# with gr.Blocks(title="AI 剧本视频工厂") as demo:
+#     gr.Markdown("# 🎬 ScriptAgent & Sora/Veo 视频生成工坊 (注意：仅供简单测试，由于成本问题在线平台内存只有18G，我们量化了模型，性能效果并不能保证，如果需要最准确的输出请自行部署即可)")
+    
+#     with gr.Tabs():
+#         # --- TAB 1: 剧本创作 ---
+#         with gr.Tab("📝 第一步：剧本创作"):
+#             with gr.Row():
+#                 with gr.Column():
+#                     llm_input = gr.Textbox(
+#                         label="剧情输入", 
+#                         placeholder="主角：你在做什么？...", 
+#                         lines=6
+#                     )
+#                     llm_btn = gr.Button("生成/续写剧本", variant="primary")
+                
+#                 with gr.Column():
+#                     llm_output = gr.Textbox(
+#                         label="生成的剧本", 
+#                         lines=10, 
+#                         interactive=True
+#                     )
+#                     to_video_btn = gr.Button("⬇️ 发送到视频生成", variant="secondary")
+            
+#             gr.Examples(
+#                 [[
+#                     "主角：你在做什么？指引图是不能随便篡改的。\n"
+#                     "蒋前：篡改指引图？不不不，你误会了。\n"
+#                     "蒋前：我才加入风物家没多久，哪有这个本事能篡改它..."
+#                 ]], 
+#                 inputs=llm_input
+#             )
 with gr.Blocks(title="AI 剧本视频工厂") as demo:
-    gr.Markdown("# 🎬 ScriptAgent & Sora/Veo 视频生成工坊")
+    gr.Markdown("# 🎬 ScriptAgent & Sora/Veo 视频生成工坊 (注意：仅供简单测试，由于成本问题在线平台内存只有18G，我们量化了模型，性能效果并不能保证，如果需要最准确的输出请自行部署即可)")
     
     with gr.Tabs():
         # --- TAB 1: 剧本创作 ---
@@ -591,7 +622,73 @@ with gr.Blocks(title="AI 剧本视频工厂") as demo:
                 ]], 
                 inputs=llm_input
             )
+            
+            # 🔥 新增：本地部署代码展示区
+            gr.Markdown("---")
+            with gr.Accordion("💻 本地部署完整代码（点击展开查看）", open=False):
+                gr.Markdown("""
+                ### 📦 完整部署步骤
+                以下代码可在本地完整运行，获得最佳性能和输出质量：
+                """)
+                
+                deployment_code = '''import os
+from huggingface_hub import snapshot_download
+
+os.environ['CUDA_VISIBLE_DEVICES'] = '0'
+
+model_name = "XD-MU/ScriptAgent"
+local_path = "./models/ScriptAgent"
+
+# 下载整个仓库的所有文件
+print("下载模型所有文件...")
+snapshot_download(
+    repo_id=model_name,
+    local_dir=local_path,
+    local_dir_use_symlinks=False,  # 直接复制文件，不使用符号链接
+    resume_download=True  # 支持断点续传
+)
 
+print(f"模型已完整下载到: {local_path}")
+
+# 然后使用 SWIFT 加载
+from swift.llm import PtEngine, RequestConfig, InferRequest
+
+engine = PtEngine(local_path, max_batch_size=1)
+request_config = RequestConfig(max_tokens=8192, temperature=0.7)
+
+infer_request = InferRequest(messages=[
+    {"role": "user", "content": "袁雪：太感谢了！幸好有你们及时赶到！否则现在我们已经变成烤肉大餐了……\\n主角：伤得怎么样，还站得起来吗？\\n赵东旭：腿一点知觉都没有……\\n蒙犽：实力不够就别逞强接甲级委托啊！跟我们走，回营地找百草家治疗。\\n蒙犽：你怎么了，发什么呆呢？\\n主角：这一带不太对劲。气温依然炎热，地面的火焰也没有消退……难道还有漏网之鱼？\\n主角：而且……我担心前面还有被困的学生。不如再去前面探查一下。\\n蒙犽：知道了。那分头行动吧，我先护送伤员去营地,回头再来找你。\\n蒙犽：哎，你可别跑太远啊！待会大眼枭联系！"}
+])
+response = engine.infer([infer_request], request_config)[0]
+
+print(response.choices[0].message.content)'''
+                
+                gr.Code(
+                    value=deployment_code,
+                    language="python",
+                    label="deploy_scriptagent.py",
+                    lines=35,
+                    interactive=False
+                )
+                
+                gr.Markdown("""
+                ### 📌 环境要求
+                ```bash
+                # 安装依赖
+                pip install ms-swift[llm] transformers torch huggingface_hub
+                
+                # GPU 推荐配置
+                - CUDA 11.8+
+                - 显存: 16GB+ (推荐 24GB)
+                - 内存: 32GB+
+                ```
+                
+                ### ⚡ 本地部署优势
+                - ✅ 完整精度模型，无量化损失
+                - ✅ 更快的推理速度
+                - ✅ 无网络限制，支持离线运行
+                - ✅ 可自定义参数（temperature, max_tokens等）
+                """)
         # --- TAB 2: 视频生成 ---
         with gr.Tab("🎥 第二步：视频生成"):
             with gr.Row():