Update app.py

app.py

@@ -10,6 +10,21 @@ import numpy as np
 from gtts import gTTS
 import textwrap
 
+# Move the script generation to a standalone cached function
+@st.cache_data
+def generate_script_cached(prompt, style, length, model_output):
+    """Cached version of script generation"""
+    style_prompts = {
+        'Professional': "Write a clear, professional script about:",
+        'Creative': "Write an engaging, creative script about:",
+        'Educational': "Write an informative educational script about:"
+    }
+    
+    # Clean up the generated text
+    script = model_output[0]['generated_text']
+    script = script.replace(style_prompts[style], '').strip()
+    return script
+
 class EnhancedVideoGenerator:
     def __init__(self):
         # Use device agnostic setup
@@ -44,7 +59,6 @@ class EnhancedVideoGenerator:
             }
         }
 
-    @st.cache_data
     def generate_script(self, prompt, style, length):
         """Generate engaging script based on style and length"""
         style_prompts = {
@@ -55,6 +69,7 @@ class EnhancedVideoGenerator:
         
         max_length = int(length * 2.5)  # Approximate words needed for duration
         
+        # Generate the text
         output = self.text_generator(
             f"{style_prompts[style]} {prompt}. Keep it concise and engaging.",
             max_length=max_length,
@@ -62,10 +77,8 @@ class EnhancedVideoGenerator:
             temperature=0.7
         )
         
-        script = output[0]['generated_text']
-        # Clean up the generated text
-        script = script.replace(style_prompts[style], '').strip()
-        return script
+        # Use the cached function to process the output
+        return generate_script_cached(prompt, style, length, output)
 
     def create_styled_frame(self, text, theme, frame_number, total_frames):
         """Create a visually appealing frame with text and progress bar"""