Update app.py

app.py

@@ -11,6 +11,7 @@ from mpl_toolkits.mplot3d import Axes3D
 import io
 import base64
 import os
+import re
 
 # Initialize text generation model (GPT-2)
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
@@ -31,37 +32,48 @@ def generate_speech(text):
     tts.tts_to_file(text=text, file_path=output_path)
     return output_path
 
-# Function to create a more advanced character animation
-def create_character_animation(text):
-    def update_point(num, data, line):
+# Function to extract keywords from the script
+def extract_keywords(script):
+    # Simple keyword extraction (you might want to use a more sophisticated method)
+    keywords = re.findall(r'\b\w+\b', script.lower())
+    return list(set(keywords))
+
+# Function to create a character animation based on script keywords
+def create_character_animation(script):
+    keywords = extract_keywords(script)
+    
+    def update_point(num, data, line, ax, keywords):
         line.set_data(data[:2, :num])
         line.set_3d_properties(data[2, :num])
-        return line
+        
+        # Update text with cycling keywords
+        keyword = keywords[num % len(keywords)]
+        ax.texts.clear()
+        ax.text2D(0.05, 0.95, keyword, transform=ax.transAxes)
+        
+        return line,
 
     fig = plt.figure()
     ax = fig.add_subplot(111, projection='3d')
 
-    # Generate a more complex 3D path for the character
-    t = np.linspace(0, 20, 100)
-    x = np.sin(t)
-    y = np.cos(t)
+    # Generate a 3D path influenced by the number of keywords
+    t = np.linspace(0, 2*np.pi*len(keywords)/10, 100)
+    x = np.sin(t) * (1 + len(keywords)/20)
+    y = np.cos(t) * (1 + len(keywords)/20)
     z = t/10
 
     data = np.array([x, y, z])
     line, = ax.plot(data[0, 0:1], data[1, 0:1], data[2, 0:1])
 
     # Setting the axes properties
-    ax.set_xlim3d([-1.0, 1.0])
+    ax.set_xlim3d([-2.0, 2.0])
     ax.set_xlabel('X')
-    ax.set_ylim3d([-1.0, 1.0])
+    ax.set_ylim3d([-2.0, 2.0])
     ax.set_ylabel('Y')
-    ax.set_zlim3d([0.0, 2.0])
+    ax.set_zlim3d([0.0, 4.0])
     ax.set_zlabel('Z')
 
-    # Adding text
-    ax.text2D(0.05, 0.95, text[:50], transform=ax.transAxes)
-
-    ani = animation.FuncAnimation(fig, update_point, frames=100, fargs=(data, line), interval=100, blit=False)
+    ani = animation.FuncAnimation(fig, update_point, frames=100, fargs=(data, line, ax, keywords), interval=100, blit=False)
     
     # Save animation as gif
     ani.save('character_animation.gif', writer='pillow')
@@ -84,7 +96,7 @@ def generate_kids_music_animation(theme):
 
 # Gradio Interface
 with gr.Blocks() as app:
-    gr.Markdown("## Advanced Character Animation Generator with Open Source Models")
+    gr.Markdown("## Script-based Character Animation Generator")
     
     with gr.Tab("Comedy Animation"):
         comedy_prompt = gr.Textbox(label="Enter comedy prompt")