Update stories.py

stories.py

@@ -244,6 +244,7 @@ def generateResponse(prompt, df):
 # -----------------------
 # DataFrame-Based Story Generation (for CSV/Excel files)
 # -----------------------
+# -----------------------
 def generate_story_from_dataframe(df, story_type):
     """
     Generate a data-based story from a CSV/Excel file.
@@ -256,22 +257,23 @@ def generate_story_from_dataframe(df, story_type):
         "free_form": "You are a professional storyteller. Using the following dataset in JSON format: " + df_json +
                      ", create an engaging and concise story. ",
         "children": "You are a professional storyteller writing stories for children. Using the following dataset in JSON format: " + df_json +
-                     ", create a fun, factual, and concise story appropriate for children. ",
+                    ", create a fun, factual, and concise story appropriate for children. ",
         "education": "You are a professional storyteller writing educational content. Using the following dataset in JSON format: " + df_json +
                      ", create an informative, engaging, and concise educational story. Include interesting facts while keeping it engaging. ",
         "business": "You are a professional storyteller specializing in business narratives. Using the following dataset in JSON format: " + df_json +
-                     ", create a professional, concise business story with practical insights. ",
+                    ", create a professional, concise business story with practical insights. ",
         "entertainment": "You are a professional storyteller writing creative entertaining stories. Using the following dataset in JSON format: " + df_json +
-                     ", create an engaging and concise entertaining story. Include interesting facts while keeping it engaging. "
+                           ", create an engaging and concise entertaining story. Include interesting facts while keeping it engaging. "
     }
     story_prompt = prompts.get(story_type, prompts["free_form"])
     full_prompt = (
         story_prompt +
-        "Write a story for a narrator meaning no labels of pages or sections the story should just flow. Divide your story into exactly 5 short and concise sections separated by [break]. " +
+        "Write a story for a narrator meaning no labels of pages or sections the story should just flow. Divide your story into exactly 5 very short and concise sections separated by [break]. " +
+        "Aim for a maximum of 3 sentences per section to ensure a quicker narration. " +
         "For each section, provide a brief narrative analysis and include, within angle brackets <>, a clear and plain-text description of a chart visualization that would represent the data. " +
         "Limit the descriptions by specifying only charts. " +
         "Ensure that your response contains only natural language descriptions examples: 'bar chart of', 'pie chart of' , 'histogram of', 'scatterplot of', 'boxplot of' , 'heatmap of etc' and nothing else."
-        
+
     )
 #
     try:
@@ -309,7 +311,7 @@ def generate_story_from_text(prompt_text, story_type):
     story_prompt = prompts.get(story_type, prompts["free_form"])
     response = model.generate_content(
         story_prompt +
-        "Write a short story for a narrator meaning no labels of pages or sections the story should just flow and narrated in 2 minutes or less. Divide your story into exactly 5 sections separated by [break]. For each section, include an image description inside <>."
+        "Write a short story for a narrator meaning no labels of pages or sections the story should just flow and narrated in 2 minutes or less. Divide your story into exactly 5 very short and concise sections separated by [break]. Aim for a maximum of 3 sentences per section. For each section, include a brief image description inside <>."
     )
     return response.text if response else None
 
@@ -323,18 +325,18 @@ def generate_story_from_wiki(wiki_url, story_type):
             "free_form": "You are a professional storyteller. Using the following Wikipedia info: " + wiki_text +
                          ", create an engaging and concise story. ",
             "children": "You are a professional storyteller for children. Using the following Wikipedia info: " + wiki_text +
-                         ", create a fun and concise story. ",
+                        ", create a fun and concise story. ",
             "education": "You are a professional storyteller. Using the following Wikipedia info: " + wiki_text +
                          ", create an educational and engaging story. ",
             "business": "You are a professional storyteller. Using the following Wikipedia info: " + wiki_text +
-                         ", create a professional business story. ",
+                        ", create a professional business story. ",
             "entertainment": "You are a professional storyteller. Using the following Wikipedia info: " + wiki_text +
-                         ", create an entertaining and concise story. "
+                             ", create an entertaining and concise story. "
         }
         story_prompt = prompts.get(story_type, prompts["free_form"])
         response = model.generate_content(
             story_prompt +
-            "Write a short story for a narrator meaning no labels of pages or sections the story should just flow and narrated in 2 minutes or less. Divide your story into exactly 5 sections separated by [break]. For each section, include an image description inside <>."
+            "Write a short story for a narrator meaning no labels of pages or sections the story should just flow and narrated in 2 minutes or less. Divide your story into exactly 5 very short and concise sections separated by [break]. Aim for a maximum of 3 sentences per section. For each section, include a brief image description inside <>."
         )
         return response.text if response else None
     except Exception as e:
@@ -397,18 +399,18 @@ def generate_story_from_bible(reference, story_type):
         "free_form": "You are a professional storyteller. Using the following Bible text: " + bible_text +
                      ", create an engaging and concise story. ",
         "children": "You are a professional storyteller for children. Using the following Bible text: " + bible_text +
-                     ", create a fun and concise story. ",
+                    ", create a fun and concise story. ",
         "education": "You are a professional storyteller. Using the following Bible text: " + bible_text +
                      ", create an educational and engaging story. ",
         "business": "You are a professional storyteller. Using the following Bible text: " + bible_text +
-                     ", create a professional business story. ",
+                    ", create a professional business story. ",
         "entertainment": "You are a professional storyteller. Using the following Bible text: " + bible_text +
-                     ", create an entertaining and concise story. "
+                         ", create an entertaining and concise story. "
     }
     story_prompt = prompts.get(story_type, prompts["free_form"])
     response = model.generate_content(
         story_prompt +
-        "Write a short story for a narrator meaning no labels of pages or sections the story should just flow and narrated in 2 minutes or less. Divide your story into exactly 5 sections separated by [break]. For each section, include a brief image description inside <>."
+        "Write a short story for a narrator meaning no labels of pages or sections the story should just flow and narrated in 2 minutes or less. Divide your story into exactly 5 very short and concise sections separated by [break]. Aim for a maximum of 3 sentences per section. For each section, include a brief image description inside <>."
     )
     return response.text if response else None
 
@@ -422,35 +424,36 @@ def generate_story_from_youtube(youtube_url, story_type):
             video_id = youtube_url.split("youtu.be/")[1].split("?")[0]
         else:
             raise ValueError("Invalid YouTube URL provided.")
-        
+
         # Retrieve the transcript as a list of dictionaries
         transcript_res = supadata.youtube.transcript(
-                video_id=video_id,
-                text=True
-            )
+            video_id=video_id,
+            text=True
+        )
         transcript_text = transcript_res.content
         # Define story prompts based on story_type, similar to the Wikipedia function
         prompts = {
             "free_form": "You are a professional storyteller. Using the following YouTube transcript: " + transcript_text +
                          ", create an engaging and concise story. ",
             "children": "You are a professional storyteller for children. Using the following YouTube transcript: " + transcript_text +
-                         ", create a fun and concise story. ",
+                        ", create a fun and concise story. ",
             "education": "You are a professional storyteller. Using the following YouTube transcript: " + transcript_text +
                          ", create an educational and engaging story. ",
             "business": "You are a professional storyteller. Using the following YouTube transcript: " + transcript_text +
-                         ", create a professional business story. ",
+                        ", create a professional business story. ",
             "entertainment": "You are a professional storyteller. Using the following YouTube transcript: " + transcript_text +
-                         ", create an entertaining and concise story. "
+                             ", create an entertaining and concise story. "
         }
         # Use the provided story_type, defaulting to free_form if not found
         story_prompt = prompts.get(story_type, prompts["free_form"])
-        
+
         # Append additional instructions for story structure
         full_prompt = story_prompt + (
-            "Write a short story for a narrator meaning no labels of pages or sections the story should just flow and narrated in 2 minutes or less. Divide your story into exactly 5 sections separated by [break]. "
+            "Write a short story for a narrator meaning no labels of pages or sections the story should just flow and narrated in 2 minutes or less. Divide your story into exactly 5 very short and concise sections separated by [break]. "
+            "Aim for a maximum of 3 sentences per section. "
             "For each section, include an image description inside <>."
         )
-        
+
         # Generate content using your model (assumes model.generate_content is available)
         response = model.generate_content(full_prompt)
         return response.text if response else None