Update app.py

app.py

@@ -91,37 +91,37 @@ user_input = st.text_input(
 if user_input:
        
 
-    # # 1. 生成剧本
-    # with st.status("🖋️ 剧本生成中...", expanded=True) as status:
-    #     try:
-    #         # 生成内容
-    #         prompt = build_script_prompt(user_input)
-    #         response = script_pipe(
-    #             prompt,
-    #             max_length=600,
-    #             temperature=0.8,
-    #             num_beams=4,
-    #             no_repeat_ngram_size=3
-    #         )[0]["generated_text"]
+    # 1. 生成剧本
+    with st.status("🖋️ 剧本生成中...", expanded=True) as status:
+        try:
+            # 生成内容
+            prompt = build_script_prompt(user_input)
+            response = script_pipe(
+                prompt,
+                max_length=600,
+                temperature=0.8,
+                num_beams=4,
+                no_repeat_ngram_size=3
+            )[0]["generated_text"]
                     
-    #         # 后处理
-    #         def format_script(text):
-    #             # 强制修正场景标题格式（如 "场景1" -> "### 场景1"）
-    #             text = re.sub(r"场景(\d+)", r"### 场景\1", text)
-    #             # 删除多余符号
-    #             text = re.sub(r"[<>()]{2,}|\[?\]?", "", text)
-    #             # 确保动作描述包裹在方括号内
-    #             return re.sub(r"(?<=\n)\s*([^\[\n]+?)\s*(?=\n)", r"[\1]", text)
+            # 后处理
+            def format_script(text):
+                # 强制修正场景标题格式（如 "场景1" -> "### 场景1"）
+                text = re.sub(r"场景(\d+)", r"### 场景\1", text)
+                # 删除多余符号
+                text = re.sub(r"[<>()]{2,}|\[?\]?", "", text)
+                # 确保动作描述包裹在方括号内
+                return re.sub(r"(?<=\n)\s*([^\[\n]+?)\s*(?=\n)", r"[\1]", text)
                     
-    #         # 显示结果
-    #         st.subheader("生成剧本")
-    #         st.markdown(f"```markdown\n{format_script(response)}\n```")
-    #         status.update(label="✅ 生成完成", state="complete")
+            # 显示结果
+            st.subheader("生成剧本")
+            st.markdown(f"```markdown\n{format_script(response)}\n```")
+            status.update(label="✅ 生成完成", state="complete")
                     
-    #     except Exception as e:
-    #         status.update(label="❌ 生成失败", state="error")
-    #         st.error(f"错误详情：{str(e)}")
-    #         st.stop()
+        except Exception as e:
+            status.update(label="❌ 生成失败", state="error")
+            st.error(f"错误详情：{str(e)}")
+            st.stop()
 
     
     #2. 生成分镜
@@ -142,9 +142,9 @@ if user_input:
             
             # 生成分镜
             storyboard = storyboard_pipe(
-                f"将以下剧本转换为分镜JSON，严格遵循模板：\n{json_template}\n剧本内容：{script[:800]}",
+                f"将以下剧本转换为分镜JSON，严格遵循模板：\n{json_template}\n剧本内容：{cleaned[:800]}",  # 关键修改点
                 max_length=1500,
-                temperature=0.3  # 降低创造性保证格式
+                temperature=0.3
             )[0]["generated_text"]
             
             # 格式修复流程