Update app.py

app.py

@@ -9,7 +9,7 @@ import traceback
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image as RLImage
 from reportlab.lib.pagesizes import letter
 from reportlab.lib.styles import getSampleStyleSheet
-
+import re
 import google.generativeai as genai
 
 
@@ -160,7 +160,7 @@ Return:
 Only provide the raw values, no headers.
 """
         response = text_model.generate_content(prompt).text.strip()
-        parts = [line.strip() for line in response.split("\n") if line.strip()]
+        parts = [re.sub(r"^\d+\.\s*", "", line.strip()) for line in response.split("\n") if line.strip()]
         char_names = parts[0] if len(parts) > 0 else ""
         dialogue = parts[1] if len(parts) > 1 else ""
         speaker = parts[2] if len(parts) > 2 else ""