Create generate_prompts.py

generate_prompts.py

@@ -0,0 +1,62 @@
+import base64
+import re
+
+def insert_description(sentence, character, description):
+    """
+    Integrates the character and its description at the beginning of the sentence if the character is mentioned.
+    Parameters:
+        - sentence (str): The original sentence.
+        - character (str): The character to be described.
+        - description (str): The description of the character.
+    Returns:
+        str: The sentence modified to include the character and description at the beginning.
+    """
+    # Inserts character and description at the beginning of the sentence if the character is found.
+    character = character.lower()
+    # Remove everything after the newline character
+    cleaned_description = re.sub(r'\n.*', '', description)
+    # Remove non-alphabetic characters and quotes from the description
+    cleaned_description = re.sub(r'[^a-zA-Z\s,]', '', cleaned_description).replace("'", '').replace('"', '')
+    # Check if the character appears in the sentence
+    if character in sentence.lower():
+        # Insert the character and its description at the beginning of the sentence
+        modified_sentence = f"{character}: {cleaned_description.strip()}. {sentence}"
+        return modified_sentence
+    else:
+        return sentence
+
+def process_text(sentence, character_dict):
+    """
+    Enhances the given sentence by incorporating descriptions for each mentioned character.
+    Parameters:
+        - sentence (str): The original sentence.
+        - character_dict (dict): Dictionary mapping characters to their descriptions.
+    Returns:
+        str: The sentence with integrated character descriptions.
+    """
+    # Modifies sentences in the given text based on character descriptions.
+    modified_sentence = sentence  # Initialize with the original sentence
+
+    # Iterate through each character in the dictionary
+    for character, descriptions in character_dict.items():
+        for description in descriptions:
+            # Update the sentence with the character and its description
+            modified_sentence = insert_description(modified_sentence, character, description)
+    return modified_sentence
+
+
+def generate_prompt(text, sentence_mapping, character_dict, selected_style):
+    """
+    Generates a prompt for image generation based on the selected style and input text.
+    Parameters:
+        - style (str): The chosen illustration style.
+        - text (str): The input text for the illustration.
+    Returns:
+        tuple: A prompt string.
+    """
+    # Retrieve the enhanced sentence associated with the original text
+    enhanced_sentence = sentence_mapping.get(text, text)
+    image_descriptions = process_text(enhanced_sentence, character_dict)
+    # Define prompts and other parameters
+    prompt = f"Make an illustration in {selected_style} style from: {image_descriptions}"
+    return prompt