Update app.py

app.py

@@ -111,12 +111,6 @@ if user_input and style_or_opening:
 
             # Post - processing
             def format_script(text):
-                # # Force - correct the scene title format (e.g., "Scene 1" -> "### Scene 1")
-                # text = re.sub(r"Scene(\d+)", r"### Scene\1", text)
-                # # Remove extra symbols
-                # text = re.sub(r"[<>()]{2,}|\[?\]?", "", text)
-                # # Ensure action descriptions are wrapped in square brackets
-                # return re.sub(r"(?<=\n)\s*([^\[\n]+?)\s*(?=\n)", r"[\1]", text)
                 return text
 
             # Display the result
@@ -131,9 +125,31 @@ if user_input and style_or_opening:
             st.stop()
 
     # 2. Generate the storyboard
-    with st.status("🎥 Converting to a storyboard script...", expanded=True) as status:
+    with st.status("🎥 Converting to storyboard script...", expanded=True) as status:
         try:
-            # JSON template guidance
+            # 示例数据模板
+            sample_storyboard = """
+            {
+              "scenes": [
+                {
+                  "scene_number": 1,
+                  "location": "Abandoned Space Station (INT. Dystopian Future)",
+                  "characters": ["Eve", "R2-D9"],
+                  "camera_angle": "Wide shot showing decaying corridors",
+                  "action": "Eve scans walls with her holographic projector, R2-D9 beeps nervously"
+                },
+                {
+                  "scene_number": 2,
+                  "location": "Alien Wasteland (EXT. Toxic Sunset)",
+                  "characters": ["Eve", "Mutant Leader"],
+                  "camera_angle": "Low-angle shot of Eve aiming plasma rifle",
+                  "action": "Mutant leader growls, acid rain hisses on Eve's protective suit"
+                }
+              ]
+            }
+            """
+
+            # JSON模板引导
             json_template = """{
               "scenes": [
                 {
@@ -146,31 +162,33 @@ if user_input and style_or_opening:
               ]
             }"""
 
-            # Generate the storyboard
+            # 生成分镜
             storyboard = storyboard_pipe(
-                f"Convert the following script into a storyboard JSON, strictly following the template:\n{json_template}\nScript content: {cleaned[:800]}",  # Key modification point
+                f"Convert to storyboard JSON: {cleaned[:800]}",
                 max_length=1500,
                 temperature=0.3
             )[0]["generated_text"]
 
-            # Format repair process
-            storyboard = storyboard.replace("'", '"')  # Unify quotes
-            storyboard = re.search(r'\{.*\}', storyboard, re.DOTALL).group()  # Extract JSON
-
-            # Parse and validate
+            # 增强的格式修复
+            storyboard = (
+                storyboard.replace("'", '"')
+                .replace("None", '""')
+                .replace("True", "true")
+                .replace("False", "false")
+            )
+            storyboard = re.search(r'\{.*\}', storyboard, re.DOTALL).group()
             parsed = json.loads(storyboard)
-            status.update(label="Storyboard generation completed ✅", state="complete")
-            st.subheader("Storyboard Script")
-            st.json(parsed)
 
-        except json.JSONDecodeError:
-            status.update(label="Storyboard parsing failed ❌", state="error")
-            st.warning("Automatically fixing the storyboard format...")
-            storyboard = "{" + storyboard.split("{", 1)[-1].rsplit("}", 1)[0] + "}"
-            st.code(storyboard)
         except Exception as e:
-            status.update(label="Storyboard generation error ❌", state="error")
-            st.error(f"Storyboard generation error: {str(e)}")
+            st.warning(f"Using sample storyboard due to error: {str(e)}")
+            parsed = json.loads(sample_storyboard)  # 回退到示例数据
+            status.update(label="⚠️ Using sample storyboard", state="warning")
+
+        finally:
+            status.update(label="Storyboard ready", state="complete")
+            st.subheader("Storyboard Script")
+            st.json(parsed)
+            st.caption("*Note: If using sample data, edit the JSON before proceeding*")
 
 
     # 3. 生成配乐