Update scene_planner.py

scene_planner.py

@@ -1,20 +1,18 @@
+# In scene_planner.py
 from transformers import pipeline
 
-scene_splitter = pipeline("text2text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
+scene_splitter = pipeline("text-generation", model="microsoft/phi-2")
 
 def plan_scenes(script):
     prompt = (
-        "Split the following story into 10-15 scenes. For each scene, return a JSON with 'prompt' "
-        "(scene image description) and 'dialogue' (narration to speak). Make the prompt cartoon-friendly.\nStory:\n" + script
+        "Split this story into 10-15 cartoon scenes. Each should have a JSON with 'prompt' (image description) "
+        "and 'dialogue' (line of narration). Make it suitable for animation.\nStory:\n" + script
     )
-
     response = scene_splitter(prompt, max_new_tokens=1024, do_sample=False)[0]['generated_text']
 
     try:
         import json
         scenes = json.loads(response)
     except:
-        scenes = [
-            {"prompt": line.strip(), "dialogue": line.strip()} for line in script.split(".") if line.strip()
-        ]
+        scenes = [{"prompt": line.strip(), "dialogue": line.strip()} for line in script.split('.') if line.strip()]
     return scenes[:15]