Spaces:
Running
Running
| import random | |
| import os | |
| import gradio as gr | |
| import google.generativeai as genai | |
| # Initialize the Gemini model | |
| Model = genai.GenerativeModel('gemini-1.5-flash') | |
| def chat_short_story(length, genre, theme, tone, writing_style): | |
| """ | |
| Generates a creative short story using Gemini API. | |
| Parameters: | |
| length (int): Approximate word count for the story. | |
| genre (str): Genre of the story. | |
| theme (str): Central theme of the story. | |
| tone (str): Tone of the story. | |
| writing_style (str): Writing style for the story. | |
| Returns: | |
| str: The generated short story, or an error message if unsuccessful. | |
| """ | |
| # Retrieve the Gemini API key from the environment | |
| GEMINI_API_KEY = os.getenv("GOOGLE_API_KEY") | |
| if not GEMINI_API_KEY: | |
| return "Error: Gemini API key not found. Please set the GEMINI_API_KEY environment variable." | |
| # Configure the Gemini API | |
| try: | |
| genai.configure(api_key=GEMINI_API_KEY) | |
| except Exception as e: | |
| return f"Error: Failed to configure the Gemini API. Details: {e}" | |
| # Construct the prompt | |
| prompt = ( | |
| f"Write a creative short story of approximately {length} words in the {genre} genre. " | |
| f"Use a {writing_style} writing style with a {tone} tone. " | |
| f"The story should revolve around the theme of {theme}. " | |
| f"Ensure it is engaging and includes a suitable title and always label the title as Title." | |
| f"Make Sure to Never Repeat a Title and Story." | |
| ) | |
| # Generate a response using the Gemini model | |
| try: | |
| chat = Model.start_chat() | |
| response = chat.send_message(prompt) | |
| return response.text | |
| except Exception as e: | |
| return f"Error: Failed to generate story. Details: {e}" | |
| # Predefined options | |
| Length = [100, 250, 750] | |
| r_length = random.choice([l for l in Length if l >= 100]) | |
| Genre = [ | |
| "Fiction", "Nonfiction", "Drama", "Poetry", "Fantasy", "Horror", "Mystery", | |
| "Science Fiction", "Suspense", "Women's fiction", "Supernatural/Paranormal", "Young adult" | |
| ] | |
| r_genre = random.choice(Genre) | |
| Themes = [ | |
| "Love", "Redemption", "Forgiveness", "Coming of age", "Revenge", "Good vs evil", | |
| "Bravery and hardship", "The power of social status", "The destructive nature of love", | |
| "The fallibility of the human condition" | |
| ] | |
| r_themes = random.choice(Themes) | |
| Writing_Styles = ["Expository", "Narrative", "Descriptive", "Persuasive", "Creative"] | |
| r_Style = random.choice(Writing_Styles) | |
| Tones = ["Formal", "Optimistic", "Worried", "Friendly", "Curious", "Assertive", "Encouraging"] | |
| r_tones = random.choice(Tones) | |
| # Gradio Interface setup | |
| iface = gr.Interface( | |
| fn=chat_short_story, | |
| inputs=[ | |
| gr.Slider(value=100, label="Story Length", minimum=100, maximum=2500, step=50), | |
| gr.Dropdown(label="Story Genre", choices=Genre), | |
| gr.Dropdown(label="Story Theme", choices=Themes), | |
| gr.Dropdown(label="Writing Style", choices=Writing_Styles), | |
| gr.Dropdown(label="Story Tone", choices=Tones) | |
| ], | |
| outputs=gr.Textbox(label="Generated Story"), | |
| title="Welcome to Patrick's Story Generator", | |
| description="Generate creative short stories tailored to your preferences." | |
| ) | |
| iface.launch() | |