Spaces:
Runtime error
Runtime error
| 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 | |